文本主题模型之ldalda求解之gibbs采样算法

hx868 hx868     2023-01-29     483

关键词:

 本文是LDA主题模型的第二篇,读这一篇之前建议先读文本主题模型之LDA(一) LDA基础,同时由于使用了基于MCMC的Gibbs采样算法,如果你对MCMC和Gibbs采样不熟悉,建议阅读之前写的MCMC系列MCMC(四)Gibbs采样

 

1. Gibbs采样算法求解LDA的思路

    首先,回顾LDA的模型图如下:

技术分享图片

    在Gibbs采样算法求解LDA的方法中,我们的α,ηα,η是已知的先验输入,我们的目标是得到各个zdn,wknzdn,wkn对应的整体z,wz→,w→的概率分布,即文档主题的分布和主题词的分布。由于我们是采用Gibbs采样法,则对于要求的目标分布,我们需要得到对应分布各个特征维度的条件概率分布。

    具体到我们的问题,我们的所有文档联合起来形成的词向量ww→是已知的数据,不知道的是语料库主题zz→的分布。假如我们可以先求出w,zw,z的联合分布p(w,z)p(w→,z→),进而可以求出某一个词wiwi对应主题特征zizi的条件概率分布p(zi=k|w,z?i)p(zi=k|w→,z→?i)。其中,z?iz→?i代表去掉下标为ii的词后的主题分布。有了条件概率分布p(zi=k|w,z?i)p(zi=k|w→,z→?i),我们就可以进行Gibbs采样,最终在Gibbs采样收敛后得到第ii个词的主题。

    如果我们通过采样得到了所有词的主题,那么通过统计所有词的主题计数,就可以得到各个主题的词分布。接着统计各个文档对应词的主题计数,就可以得到各个文档的主题分布。

    以上就是Gibbs采样算法求解LDA的思路。

2. 主题和词的联合分布与条件分布的求解

    从上一节可以发现,要使用Gibbs采样求解LDA,关键是得到条件概率p(zi=k|w,z?i)p(zi=k|w→,z→?i)的表达式。那么这一节我们的目标就是求出这个表达式供Gibbs采样使用。

    首先我们简化下Dirichlet分布的表达式,其中(α)△(α)是归一化参数:

Dirichlet(p|α)=Γ(k=1Kαk)Kk=1Γ(αk)k=1Kpαk?1k=1(α)k=1Kpαk?1kDirichlet(p→|α→)=Γ(∑k=1Kαk)∏k=1KΓ(αk)∏k=1Kpkαk?1=1△(α→)∏k=1Kpkαk?1

 

    现在我们先计算下第d个文档的主题的条件分布p(zd|α)p(z→d|α),在上一篇中我们讲到αθdzdα→θd→z→d组成了Dirichlet-multi共轭,利用这组分布,计算p(zd|α)p(z→d|α→)如下:

p(zd|α)=p(zd|θd)p(θd|α)dθd=k=1Kpn(k)dkDirichlet(α)dθd=k=1Kpn(k)dk1(α)k=1Kpαk?1kdθd=1(α)k=1Kpn(k)d+αk?1kdθd=(nd+α)(α)(1)(2)(3)(4)(5)(1)p(z→d|α→)=∫p(z→d|θ→d)p(θd|α→)dθ→d(2)=∫∏k=1Kpknd(k)Dirichlet(α→)dθ→d(3)=∫∏k=1Kpknd(k)1△(α→)∏k=1Kpkαk?1dθ→d(4)=1△(α→)∫∏k=1Kpknd(k)+αk?1dθ→d(5)=△(n→d+α→)△(α→)

 

    其中,在第d个文档中,第k个主题的词的个数表示为:n(k)dnd(k), 对应的多项分布的计数可以表示为

nd=(n(1)d,n(2)d,...n(K)d)n→d=(nd(1),nd(2),...nd(K))

 

    有了单一一个文档的主题条件分布,则可以得到所有文档的主题条件分布为:

p(z|α)=d=1Mp(zd|α)=d=1M(nd+α)(α)p(z→|α→)=∏d=1Mp(z→d|α→)=∏d=1M△(n→d+α→)△(α→)

 

    同样的方法,可以得到,第k个主题对应的词的条件分布p(w|z,η)p(w→|z→,η→)为:

p(w|z,η)=k=1Kp(wk|z,η)=k=1K(nk+η)(η)p(w→|z→,η→)=∏k=1Kp(w→k|z→,η→)=∏k=1K△(n→k+η→)△(η→)

 

    其中,第k个主题中,第v个词的个数表示为:n(v)knk(v), 对应的多项分布的计数可以表示为

