为什么贝叶斯统计中后验分布(posterior distortion)没有闭式解,以及变分推断的形式

参考:
StackExchange
Variational Inference: A Review for Statisticians

这个问题是变分推断(variational inference)的一个基础问题。很奇怪的是,我没有在中文社区发现这个问题的详细解释。一句话总结,如果求解过程出现了不可约掉的积分,该过程就是不可直接计算的(intractable),或者说没有闭式解。而机器学习所有解决该问题的办法,就是巧妙地去掉积分
下面通过一个例子,来解释这个问题。

问题描述

符号定义\textbf{x}=x_{1:n}表示观察变量,\textbf{z}=z_{1:m}表示隐变量,它们的联合概率密度为p(\textbf{x}, \textbf{z})
目标:求解条件概率p(\textbf{z}|\textbf{x})。在贝叶斯框架下,这一过程叫做推断,而p(\textbf{z}|\textbf{x})叫做后验概率。
后验概率密度
p(\textbf{z}|\textbf{x})=\frac{p(\textbf{z}, \textbf{x})}{p(\textbf{x})}=\frac{p(\textbf{z}, \textbf{x})}{\int p(\textbf{z}, \textbf{x})d\textbf{z}}
其中p(\textbf{x})被称为“证据”(evidence)
下面的例子用来说明,证据的积分形式通常没有闭式解

例子:混合高斯模型的后验概率估计

设有K个混合分支,分支的均值集合为\mathbf{\mu}=\{\mu_1,...,\mu_K\},这些均值服从高斯分布\mathcal{N}(0,\sigma^2),其中参数\sigma^2表示方差。采样K次得到\mathbf{\mu}
每个观察数据x_i的产生过程如下:从类别{1,...,K}中选出一个分支,记为c_i,其数学形式为one-hot(一个长为K的向量,其中一位是1,其余位是0)。从(另一个)高斯分布\mathcal{N}({c_i}^T\mu,1)采样,得到x_i。连续采样n次。
即:
\mu_k \sim \mathcal{N}(0,\sigma^2),其中k=1,...,K
c_i \sim Categorical(\frac{1}{K},...,\frac{1}{K}),其中i=1,...,n
x_i|c_i,\mu \sim \mathcal{N}({c_i}^T\mu,1),其中i=1,...,n
联合概率密度为:
p(\mathbf{\mu},\textbf{c},\textbf{x})=p(\mathbf{\mu})\prod_{i=1}^{n}p(c_i)p(x_i|c_i,\mathbf{\mu})
隐变量\mathbf{z}=\{\mathbf{\mu},\mathbf{c}\}
证据为:
p(\textbf{x})=\int p(\mathbf{\mu})\prod_{i=1}^{n}\sum_{c_i}p(c_i)p(x_i|c_i,\mathbf{\mu})d\mu
=\int p(\mu_1,...\mu_K)\prod_{i=1}^{n}\sum_{c_i}p(c_i)p(x_i|c_i,\mu_1,...\mu_K)d\mu_1d\mu_2...d\mu_K
上式中,每一个\mu_k都无法从连乘项里分离出来,所以计算该积分的复杂度为\mathcal{O}(K^n),为K的指数级别的复杂度。所以一般没有闭式解(intractable)。
(想象某个场景,在10个混合分支中,采样100个点,计算该积分的复杂度高达10^{100}!)

ELBO(证据下界)

对目标p(\mathbf{z}|\mathbf{x})求解,一味蛮干是行不通的。通行的做法有两种,要么通过蒙特卡洛法近似,要么通过变分推断近似。这里介绍后一种。
我们利用容易处理的“代理人”函数q(\mathbf{z})近似p(\mathbf{z}|\mathbf{x})
q^*(z)=argmin_{q(z)\in \mathcal{Q}} KL[q(z)||p(\mathcal{z}|\mathcal{x})]
其中KL表示相对熵,由定义,
KL[q(z)||p(\mathcal{z}|\mathcal{x})]=\mathbb{E}[\log q(\mathcal{z})]-\mathbb{E}[\log p(\mathcal{z}|\mathcal{x})]=\mathbb{E}[\log q(\mathcal{z})]-\mathbb{E}[\log p(\mathcal{z},\mathcal{x})]+\log p(\mathcal{x})
由此可见,
\mathbb{E}[\log p(\mathcal{z},\mathcal{x})]-\mathbb{E}[\log q(\mathcal{z})]=\log p(\mathcal{x})-KL[q(z)||p(\mathcal{z}|\mathcal{x})] \leq \log p(\mathcal{x})
定义
ELBO(q) =\mathbb{E}[\log p(\mathcal{z},\mathcal{x})]-\mathbb{E}[\log q(\mathcal{z})]
最大化ELBO就是最小化KL[q(z)||p(\mathcal{z}|\mathcal{x})],成功避开计算p(\mathcal{z}|\mathcal{x})

拿掉积分!

上面的混合高斯的例子,应用变分方法,我们得到
\int p(\mu_1,...\mu_K)\prod_{i=1}^{n}\sum_{c_i}p(c_i)p(x_i|c_i,\mu_1,...\mu_K)d\mu_1d\mu_2...d\mu_K \\ \approx \mathbb{E}_{q(\mu_1,...\mu_K;\lambda)}\prod_{i=1}^{n}\sum_{c_i}p(c_i)p(x_i|c_i,\mu_1,...\mu_K)+KL[q(\mu;\lambda)||p(\mathcal{\mu}|\mathcal{x})]
虽然多引入一个参数,但是不用求积分,只要算乘法和加减就够了。生活一下子美好了有木有!

求积分为啥这么难

来自:https://xkcd.com/2117/

微分与积分

至今的算法中,求微分(导数)的一大堆,求积分的寥寥可数。计算机和人类一样不擅长算积分啊!

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

推荐阅读更多精彩内容

  • 残觞酒凉,彼时西厢,玉屏风上雕鸳鸯,坐对铜镜贴花黄,薄唇点绛,影斜云纱窗,“梧桐叶上已三更。”反手琵琶语,弦月满西...
    越儿笑倾城阅读 431评论 2 7
  • 当你睡了,衾拥着梦,窗外的 月亮,静静的迈着微步, 草虫儿也不忍作声。 我独坐在你不知晓的窗下 细数着昔日见你的光...
    南山野客阅读 182评论 0 1
  • 早上起床后,妹妹便缠着奶奶要读昨天刚从图书馆借来的绘本。 环境层面: 肯定孩子有图书馆借来的各种绘本,资源丰富。 ...
    Selena_5b5g阅读 25评论 0 0
  • 一家人说好下午带孩子去市民广场玩。 一到那里,呐喊声,欢呼声,音乐声,原来正在举办音乐会。儿子被一块大广告牌吸引了...
    暮羽初心阅读 296评论 0 0