【业界】基于腾讯Angel的LDA*入选VLDB，超越微软LightLDA

港控/mmm° 2022-06-09 11:14 208阅读 0赞

--------------------

> 2017年8月27日-9月1日， 数据库领域的国际顶级学术会议VLDB(Very Large DataBase) 在德国慕尼黑召开。 腾讯TEG数据平台部和北京大学联合撰写的大规模主题模型系统的论文《`LDA*`：A Robust and Large-scale Topic Modeling System》， 入选了今年VLDB的Research Track，并获邀在大会上进行Oral Presentation。

![1240][]

VLDB会议由VLDB基金会赞助，和另外两大会议SIGMOD、ICDE构成了数据库领域的三个顶级会议，而它是这3个会议中，含金量最高的一个。VLDB的论文接受率总体很低，只有创新性很高、贡献很大的论文才有机会被录用。近年来随着大数据的发展，VLDB的领域也从数据库扩展到大数据相关领域，与时俱进的同时，依然保持着对文章质量的严格把控。

本次论文的主要内容，是研究如何在工业界的环境中建立一个大规模的主题模型训练系统。在这之前，已经有很多的相关的系统工作，包括YahooLDA，微软的LightLDA以及Petuum等，而`LDA*` 则解决了 **鲁棒的采样性能** 以及 **词分布倾斜** 这两个难题，取得了更好的性能。`LDA*` 构建于腾讯开源的系统Angel之上，得益于Angel开放的参数服务器架构，良好的扩展性以及优秀的编程接口设计，它可以轻松处理TB级别的数据和十亿维度的主题模型。