nk=(n(1)k,n(2)k,...n(V)k)n→k=(nk(1),nk(2),...nk(V))

 

    最终我们得到主题和词的联合分布p(w,z|α,η)p(w→,z→|α→,η→)如下:

p(w,z)p(w,z|α,η)=p(z|α)p(w|z,η)=d=1M(nd+α)(α)k=1K(nk+η)(η)p(w→,z→)∝p(w→,z→|α→,η→)=p(z→|α→)p(w→|z→,η→)=∏d=1M△(n→d+α→)△(α→)∏k=1K△(n→k+η→)△(η→)

 

    有了联合分布,现在我们就可以求Gibbs采样需要的条件分布p(zi=k|w,z?i)p(zi=k|w→,z→?i)了。需要注意的是这里的i是一个二维下标,对应第d篇文档的第n个词。

    对于下标ii,由于它对应的词wiwi是可以观察到的,因此我们有:

p(zi=k|w,z?i)p(zi=k,wi=t|w?i,z?i)p(zi=k|w→,z→?i)∝p(zi=k,wi=t|w→?i,z→?i)

 

    对于zi=k,wi=tzi=k,wi=t,它只涉及到第d篇文档和第k个主题两个Dirichlet-multi共轭,即:

αθdzdα→→θ→d→z→d
ηβkw(k)η→→β→k→w→(k)

 

    其余的M+K?2M+K?2个Dirichlet-multi共轭和它们这两个共轭是独立的。如果我们在语料库中去掉zi,wizi,wi,并不会改变之前的M+KM+K个Dirichlet-multi共轭结构,只是向量的某些位置的计数会减少,因此对于θd,βkθ→d,β→k,对应的后验分布为:

p(θd|w?i,z?i)=Dirichlet(θd|nd,?i+α)p(θ→d|w→?i,z→?i)=Dirichlet(θ→d|n→d,?i+α→)
p(βk|w?i,z?i)=Dirichlet(βk|nk,?i+η)p(β→k|w→?i,z→?i)=Dirichlet(β→k|n→k,?i+η→)

 

    现在开始计算Gibbs采样需要的条件概率:

p(zi=k|w,z?i)p(zi=k,wi=t|w?i,z?i)=p(zi=k,wi=t,θd,βk|w?i,z?i)dθddβk=p(zi=k,θd|w?i,z?i)p(wi=t,βk|w?i,z?i)dθddβk=p(zi=k|θd)p(θd|w?i,z?i)p(wi=t|βk)p(βk|w?i,z?i)dθddβk=p(zi=k|θd)Dirichlet(θd|nd,?i+α)dθd?p(wi=t|βk)Dirichlet(βk|nk,?i+η)dβk=θdkDirichlet(θd|nd,?i+α)dθdβktDirichlet(βk|nk,?i+η)dβk=EDirichlet(θd)(θdk)EDirichlet(βk)(βkt)(6)(7)(8)(9)(10)(11)(12)(13)(6)p(zi=k|w→,z→?i)∝p(zi=k,wi=t|w→?i,z→?i)(7)=∫p(zi=k,wi=t,θ→d,β→k|w→?i,z→?i)dθ→ddβ→k(8)=∫p(zi=k,θ→d|w→?i,z→?i)p(wi=t,β→k|w→?i,z→?i)dθ→ddβ→k(9)=∫p(zi=k|θ→d)p(θ→d|w→?i,z→?i)p(wi=t|β→k)p(β→k|w→?i,z→?i)dθ→ddβ→k(10)=∫p(zi=k|θ→d)Dirichlet(θ→d|n→d,?i+α→)dθ→d(11)?∫p(wi=t|β→k)Dirichlet(β→k|n→k,?i+η→)dβ→k(12)=∫θdkDirichlet(θ→d|n→d,?i+α→)dθ→d∫βktDirichlet(β→k|n→k,?i+η→)dβ→k(13)=EDirichlet(θd)(θdk)EDirichlet(βk)(βkt)

 

    在上一篇LDA基础里我们讲到了Dirichlet分布的期望公式,因此我们有:

EDirichlet(θd)(θdk)=nkd,?i+αks=1Knsd,?i+αsEDirichlet(θd)(θdk)=nd,?ik+αk∑s=1Knd,?is+αs
EDirichlet(βk)(βkt)=ntk,?i+ηtf=1Vnfk,?i+ηfEDirichlet(βk)(βkt)=nk,?it+ηt∑f=1Vnk,?if+ηf

 

    最终我们得到每个词对应主题的Gibbs采样的条件概率公式为:

p(zi=k|w,z?i)=nkd,?i+αks=1Knsd,?i+αsntk,?i+ηtf=1Vnfk,?i+ηfp(zi=k|w→,z→?i)=nd,?ik+αk∑s=1Knd,?is+αsnk,?it+ηt∑f=1Vnk,?if+ηf

 

    有了这个公式,我们就可以用Gibbs采样去采样所有词的主题,当Gibbs采样收敛后,即得到所有词的采样主题。

    利用所有采样得到的词和主题的对应关系,我们就可以得到每个文档词主题的分布θdθd和每个主题中所有词的分布βkβk。

3. LDA Gibbs采样算法流程总结

    现在我们总结下LDA Gibbs采样算法流程。首先是训练流程:

    1) 选择合适的主题数KK, 选择合适的超参数向量α,ηα→,η→

    2) 对应语料库中每一篇文档的每一个词,随机的赋予一个主题编号zz

    3)  重新扫描语料库,对于每一个词,利用Gibbs采样公式更新它的topic编号,并更新语料库中该词的编号。

    4) 重复第2步的基于坐标轴轮换的Gibbs采样,直到Gibbs采样收敛。

    5) 统计语料库中的各个文档各个词的主题,得到文档主题分布θdθd,统计语料库中各个主题词的分布,得到LDA的主题与词的分布βkβk。

 

    下面我们再来看看当新文档出现时,如何统计该文档的主题。此时我们的模型已定,也就是LDA的各个主题的词分布βkβk已经确定,我们需要得到的是该文档的主题分布。因此在Gibbs采样时,我们的EDirichlet(βk)(βkt)EDirichlet(βk)(βkt)已经固定,只需要对前半部分EDirichlet(θd)(θdk)EDirichlet(θd)(θdk)进行采样计算即可。

    现在我们总结下LDA Gibbs采样算法的预测流程:

    1) 对应当前文档的每一个词,随机的赋予一个主题编号zz

    2)  重新扫描当前文档,对于每一个词,利用Gibbs采样公式更新它的topic编号。

    3) 重复第2步的基于坐标轴轮换的Gibbs采样,直到Gibbs采样收敛。

    4) 统计文档中各个词的主题,得到该文档主题分布。

 

4. LDA Gibbs采样算法小结    

    使用Gibbs采样算法训练LDA模型,我们需要先确定三个超参数K,α,ηK,α→,η→。其中选择一个合适的KK尤其关键,这个值一般和我们解决问题的目的有关。如果只是简单的语义区分,则较小的KK即可,如果是复杂的语义区分,则KK需要较大,而且还需要足够的语料。

    由于Gibbs采样可以很容易的并行化,因此也可以很方便的使用大数据平台来分布式的训练海量文档的LDA模型。以上就是LDA Gibbs采样算法。

    后面我们会介绍用变分推断EM算法来求解LDA主题模型,这个方法是scikit-learn和spark MLlib都使用的LDA求解方法。

转载自 文本主题模型之LDA(二) LDA求解之Gibbs采样算法 - 刘建平Pinard - 博客园  https://www.cnblogs.com/pinard/p/6867828.html

文本主题模型之ldalda求解之变分推断em算法

 本文是LDA主题模型的第三篇,读这一篇之前建议先读文本主题模型之LDA(一)LDA基础,同时由于使用了EM算法,如果你对EM算法不熟悉,建议先熟悉EM算法的主要思想。LDA的变分推断EM算法求解,应用于SparkMLlib和Scikit-learn的LDA算法... 查看详情

文本主题模型之ldalda基础

https://www.cnblogs.com/pinard/p/6831308.html     http://www.360doc.com/content/16/0428/10/478627_554452907.shtml     LDA(LatentDirichletAllocation)是一 查看详情

文本主题模型之ldalda基础

...这个LDA的信息,参看之前写的线性判别分析LDA原理总结。文本关注于隐含狄利克雷分布对应的LDA。1.LDA贝叶斯模型    LDA是基 查看详情

采样之gibbs采样

前面我们讲到了M-H采样已经可以很好的解决蒙特卡罗方法需要的任意概率分布的样本集的问题。但是M-H采样有两个缺点:一是需要计算接受率,在高维时计算量大。并且由于接受率的原因导致算法收敛时间变长。二是有些高维数... 查看详情

