舍弃 Python,为什么知乎选用 Go 重构推荐系统?

阅读数:12422 2018 年 12 月 26 日

AI 前线导读:知乎从问答起步,已逐步成长为一个大规模的综合性知识内容平台,截止目前,用户数突破 2.2 亿,有超过 3000 万的问题被提出,并获得超过 1.3 亿个回答。同时,知乎内还沉淀了数量众多的优质文章、电子书以及其它付费内容。

因此,在链接人与知识的路径中,知乎存在着大量的推荐场景。粗略统计,目前除了首页推荐之外,我们已存在着 20 多种推荐场景;并且在业务快速发展中,不断有新的推荐业务需求加入。在这个背景之下,构建一个较通用的且便于业务接入的推荐系统就变成不得不做的事了。

重构推荐系统需要考虑哪些因素?如何做技术选型?重构的过程中会遇到哪些坑?希望知乎的踩坑经验能给你带来一些思考。

更多优质内容请关注微信公众号“AI 前线”(ID:ai-front)

背景

image

在讲通用架构的设计之前,我们一起回顾一下推荐系统的总体流程和架构。通常,因为模型所需特征及排序的性能考虑,我们通常将简单的推荐系统分为召回层和 ranking 层,召回层主要负责从候选集合中粗排选择待排序集合,之后获取 ranking 特征,经过排序模型,挑选出推荐结果给用户。

image

简单推荐模型适合一些推荐结果要求单一,只对单目标负责的推荐场景,比如新闻详情页推荐、文章推荐等等。但在现实中,作为通用的推荐系统来说,其需要考虑用户的多维度需求,比如用户的多样性需求、时效性需求、结果的满足性需求等。因此就需要在推荐过程中采用多个不同队列,针对不同需求进行排序,之后通过多队列融合策略,从而满足用户不同的需求。

image

从我们知乎来说,也大体是这样一个发展路线,比如今年的 7 月份时,因为一些业务快速发展且架构上相对独立的历史原因,我们的推荐系统存在多套,并且架构相对简单。以其中一个推荐架构设计相对完善的系统为例,其总体架构是这样的。可以看出,这个架构已经包含了召回层和 ranking 层,并且还考虑了二次排序。

image

那么存在哪些问题呢?

首先,对多路召回支持不友好。现有架构的召回是耦合在一起的,因此开发调研成本高,多路召回接入相对困难。

然后,召回阶段只使用 redis 作为召回基础。redis 有很多优点,比如查询效率高,服务较稳定。但将其作为所有召回层的基础,就放大了其缺点,第一不支持稍复杂的召回逻辑,第二无法进行大量结果的召回计算,第三不支持 embedding 的召回。

第三点,总体架构在实现时,架构逻辑剥离不够干净,使得架构抽样逻辑较弱,各种通用特征和通用监控建设都较困难。

第四点,我们知道,在推荐系统中,特征日志的建设是非常重要的一个环节,它是推荐效果好坏的重要基础之一。但现有推荐系统框架中,特征日志建设缺乏统一的校验和落地方案,各业务『各显神通』。

第五点,当前系统是不支持多队列融合的,这就严重限制了通用架构的可扩展性和易用性。
因此,我们就准备重构知乎的通用推荐服务框架。

重构之路

在重构前的考虑

第一,语言的选择。早期知乎大量的服务都是基于 Python 开发的,但在实践过程中发现 Python 资源消耗过大、不利用多人协同开发等各种问题,之后公司进行了大规模的重构,现在知乎在语言层面的技术选型上比较开放,目前公司内部已有 Python、Scala、Java、Golang 等多种开发语言项目。那么对于推荐系统服务来说,由于其重计算,多并发的特点,语言的选择还是需要考虑的。

第二,架构上的考虑,要解决支持多队列混排和支持多路召回的问题,并且其设计最好是支持可插拔的。

第三,召回层上,除了传统的 redis 的 kv 召回(部分 cf 召回,热门召回等等应用场景),我们还需要考虑一些其他索引数据库,以便支持更多索引类型。

首先我们先看语言上的选择,先总体上比较一下各种语言的特点,我们简单从如下几个方面进行比较。

从性能上,依照公开的 benchmark,Golang 和 Java、Scala 大概在一个量级,是 Python 的 30 倍左右。其次 Golang 的编译速度较快,这点相对于 Java、Scala 具有比较明显的优势,再次其语言特性决定了 Golang 的开发效率较高,此外因为缺乏 trycatch 机制,使得使用 Golang 开发时对异常处理思考较多,因此其上线之后维护成本相对较低。但 Golang 有个明显缺陷就是目前第三方库较少,特别跟 AI 相关的库。