*  VLDB: [http://www.vldb.org/pvldb/vol10/p1406-yu.pdf][http_www.vldb.org_pvldb_vol10_p1406-yu.pdf]
 *  Github: [https://github.com/tencent/angel][https_github.com_tencent_angel]

## 主题模型：用数学框架捕捉文档潜在语义概率分布 ##

**主题模型（Topic Model）**在机器学习和自然语言处理等领域是用来在一系列文档中发现抽象主题的一种统计模型。一篇文章通常包含多种主题（比如“猫”、“狗”），而且每个主题所占比例各不相同。主题模型试图用数学框架来体现文档的这种特点，本质上，它是一个对文本建模的概率图模型。

在主题模型中，每个文档被看成一个**话题（Topic）** 的分布，而将每个话题被看成一个在词语上的分布**(Topic Distribution of Words)**。通过主题模型对文本进行建模，将文档表示成**话题的分布概率（Topic Probability）**，从而可以对文档进行聚类等分析。主题模型最初是运用于自然语言处理相关方向，现在已经用于多个领域，比如推荐系统、广告CTR预估，用户兴趣分类……

![1240 1][]

在工业界的场景下训练主题模型，经常会遇到3个问题：

1.  训练的数据规模非常大，2T大小的样本，3000亿个Token，要求在几个小时内跑出结果
2.  面对的数据和参数复杂，在各种各样的数据和参数配置下，都必须要有较好的性能
3.  所有的任务都是在一个集中的集群上运行，系统需要具有较高的可扩展性和鲁棒性

为此，`LDA*`从模型和工程上，对传统的LDA进行了2个针对性优化，使得它有广泛的适用性和良好的性能，成为一个大规模的高性能主题模型系统。

## 优化1：吉布斯采样（Hybrid Sampler） ##

吉布斯采样是一种基于马尔科夫蒙特卡罗的采样方法，用于从一个复杂的概率分布中进行采样，常常用于求解大规模主题模型。但最原始的吉布斯采样算法，Collapsed Gibbs Sampling（CGS）拥有O(K)的算法复杂度（K->Topic Number），使得在大规模数据以及话题个数较多的情况下，求解效率非常低下。因此有很多研究工作考虑如何降低采样算法的复杂度。

目前业界已有的LDA方法包括**AliasLDA，F+LDA，LightLDA和WarpLDA**。其中AliasLDA和F+LDA利用主题模型参数的稀疏性质使得模型的采样复杂度降低到了O(Kd)，而的上限是文档的长度和话题个数K中的较小值；LightLDA和WarpLDA则利用Metropolis-Hastings方法，逐一地从两个较为简单的proposal分布中进行采样(qdoc和qword)，再根据Metropolis-Hastings方法计算采样结果的接受率，因为利用了Alias Table来进行采样，LightLDA和WarpLDA每次采样只需要O(1)的计算复杂度。

![1240 2][]

但是，O(1)的采样复杂度并不意味着更快的收敛速度。由于Metropolis-Hastings方法中存在接受率，因此从概率的角度上来看，LightLDA和WarpLDA平均需要O(Kd)次采样操作才能产生一个可接受的采样样本，从而发生一次状态转移；而AliasLDA和F+LDA虽然每次采样操作需要O(

1π

)的复杂度，但是每次采样都能够产生一个样本。因此，Sparse-Aware Samplers和Metropolis-Hastings Samplers之间存在一个tradeoff，单一的一种sampler并不能够适用于所有的数据集或者参数设置，只有发现这个tradeoff并且有效地将这两种samplers结合起来，才能达到最佳的性能。

![1240 3][]

为了发现这个tradeoff，Angel团队做了详尽的实验，并找到两种samplers的交叉点。基于这个交叉点，`LDA*`有效地将F+LDA和WarpLDA结合起来，设计了一个新的Hybrid Sampler。它采用了两个启发式的规则来构建这个Hybrid Sampler。在第一个规则中，将数据集分成两个部分，文档较长的数据集和文档较短的数据集，然后使用F+LDA来采样较短的文档集合，使用WarpLDA采样较长的文档集合。在第二个规则中，将两个具有不同收敛速度的sampler结合起来，在一次迭代中，每个token话题的采样都能产生一个样本，因此对于WarpLDA来说，需要动态地设定其MH步长，而对于F+LDA来说，则不需要进行任何改变。

根据实验，Hybrid Sampler在PubMED数据集上获得与F+LDA相当的甚至更好的性能，而在Tencent数据集上也可以获得比WarpLDA更好的性能。总的来说Hybrid Sampler在所有实验数据集上和参数配置下，都能获得最好的性能。

## 优化2：非对称架构，实现模型并行 ##

![1240 4][]

主题模型的训练常常需要大规模的数据集和较大的话题个数，因此分布式训练常常用于进行大规模的主题模型训练。已有的系统，包括**LightLDA，YahooLDA和Petuum**，其实都是采用参数服务器的架构来进行分布式训练，但是由于词分布的倾斜特性，标准的参数服务器架构会在Worker端产生较大的网络通信开销，因此每个worker几乎都需要把整个词-话题（Word-Topic）矩阵模型从PSServer上拉取下来，这个对性能会有很大的影响。

在实际的测试下，由于词分布的倾斜性，导致大部分的网络开销都产生于长尾的词语，这些长尾的词语产生的网络开销造成了在数据量和模型参数较大时带来的性能损失。因此，`LDA*` 对于这类长尾词语进行了特殊的处理，将一部分长尾词语的采样推送到PSServer端进行，从而避免了对词-话题矩阵的拉取操作。由于在这样的架构中，PSServer不仅仅作为一个分布式存储，还参与了一定的计算任务，从而某种程度上实现了模型并行，这种架构为非对称架构。依托于Angel良好的接口设计和psFunc功能，`LDA*`的这种非对称架构可以非常轻松的实现，而不用对Core进行任何的修改。

## 性能数据：在腾讯真实的推荐数据集上最高是原有系统的 9 倍 ##

为了进行全面的评测，`LDA*` 和之前开源的Petuum，以及微软的LightLDA，在2个数据集上进行对比。一个是开源的PubMED，一个是腾讯真实的推荐数据集。实验结果表明，在数据量越大的情况下，`LDA*`的优势越明显。

![1240 5][]

PubMED数据集上，`LDA*`的速度，分别是Petuum和LightLDA的5倍。

![1240 6][]

在腾讯真实的推荐数据集上，`LDA*`是Petuum的9倍，是lightLDA的2.6倍。

为了更好的体现`LDA*` 的性能，在腾讯内部生产集群上，`LDA*`使用更大的数据集进行训练。从下图的数据集可以看到，最大的一个数据集有3000亿个token，大约1.8TB的数据量。在所有的任务中，话题数设置为8000，具体数据集大小如下：

<table> 
 <thead> 
  <tr> 
   <th>Dataset</th> 
   <th>docs</th> 
   <th>words</th> 
   <th>tokens</th> 
   <th>Average doc length</th> 
   <th>Sparse text size</th> 
  </tr> 
 </thead> 
 <tbody> 
  <tr> 
   <td>Production1</td> 
   <td>159M</td> 
   <td>3759618</td> 
   <td>19B</td> 
   <td>122</td> 
   <td>90GB</td> 
  </tr> 
  <tr> 
   <td>Production2</td> 
   <td>507M</td> 
   <td>5588297</td> 
   <td>60B</td> 
   <td>119</td> 
   <td>270GB</td> 
  </tr> 
  <tr> 
   <td>Production3</td> 
   <td>2.2B</td> 
   <td>3911813</td> 
   <td>308B</td> 
   <td>138</td> 
   <td>1.8TB</td> 
  </tr> 
 </tbody> 
</table>

测试性能如下，可以看出`LDA*`能够在资源充足的情况下，能得到非常好的扩展性，并且能够扩展至数千个Worker，即便对TB级别的数据进行训练，时间可以控制在小时级别，很好的满足了生产系统的需要。

![1240 7][]

## 总结 ##

`LDA*` 无论是在模型准确度还是性能上，都有不俗的表现和可扩展性，目前已经应用于腾讯的多个业务场景中，包括广点通推荐，用户画像，腾讯视频等……

未来，伴随着Angel的发展和推广，希望业界会有更多的公司，能够从`LDA*`中受益，轻松拥有大规模主题建模的能力。

**新智元源链接：** \[【业界】基于腾讯Angel的`LDA*`入选VLDB，超越微软LightLDA  
\] ([https://mp.weixin.qq.com/s/Ib0Q50c9OsSZRWPQPcIn0g][https_mp.weixin.qq.com_s_Ib0Q50c9OsSZRWPQPcIn0g])

[1240]: /images/20220609/d7bb92ac847b416fa6274c6e5bf2e10f.png
[http_www.vldb.org_pvldb_vol10_p1406-yu.pdf]: http://www.vldb.org/pvldb/vol10/p1406-yu.pdf
[https_github.com_tencent_angel]: https://github.com/tencent/angel
[1240 1]: /images/20220609/63233d9f7897404f8eaee93448456630.png
[1240 2]: /images/20220609/cb5bcb78d18f48e7820c7c674cca4b2a.png
[1240 3]: /images/20220609/461559d14154465b8e815e931a82033d.png
[1240 4]: /images/20220609/80d70a7ca380455c90fa3674b4cf41f0.png
[1240 5]: /images/20220609/cb34d98778814fa7b59869db1f1ad54a.png
[1240 6]: /images/20220609/e8aa37eff42c4d12a720289918812632.png
[1240 7]: /images/20220609/c96d31a601794f07b36539a705066f12.png
[https_mp.weixin.qq.com_s_Ib0Q50c9OsSZRWPQPcIn0g]: https://mp.weixin.qq.com/s/Ib0Q50c9OsSZRWPQPcIn0g