gibbs采样完整解析与理解

...gibbs采样属于随机模拟抽样算法中的一种(一类近似求解的方法)。随机模拟的核心是对一个分布进行抽样,常用的抽样算法包括:1.接受-拒绝抽样;2)重要性抽 查看详情

文本主题模型之非负矩阵分解(nmf)(代码片段)

在文本主题模型之潜在语义索引(LSI)中,我们讲到LSI主题模型使用了奇异值分解,面临着高维度计算量太大的问题。这里我们就介绍另一种基于矩阵分解的主题模型:非负矩阵分解(NMF),它同样使用了矩阵分解,但是计算量和处... 查看详情

线性混合效应模型linearmixed-effectsmodels的部分折叠gibbs采样

本文介绍了线性混合效应模型的新型贝叶斯分析。该分析基于部分折叠的方法,该方法允许某些组件从模型中部分折叠。得到的部分折叠的Gibbs(PCG)采样器被构造成适合线性混合效应模型,预计会比相应的Gibbs采样器表现出更... 查看详情

使用r语言进行metroplis-in-gibbs采样和mcmc运行分析

原文链接:http://tecdat.cn/?p=12200对于许多模型,例如物流模型,没有共轭先验。因此,吉布斯采样不适用。这篇文章展示了我们如何使用Metropolis-Hastings(MH)从每次Gibbs迭代中的非共轭条件后验对象中进行采样–比网格方法更... 查看详情

机器学习:gibbs采样

1.什么是Gibbs采样Gibbs采样是MH算法的一种特例(α==1),因此可以保证Gibbs抽取的样本,也构成一个非周期不可约稳定收敛的马氏链;Gibbs采样适用于样本是两维或以上的情况;通过积分去除掉相关但是不感兴趣的变量,称为“collap... 查看详情

概率图模型之em算法

...极大算法)是一种迭代算法,用于求解含有隐变量的概率模型参数的极大似然估计(MLE)或极大后验概率估计(MAP)。EM算法是一种比较通用的参数估计算法,被广泛用于朴素贝叶斯、GMM(高斯混合模型)、K-means(K均值聚类)... 查看详情

c++算法主题系列之贪心算法的贪心之术(代码片段)

1.前言贪心算法是一种常见算法。是以人性之念的算法,面对众多选择时,总是趋利而行。因贪心算法以眼前利益为先,故总能保证当前的选择是最好的,但无法时时保证最终的选择是最好的。当然,在局部利益最大化的同时,... 查看详情

通信算法之120:数字信号处理-采样

       查看详情

dl:rbm学习算法——gibbs采样变分方法对比散度模拟退火

RBM学习算法——Gibbs采样、变分方法、对比散度、模拟退火在学习Hinton的stackautoencoder算法(论文 Reducing the Dimensionality of Data with Neural Networks)之前需要了解什么是RBM,现在就我学习的情况,查找的资料(... 查看详情

mcmc笔记:吉布斯采样(gibbs)

1介绍吉布斯采样是一种特殊的MH采样MCMC笔记Metropilis-Hastings算法(MH算法)_UQI-LIUWJ的博客-CSDN博客此时我们要采样的分布是一个高维的情况吉布斯采样的思想就是一维一维地进行采样,采某一个维度的时候固定其他的... 查看详情

c++算法主题系列之集结0-1背包问题的所有求解方案(代码片段)

1.前言背包问题是类型问题,通过对这一类型问题的理解和掌握,从而可以归纳出求解此类问题的思路和模板。背包问题的分类有:0-1背包问题,也称为不可分割背包问题。无限背包问题。判定性背包问题.带附属关系的背包问题... 查看详情

特定领域知识图谱融合方案:文本匹配算法之预训练simberternie-gram单塔模型等诸多模型

特定领域知识图谱融合方案:文本匹配算法之预训练模型SimBert、ERNIE-Gram文本匹配任务在自然语言处理中是非常重要的基础任务之一,一般研究两段文本之间的关系。有很多应用场景;如信息检索、问答系统、智能对话、文本鉴... 查看详情

svm之序列最小最优化求解算法(smo)

  查看详情

数据结构与算法之深入解析“文本左右对齐”的求解思路与算法示例(代码片段)

...其成为每行恰好有maxWidth个字符,且左右两端对齐的文本。你应该使用“贪心算法”来放置给定的单词;也就是说,尽可能多地往每行中放置单词。必要时可用空格’’填充,使得每行恰好有maxWidth个字符。要求尽... 查看详情