image

那么基于以上优缺点,我们重构为什么选择 Golang?

1、Golang 天然的优势,支持高并发并且占用资源相对较少。这个优势恰恰是推荐系统所需要的,推荐系统存在大量需要高并发的场景,比如多路召回,特征计算等等。

2、知乎内部基础组件的 Golang 版生态比较完善。目前我们知乎内部对于 Golang 的使用越来越积极,大量基础组件都已经 Golang 化,包括基础监控组件等等,这也是我们选择 Golang 的重要原因。

但我需要强调一点,语言的选择不是只有唯一答案的,这是跟公司技术和业务场景结合的选择。

讲完语言上的选择,那么为了在重构时支持多队列混排和支持多路召回,我们架构上是如何来解决的?

这点在设计模式比较常见,就是『抽象工厂模式』:首先我们构建队列注册管理器,将回调注册一个 map 中,并将当前服务所有队列做成 json 配置的可自由插拔的模式,比如如下配置,指定一个服务所需要的全部队列,存入 queues 字段中,

image

通过 name 来从注册管理器的 map 中调取相应的队列服务,

image

之后呢我们就可以并发进行多队列的处理。

image

对于多路召回,及整个推荐具体流程的可插拔,与上面处理手法类似,比如如下队列

image

我们可以指定所需召回源,指定 merger 策略等等,当某个过程不需处理,会按自动默认步骤处理,这样在具体 Queue 的实现中就可以通过如下简单操作进行自由配置。

image

我们讲完了架构上一些思考点和具体架构实现方案,下面就是关于召回层具体技术选型问题。

我们先回顾一下,在常用的推荐召回源中,有基于 topic(tag)的召回、实体的召回、地域的召回、CF(协同过滤)的召回以及 NN 产生的 embedding 召回等等。那么为了能够支持这些召回,技术上我们应该如何来实现呢?

我们先从使用角度看一下常用的 NoSQL 数据库产品,Redis 是典型的 k-v 存储,其简单、并且高性能使得其在业内得到大量使用,但不支持复杂查询的问题也让 Redis 在召回复杂场景下并不占优,因此一般主要适用于 kv 类的召回,比如热门召回,地域召回,少量的 CF 召回等等。而 HBase 作为海量数据的列式存储数据库,有一个明显缺点就是复杂查询性能差,因此一般适合数据查询量大,但查询简单的场景,比如推荐系统当中的已读已推等等场景。而 ES 其实已经不算是一个数据库了,而是一个通用搜索引擎范畴,其最大优点就是支持复杂聚合查询,因此将其用户通用基础检索,是一个相对适合的选择。

image

我们上面介绍了通用召回的技术选型,那么 embedding 召回如何来处理呢,我们的方案是基于 Facebook 开源的 faiss 封装,构建一个通用 ANN(近似最近邻)检索服务。faiss 是为稠密向量提供高效相似度搜索和聚类的框架。其具有如下特性:1、提供了多种 embedding 召回方法;2、检索速度快,我们基于 python 接口封装,影响时间在几 ms-20ms 之间;3、c++ 实现,并且提供了 python 的封装调用;4、大部分算法支持 GPU 实现。

从以上介绍可以看出,在通用的推荐场景中,我们召回层大体是基于 ES+Redis+ANN 的模式进行构建。ES 主要支持相对复杂的召回逻辑,比如基于多种 topic 的混合召回;redis 主要用于支持热门召回,以及规模相对较小的 CF 召回等;ANN 主要支持 embedding 召回,包括 nn 产出的 embedding、CF 训练产出的 embedding 等等。

介绍完以上思考点,我们总体的架构就基本成型了,具体如下图所示。该框架可以支持多队列融合,并且每个队列也支持多路召回,从而对于不同推荐场景能够较好的支持,另外,我们召回选择了 ES+Redis+ANN 的技术栈方案,可以较好支持多种不同类型召回,并达到服务线上的最终目的。

image

重构遇到的一些问题及解决方案

1、离线任务和模型的管理问题。我们做在线服务的都有体会,我们经常容易对线上业务逻辑代码更关注一些,而往往忽视离线代码任务的管理和维护。但离线代码任务和模型在推荐场景中又至关重要。因此如何有效维护离线代码和任务,是我们面临的第一个问题。

