变分推断的原理推导

一.原理推导

变分推断(VI)要做的事情很朴素,那就是有一个复杂的难以求解的分布,比如后验概率分布:p(Z\mid X),这里X表示观测数据,Z表示参数或隐变量,VI就是利用一个简单可控的近似分布q(Z)去逼近目标p(Z\mid X),即:

q(Z)\rightarrow p(Z\mid X)

比如下图,黄色区域便是我们的目标分布,红线和绿线是我们构建的高斯分布,去近似目标分布


那么,自然地有个问题就产生了,红线近似的更好还是绿线近似的更好?显然,上面的图我们很难肉眼区分的开,所以我们需要找到一个量化的指标来评估两个分布的近似程度,我们可以使用KL距离,它的定义如下:

KL(q\mid\mid p)=\int q(Z)ln\{\frac{q(Z)}{p(Z\mid X)}\}dZ

显然,当q(Z)=p(Z\mid X)时,KL(q\mid\mid p)取得最小值0,所以我们接下来求解下面优化问题就可以得到最优的近似分布q^*(Z)了:

q^*(Z)=arg\min_{q(Z)}KL(q\mid\mid p)

但是KL公式中同样还包含有P(Z\mid X),这样我们在求解时依然会很困难,接下来我们推到一种等价的方式,我们首先可以将p(X)拆解为如下的等式:

p(X)=\frac{p(X,Z)}{p(Z\mid X)}

显然,上面的等式恒成立,然后对两边取对数,有:

ln\ p(X)=ln\ p(X,Z)-ln\ p(Z\mid X)

继续加入我们的q(Z),有:

ln\ p(X)=ln\ p(X,Z)-ln\ p(Z\mid X)\\ =ln\ \frac{p(X,Z)}{q(Z)}-ln\ \frac{p(Z\mid X)}{q(Z)}

接下里,对两边求在近似分布q(Z)上的期望,由于左边与Z无关,求期望后还是其自身,所以:

ln\ p(X)=\int q(Z)ln\{\frac{p(X,Z)}{q(Z)}\}dZ-\int q(Z)ln\{\frac{p(Z\mid X)}{q(Z)}\}dZ\\ =\int q(Z)ln\{\frac{p(X,Z)}{q(Z)}\}dZ+\int q(Z)ln\{\frac{p(q(Z)}{Z\mid X)}\}dZ\\ =\mathcal{L}(q)+KL(q\mid\mid p)

这里,\mathcal{L}被称为证据下界(evidence lower bound,ELBO),这时,对数似然,ELBO以及KL距离三者之间具有如下的关系:

所以,对KL(q\mid\mid p)的极小化等价于对\mathcal{L}(q)做极大化,而ELBO函数中包含的联合概率分布p(X,Z)往往易于求解,所以,我们的最终计算目标便是:

q^*(Z)=arg\max_{q(Z)}\int q(Z)ln\{\frac{p(X,Z)}{q(Z)}\}dZ

二.对q(Z)进行简化

有时候,我们为了方便计算会将Z划分为若干个互不相交的,每组记作Z_i,i=1,2,...,M(注意,每个Z_i可能包含多个变量),同时假设这些分组变量是相互独立的,那么有:

q(Z)=\prod_{i=1}^Mq_i(Z_i)

注意,这里每个q_i(Z_i)可以有不同的函数形式。接下来,我们考虑该形式下的最优解,由于上面的独立假设,我们对于q(Z)的优化问题,可以转换为依次对不同的q_i(Z_i)优化问题(其余的可以看做常数项,不会影响最优解),我们将上面的等式带入ELBO函数\mathcal{L}(\cdot)中,并分离出仅依赖于某一组因子的形式,比如q_j(Z_j)

\mathcal{L}(q)=\int\prod_i q_i(Z_i)[ln\ p(X,Z)-\sum_i ln\ q_i(Z_i)]dZ\\ =\int q_j(Z_j)[\int ln\ p(X,Z)\prod_{i\neq j}q_i(Z_i)dZ_i]dZ_j-\int q_j(Z_j)ln\ q_j(Z_j)dZ_j+const\\ =\int q_j(Z_j)ln\ \tilde{p}(X,Z_j)dZ_j-\int q_j(Z_j)ln\ q_j(Z_j)dZ_j\\ =-KL(q_j(Z_j)\mid\mid\tilde{p}(X,Z_j))
这里,我们定义:

ln\ \tilde{p}(X,Z_j)=\int ln\ p(X,Z)\prod_{i\neq j}q_i(Z_i)dZ_i+const=E_{i\neq j}[ln\ p(X,Z)]+const

这里const为常数项,最优解可以直接观测出来了,那就是使得KL(q_j(Z_j)\mid\mid\tilde{p}(X,Z_j))为0的解,那就只有两个分布相等情况下成立,即:

q_j^*(Z_j)=\tilde{p}(X,Z_j)\\

消去const,可以写作如下:

\tilde{p}(X,Z_j)=\frac{exp(E_{i\neq j}[ln\ p(X,Z)])}{\int exp(E_{i\neq j}[ln\ p(X,Z)])dZ_j}(const即是分母部分取负对数)

通常,为了方便计算,用的更多的还是下面的表达式(下面的表达式后续会反复用到):

ln\ q_j^*(Z_j)=E_{i\neq j}[ln\ p(X,Z)]+const

因为对于复杂的计算,最后将多项的const合并在一起处理更为方便,因为它主要起着归一化系数的作用,而这个系数可以通过观测得出(比如上面等式中的分母项),不必刻意去计算。

©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 216,496评论 6 501
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 92,407评论 3 392
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 162,632评论 0 353
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 58,180评论 1 292
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 67,198评论 6 388
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 51,165评论 1 299
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 40,052评论 3 418
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 38,910评论 0 274
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 45,324评论 1 310
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 37,542评论 2 332
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 39,711评论 1 348
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 35,424评论 5 343
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 41,017评论 3 326
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 31,668评论 0 22
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 32,823评论 1 269
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 47,722评论 2 368
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 44,611评论 2 353

推荐阅读更多精彩内容