2、特征日志问题。在推荐系统中,我们常常会遇到特征拼接和特征的『时间穿越』的问题。所谓特征时间穿越,指的是模型训练时用到了预测时无法获取的『未来信息』,这主要是训练 label 和特征拼接时时间上不够严谨导致。如何构建便捷通用的特征日志,减少特征拼接错误和特征穿越,是我们面临的第二个问题。

3、服务监控问题。一个通用的推荐系统应该在基础监控上做到尽可能通用可复用,减少具体业务对于监控的开发量,并方便业务定位问题。

4、离线任务和模型的管理问题。

在包括推荐系统的算法方向中,需要构建大量离线任务支持各种数据计算业务,和模型的定时训练工作。但实际工作中,我们往往忽略离线任务代码管理的重要性,当时间一长,各种数据和特征的质量往往无法保证。为了尽可能解决这样的问题,我们从三方面来做,第一,将通用推荐系统依赖的离线任务的代码统一到一处管理;第二,结合公司离线任务管理平台,将所有任务以通用包的形式进行管理,这样保证所有任务的都是依赖最新包;第三,建设任务结果的监控体系,将离线任务的产出完整监控起来。

image

5、特征日志问题。

Andrew Ng 之前说过:『挖掘特征是困难、费时且需要专业知识的事,应用机器学习其实基本上是在做特征工程。』我们理想中的推荐系统模型应该是有干净的 Raw Data,方便处理成可学习的 Dataset,通过某种算法学习 model,来达到预测效果不断优化的目的。

image

但现实中,我们需要处理各种各样的数据源,有数据库的,有日志的,有离线的,有在线的。这么多来源的 Raw Data,不可避免的会遇到各种各样的问题,比如特征拼接错误,特征『时间穿越』等等。

image

这里边反应的一个本质问题是特征处理流程的规范性问题。那么我们是如何来解决这一点呢,首先,我们用在线代替了离线,通过在线落特征日志,而不是 Raw Data,并统一了特征日志 Proto,如此就可以统一特征解析脚本。

image

6、服务监控问题。

在监控问题上,知乎搭建了基于 StatsD + Grafana + InfluxDB 的监控系统,以支持各种监控日志的收集存储及展示。基于这套系统,我们可以便捷的构建自己微服务的各种监控。

image

我们这里不过多介绍通用监控系统,主要介绍下,基于推荐系统我们监控建设的做法。首先先回顾一下我们推荐系统的通用设计上,我们采用了『可插拔』的多队列和多召回的设计,那么可以在通用架构设计获取到各种信息,比如业务线名,业务名,队列名,process 名等等。如此,我们就可以将监控使用如下方式实现,这样就可以通用化的设计监控,而不需各个推荐业务再过多设计监控及相关报警。

image

image

按照如上实现之后,我们推荐系统的监控体系大概是什么样子?首先各个业务可以通过 grafana 展示页面进行设置。我们可以看到各个 flow 的各种数据,以及召回源的比例关系,还有特征分布,ranking 得分分布等等。

image

未来挑战

讲完了遇到的一些问题之后,我们来看一下未来的挑战。随着业务的快速发展,数据和规模还在不断扩张,架构上还需要不断迭代;第二点,随着推荐业务越来越多,策略的通用性和业务之间的隔离如何协调一致;第三点,资源调度和性能开销也需要不断优化;最后,多机房之间数据如何保持同步也是需要考虑的问题。

总结

最后,我们做个简单总结:第一点,重构语言的选择,关键要跟公司技术背景和业务场景结合起来;第二点,架构尽量灵活,并不断自我迭代;第三点,监控要早点开展,并尽可能底层化、通用化。

团队介绍

我们是知乎的推荐技术团队,属于知乎的技术中台,主要为公司各个业务方提供完整的推荐服务,其中包括问题路由、知乎大学的推荐和搜索、回答推荐、文章推荐、视频推荐、和知乎个性化 Push 等多个推荐场景。在这里,我们除了建设通用推荐服务架构之外,还不断将最新的推荐算法应用到具体推荐业务中,提升用户体验,满足用户需求。

收藏

评论

微博

发表评论

注册/登录 InfoQ 发表评论

最新评论

Zhranklin 2019 年 01 月 02 日 10:46 0 回复
scala语法特性都不知道甩开go几条街了。。当然了本身也是有缺陷的, 包括语法特性有些过于复杂, 编译慢(比java还慢很多), 以及jvm相对于golang与c等语言内存占用比较大等
没有更多了