数据采样

蓄水池采样法(Reservoir Sampling)

蓄水池算法适用于对一个不清楚规模的数据集进行采样。尤其适合针对数据流进行采样,并不知道这个流什么时候结束,且须保证每单位数据被采样到的几率相同。

假设数据序列规模为 nn ,需要采样的数量为 kk 。内存中开辟可容纳 kk 个元素的空间,并将序列前 kk 个元素放入空间中。然后从第 k+1k+1 个元素开始,以 kn\frac{k}{n} 的概率来决定该元素是否被替换到数组中(数组中的元素被替换的概率是相同的)。当遍历完所有元素后,内存中所存储的即为采样的样本。

证明:

对于第 ii 个数( iki\leq k ),被选中的概率为 11 。当走到第 k+1k+1 步时, 被第 k+1k+1 个元素替换的概率为 k+1k+1 个元素被选中的概率乘以 ii 被选择替换的概率,即 kk+1×1k=1k+1\frac{k}{k+1}\times \frac{1}{k}=\frac{1}{k+1} 。则被保留的概率为 11k+1=kk+11-\frac{1}{k+1}=\frac{k}{k+1} 。依次类推,不被 k+2k+2 个元素替换的概率为 1kk+2×1k=k+1k+21-\frac{k}{k+2}\times\frac{1}{k}=\frac{k+1}{k+2} 。则运行到第 nn 步时,被保留的概率为被选中的概率乘以不被替换的概率,即 1×kk+1×k+1k+2××n1n=kn1\times\frac{k}{k+1}\times\frac{k+1}{k+2}\times\cdots\times\frac{n-1}{n}=\frac{k}{n}

对于第 jj 个数( iki\leq k )。在第 jj 步被选中的概率为 kj\frac{k}{j} 。不被第 j+1j+1 个元素替换的概率为 1kj+1×1k=jj+11-\frac{k}{j+1}\times\frac{1}{k}=\frac{j}{j+1} 。则运行到第 nn 步时,被保留的概率为被选中的概率乘以不被替换的概率,即 kj×jj+1×j+1j+2××n1n=kn\frac{k}{j}\times\frac{j}{j+1}\times\frac{j+1}{j+2}\times\cdots\times\frac{n-1}{n}=\frac{k}{n}

所以,对于每一个元素,被保留的概率都为 kn\frac{k}{n}

逆采样(Inverse Sampling)

在蒙特卡罗方法中,有一个关键的问题需要解决,即如何基于概率密度函数去采得 nnxx 的样本。

逆采样(Inverse Sampling)和拒绝采样(Reject Sampling)就是用于解决这个问题的。

我们知道,对于常见的均匀分布 uniform(0,1)\text{uniform}(0,1) 是非常容易采样的,一般通过线性同余发生器就可以很方便的生成 (0,1)(0,1) 之间的伪随机数样本。而其他常见的概率分布,无论是离散的分布还是连续的分布,它们的样本都可以通过 uniform(0,1)\text{uniform}(0,1) 的样本转换而得。那么应该如何得到呢?这就是逆采样。

下面我们以指数分布为例,说明如何通过均匀分布来采样服从指数分布的样本集。指数分布的概率密度函数PDF为:

pexp(x)={λexp(λx),    x00                  ,    x<0p_{\exp}(x)=\begin{cases} \lambda\exp(-\lambda x),\ \ \ \ x\geq0\\ 0\ \ \ \ \ \ \ \ \ \ \ \ \ \ \ \ \ \ ,\ \ \ \ x<0 \end{cases}

那么它的概率分布函数为:

F(x)=p(x)dxF(x) = \int_{-\infty}^{\infty}p(x)dx

下图为指数分布和均匀分布的CDF图。从左图上看,在 x0x\geq 0 的部分是一个单调递增的函数(在定义域上单调非减),定义于和值域是 [0,)[0,1)[0,\infty)\to[0,1) ,在 p(x)p(x) 大的地方它增长快,反之亦然。

因为它是唯一映射的(在 >0>0 的部分,接下来我们只考虑这一部分),所以它的反函数可以表示为 F1(a),a[0,1)F^{-1}(a), a \in [0,1) ,值域为 [0,)[0,\infty) 。因为 FF 单调递增,所以 F1F^{-1} 也是单调递增的:

xyF(x)F(y)x\leq y \Rightarrow F(x)\leq F(y)

abF1(a)F1(b)a\leq b \Rightarrow F^{-1}(a)\leq F^{-1}(b)

利用反函数的定义,我们有:

F1(a)<x, iff a<F(x)F^{-1}(a)<x,\ \text{iff}\ a<F(x)

接下来,我们定义一下 [0,1][0,1] 均匀分布的CDF,这个很好理解:

P(ax)=H(x)={1  ,x1x  ,0x10  ,x<0P(a\leq x)=H(x)=\begin{cases}1\ \ ,x\geq 1\\ x\ \ ,0\leq x\leq1\\ 0\ \ ,x<0\end{cases}

根据上两式,有:

P(F1(a)x)=P(aF(x))=H(F(x))P(F^{-1}(a)\leq x)=P(a\leq F(x))=H(F(x))

因为 F(x)F(x) 的值域 a=ba = b ,根据均匀分布的CDF,上式可改写为:

P(F1(a)x)=H(F(x))=F(x)P(F^{-1}(a)\leq x)=H(F(x))=F(x)

F(x)F(x) 的定义,它是 exp\exp 分布的CDF,所以上式的意思是 F1(a)F^{-1}(a) 符合 exp\exp 分布,我们通过 FF 的反函数将一个 0011 均匀分布的随机数转换成了符合 exp\exp 分布的随机数,注意,以上推导对于CDF可逆的分布都是一样的。对于 exp\exp 来说,它的反函数的形式是:

Fexp1(a)=1λlog(1a)F^{-1}_{\exp}(a)=-\frac{1}{\lambda}*\log(1-a)

具体的映射关系可以看下图(a),我们从y轴0-1的均匀分布样本(绿色)映射得到了服从指数分布的样本(红色)

最后绘制出来的直方图可以看出来就是 exp\exp 分布图,见上图(b)。可以看到随着采样数量的变多,概率直方图和真实的CDF就越接近。以上就是逆采样的过程。我们的结论是:因为CDF是单调函数(累积的概率只能越来越大,直到为1),因此,只要某分布的CDF可逆,那么就可以通过均匀分布来采样服从该分布的样本集

拒绝采样(Reject Sampling)

对于常见的分布,如均匀分布,高斯分布,指数分布,t分布,F分布,Beta分布,Gamma分布等,可以采用逆采样的方法进行采样;不过很多时候,我们的概率分布不是常见的分布,这些分布的概率分布函数CDF 不可逆,因此没有办法用逆采样来采样,这意味着我们没法方便的得到这些非常见的概率分布的样本集。拒绝采样就是用来解决这个问题的 一种随机采样方法。

我们以求圆周率 π\pi 的例子入手,讲解拒绝采样的思想。通过采样的方法来计算π\pi值,也就是在一个1×11\times 1 的范围内随机采样一个点,如果它到原点的距离小于 11 ,则说明它在 1/41/4 圆内,则接受它,最后通过接受的占比来计算 1/41/4 圆形的面积,从而根据公式反算出预估 π^\hat{\pi} 的值,随着采样点的增多,最后的结果 π^\hat{\pi} 会愈加准确。

上面这个例子里说明一个问题,我们想求一个空间里均匀分布的集合面积,可以尝试在更大范围内按照均匀分布随机采样,如果采样点在集合中,则接受,否则拒绝。最后的接受概率就是集合在”更大范围“的面积占比。接下来,我们来形式化地说明拒绝采样。

给定一个概率分布 p(z)=1Zpp~(z)p(z)=\frac{1}{Z_p}\tilde{p}(z) ,其中, p~(z)\tilde{p}(z) 已知, ZpZ_p 为归一化常数,未知。要对该分布 p(z)p(z) 进行拒绝采样,首先需要借用一个简单的参考分布(proposal distribution),记为 q(x)q(x) ,该分布的采样易于实现,如均匀分布、高斯分布。然后引入常数 kk ,使得对所有的 zz ,满足 kq(z)p~(z)kq(z)\geq\tilde{p}(z) ,如下图所示,红色的曲线为 p~(z)\tilde{p}(z) ,蓝色的曲线为 kq(z)kq(z)

在每次采样中,首先从 q(z)q(z) 采样一个数值 z0z_0 ,然后在区间 [0,kq(z0)][0,kq(z_0)] 进行均匀采样,得到 u0u_0 。如果 u0<p~(z0)u_0<\tilde{p}(z_0) ,则保留该采样值,否则舍弃该采样值。最后得到的数据就是对 p(z)p(z) 分布的一个近似采样。结合图,直观来理解上述的过程:在 x=z0x = z_0 这条线上,从 [0,kq(z0)][0,kq(z_0)] 均匀采样中一个值,如果这个值小于 p~(z0)\tilde{p}(z_0) ,即这个均匀采样的这个值落在了 p~(z0)\tilde{p}(z_0) 下方,我们就接受 z0z_0 这个采样值。

我们知道,每次采样的接受概率计算如下:

p(accept)=p~(z)kq(z)q(z)dz=1kp~(z)dzp(\text{accept})=\int\frac{\tilde{p}(z)}{kq(z)}q(z)dz=\frac{1}{k}\int\tilde{p}(z)dz

所以,为了提高接受概率,防止舍弃过多的采样值而导致采样效率低下, kk 的选取应该在满足 kq(z)p~(z)kq(z)\geq \tilde{p}(z) 的基础上尽可能小。

拒绝采样问题可以这样理解, p~(z)\tilde{p}(z)xx 轴之间的区域为要估计的问题,类似于求 π\pi 提到的圆形区域, kq(z)kq(z)xx 轴之间的区域为参考区域,类似于上面提到的正方形。由于 kq(z)kq(z)xx 轴之间的区域面积为 kk (原来的概率密度函数的面积为 11 ,现在扩大了 kk 倍)所以, p~(z)\tilde{p}(z)xx 轴之间的区域面积除以 kk 即为对 p(z)p(z) 的估计。在每一个采样点,以 [0,kq(z0)][0,kq(z_0)] 为界限,落在 p~(z)\tilde{p}(z) 曲线以下的点就是服从 p(z)p(z) 分布的点。

我们必须知道复杂分布的概率密度函数PDF,才可以进行拒绝采样。然而,在现实情况中:

  • 对于一些二维分布 p(x,y)p(x,y) ,有时候我们只能得到条件分布 p(xy)p(x∣y)p(yx)p(y∣x) ,却很难得到二维分布的概率密度函数 p(x,y)p(x,y) 的一般形式,这时我们无法用拒绝采样得到其样本集。

  • 对于一些高维的复杂非常见分布 p(x1,x2,...,xn)p(x_1,x_2,...,x_n) ,我们要找到一个合适的 q(x)q(x)kk 非常困难。

因此,实际上,我们仍然要找到一种方法可以解决如何方便得到各种复杂概率分布的对应的采样样本集的问题。马尔科夫链有能力帮助找到这些复杂概率分布的对应的采样样本集,而这也是MCMC采样的基础。

采样和蒙特卡罗方法

有许多原因使我们希望从某个分布中采样。当我们需要以较小的代价近似许多项或某个积分时,采样是一种很灵活的选择。有时候,我们使用它加速一些很费时却易于处理的求和估计,就像我们使用小批量对整个训练代价进行子采样一样。在其他情况下,我们需要近似一个难以处理的求和或积分,例如估计一个无向模型中配分函数对数的梯度时。在许多其他情况下,抽样实际上是我们的目标,例如我们想训练一个可以从训练分布采样的模型。

我们知道,随机算法在采样不全时,通常不能保证找到最优解,只能说是尽量找。那么根据怎么个“尽量”法儿,我们我们把随机算法分成两类:

  • 蒙特卡罗算法:采样越多,越近似最优解;

  • 拉斯维加斯算法:采样越多,越有机会找到最优解;

举个例子,假如筐里有100个苹果,让我每次闭眼拿1个,挑出最大的。于是我随机拿1个,再随机拿1个跟它比,留下大的,再随机拿1个……我每拿一次,留下的苹果都至少不比上次的小。拿的次数越多,挑出的苹果就越大,但我除非拿100次,否则无法肯定挑出了最大的。这个挑苹果的算法,就属于蒙特卡罗算法——尽量找好的,但不保证是最好的

而拉斯维加斯算法,则是另一种情况。假如有一把锁,给我100把钥匙,只有1把是对的。于是我每次随机拿1把钥匙去试,打不开就再换1把。我试的次数越多,打开(最优解)的机会就越大,但在打开之前,那些错的钥匙都是没有用的。这个试钥匙的算法,就是拉斯维加斯的——尽量找最好的,但不保证能找到

这两类随机算法之间的选择,往往受到问题的局限。如果问题要求在有限采样内,必须给出一个解,但不要求是最优解,那就要用蒙特卡罗算法。反之,如果问题要求必须给出最优解,但对采样没有限制,那就要用拉斯维加斯算法。

比如机器下棋的算法本质都是搜索树,围棋难在它的树宽可以达到好几百。在有限时间内要遍历这么宽的树,就只能牺牲深度(俗称“往后看几步”),但围棋又是依赖远见的游戏,甚至不仅是看“几步”的问题。所以,要想保证搜索深度,就只能放弃遍历,改为随机采样——这就是为什么在没有MCTS(蒙特卡罗搜树)类的方法之前,机器围棋的水平几乎是笑话。而采用了MCTS方法后,搜索深度就大大增加了。

蒙特卡罗采样的基础

蒙特卡罗原来是一个赌场的名称,用它作为名字大概是因为蒙特卡罗方法是一种随机模拟的方法,这很像赌博场里面的扔骰子的过程。最早的蒙特卡罗方法都是为了求解一些不太好求解的求和或者积分问题。比如积分:

θ=abf(x)dx\theta=\int_a^bf(x)dx

如果我们很难求解出 f(x)f(x) 的原函数,那么这个积分比较难求解。当然我们可以通过蒙特卡罗方法来模拟求解近似值。如何模拟呢?假设我们函数图像如下图:

一个简单的近似求解方法是在 [a,b][a,b] 之间随机的采样一个点。比如 x0x_0 ,然后用 f(x0)f(x_0) 代表在 [a,b][a,b] 区间上所有的 f(x)f(x) 的值。那么上面的定积分的近似求解为:

(ba)f(x0)(b-a)f(x_0)

当然,用一个值代表 [a,b][a,b] 区间上所有的 f(x)f(x) 的值太粗糙了。我们可以采样 [a,b][a,b] 区间的 nn 个值: x0,x1,,xn1x_0,x_1,\dots,x_{n-1} ,用它们的均值来代表 [a,b][a,b] 区间上所有的 f(x)f(x) 的值。这样我们上面的定积分的近似求解为:

bani=0n1f(xi)\frac{b-a}{n}\sum\limits_{i=0}^{n-1}f(x_i)

虽然上面的方法可以一定程度上求解出近似的解,但是它隐含了一个假定,即 xx[a,b][a,b] 之间是均匀分布的,而绝大部分情况, xx[a,b][a,b] 之间不是均匀分布的。如果我们用上面的方法,则模拟求出的结果很可能和真实值相差甚远。

如果我们可以得到 xx[a,b][a,b] 的概率分布函数 p(x)p(x) ,那么我们的定积分求和可以这样进行:

θ=abf(x)dx=abf(x)p(x)p(x)dx1ni=0n1f(xi)p(xi)\theta=\int_a^bf(x)dx=\int_a^b\frac{f(x)}{p(x)}p(x)dx\approx\frac{1}{n}\sum\limits_{i=0}^{n-1}\frac{f(x_i)}{p(x_i)}

上式最右边的这个形式就是蒙特卡罗方法的一般形式。当然这里是连续函数形式的蒙特卡罗方法,但是在离散时一样成立。关于上式,我们可以这么理解最后一步转换:由于 abf(x)p(x)p(x)dx\int_a^b\frac{f(x)}{p(x)}p(x)dx 可以看做是 f(x)p(x)\frac{f(x)}{p(x)} 基于概率分布 p(x)p(x) 的期望,那么我们可以用期望的方法来求这个式子的值。而计算期望的一个近似方法是取 f(x)p(x)\frac{f(x)}{p(x)} 的若干个基于分布 p(x)p(x) 的采样点,然后求平均值得到。

举例如下

假设 f(x)f(x) 的取值只有 22 个, x=1,2x = 1,2 。对应的 yy 值分别是 f(1)=1,f(2)=4f(1)=1,f(2)=4 。其中 xx 的取值不是平均的,取 x=1x = 1 的概率 p(1)=0.25p(1)=0.25 ,取 x=2x = 2 的概率 p(2)=0.75p(2)=0.75

那么严格来说,根据 θ=abf(x)dx=abf(x)p(x)p(x)dx1ni=0n1f(xi)p(xi)\theta=\int_a^bf(x)dx=\int_a^b\frac{f(x)}{p(x)}p(x)dx\approx\frac{1}{n}\sum\limits_{i=0}^{n-1}\frac{f(x_i)}{p(x_i)} ,对应的 f(x)f(x) 的积分等于 f(x)p(x)\frac{f(x)}{p(x)} 基于概率分布 p(x)p(x) 的期望,根据公式,则有 10.250.25+40.750.75=5\frac{1}{0.25}*0.25+\frac{4}{0.75}*0.75=5

此时我们采样三次,期望求近似结果。假设第一次采样到 11 ,第二三次采样到 22 ,那么最后的近似结果是

13(10.25+40.75+40.75)=4.89\frac{1}{3}(\frac{1}{0.25}+\frac{4}{0.75}+\frac{4}{0.75})=4.89 。这个 4.894.89 就是我们 55 的近似。虽然有些距离,但是由于采样太少原因。若我们采样100次,得到26次 11 ,74次 22 ,那么近似结果是 1100(2610.25+7440.75)=4.99\frac{1}{100}(26*\frac{1}{0.25}+74*\frac{4}{0.75})=4.99 。可见结果越来越接近。接近的原因是随着采样数的增多,采样的样本分布越来越接近于 xx 本来的分布。

基于马尔科夫链采样

马尔可夫链转移矩阵性质

马尔科夫链定义本身比较简单,它假设某一时刻状态转移的概率只依赖于它的前一个状态。举个形象的比喻,假如每天的天气是一个状态的话,那个今天是不是晴天只依赖于昨天的天气,而和前天的天气没有任何关系。当然这么说可能有些武断,但是这样做可以大大简化模型的复杂度,因此马尔科夫链在很多时间序列模型中得到广泛的应用,比如循环神经网络RNN,隐式马尔科夫模型HMM等,当然MCMC也需要它。

如果用精确的数学定义来描述,则假设我们的序列状态是 ,Xt2,Xt1,Xt,Xt+1,\dots,X_{t-2},X_{t-1},X_{t},X_{t+1},\dots 那么我们在时刻 Xt+1X_{t+1} 的状态的条件概率仅仅依赖于时刻 XtX_t ,即:

P(Xt+1Xt2,Xt1,Xt)=P(Xt+1Xt)P(X_{t+1}|\dots X_{t-2},X_{t-1},X_{t})=P(X_{t+1}|X_t)

既然某一时刻状态转移的概率只依赖于它的前一个状态,那么我们只要能求出系统中任意两个状态之间的转换概率,这个马尔科夫链的模型就定了。

这个马尔科夫链是表示股市模型的,共有三种状态:牛市(Bull market), 熊市(Bear market)和横盘(Stagnant market)。每一个状态都以一定的概率转化到下一个状态。比如,牛市以0.025的概率转化到横盘的状态。这个状态概率转化图可以以矩阵的形式表示。如果我们定义矩阵 PP 某一位置 P(i,j)P(i,j) 的值为 P(j,i)P(j,i) ,即从状态 ii 转化到状态 jj 的概率,并定义牛市为状态0, 熊市为状态1, 横盘为状态2。这样我们得到了马尔科夫链模型的状态转移矩阵为:

P=[0.9  0.075  00250.15  0.8  0.050.25  0.25  0.5]P = \left[ \begin{matrix} 0.9 \ \ 0.075\ \ 0025\\ 0.15 \ \ 0.8\ \ 0.05\\ 0.25\ \ 0.25\ \ 0.5 \end{matrix} \right]

讲了这么多,那么马尔科夫链模型的状态转移矩阵和我们蒙特卡罗方法需要的概率分布样本集有什么关系呢?这需要从马尔科夫链模型的状态转移矩阵的收敛性质讲起。马尔科夫链模型的状态转移矩阵收敛到的稳定概率分布与我们的初始状态概率分布无关,且这个稳定分布是唯一的。这是一个非常好的性质,也就是说,如果我们得到了这个稳定概率分布对应的马尔科夫链模型的状态转移矩阵,则我们可以用任意的概率分布样本开始,带入马尔科夫链模型的状态转移矩阵,这样经过一些序列的转换,最终就可以得到符合对应稳定概率分布的样本。

基于马尔可夫链的采样

如果我们得到了某个平稳分布所对应的马尔科夫链状态转移矩阵 PP ,我们就很容易采用出这个平稳分布的样本集。

假设我们任意初始的概率分布是 π0(x)\pi_0(x) ,经过第一轮马尔可夫链状态转移后的概率分布是 π1(x)\pi_1(x) ,第 ii 轮的概率分布是 πi(x)\pi_i(x) 。假设经过 nn 轮后马尔可夫链收敛到我们的平稳分布 π(x)\pi(x) ,即:

πn(x)=πn+1(x)=πn+2(x)==π(x)\pi_n(x)=\pi_{n+1}(x)=\pi_{n+2}(x)=\dots=\pi(x)

对于每个分布 πi(x)\pi_i(x),我们有:

πi(x)=πi1(x)P=πi2(x)P2=π0(x)Pi\pi_i(x)=\pi_{i-1}(x)P=\pi_{i-2}(x)P^2=\pi_{0}(x)P^i

现在我们可以开始采样了,首先,基于初始任意简单概率分布比如高斯分布 π0(x)\pi_0(x) 采样得到状态值 x0x_0 ,基于条件概率分布 P(xx0)P(x|x_0) 采样状态值 x1x_1 , 一直进行下去,当状态转移进行到一定的次数时,比如到 nn 次时,我们认为此时的采样集 (xn,xn+1,xn+2,)(x_n,x_{n+1},x_{n+2},\dots) 即是符合我们的平稳分布的对应样本集,可以用来做蒙特卡罗模拟求和了。

采样过程

(1)输入马尔可夫链状态转移矩阵 PP ,设定状态转移次数阈值 n1n_1 ,需要的样本个数 n2n_2

(2)从任意简单概率分布采样得到初始状态值 x0x_0

(3) for t=0 to (n1+n21)\text{for}\ t=0\ \text{to}\ (n_1+n_2-1) :从条件概率分布 P(xxt)P(x|x_t) 中采样得到样本 xt+1x_{t+1}

(4)样本集 (xn1,xn1+1,,xn1+n21)(x_{n_1},x_{n_{1}+1},\dots,x_{n_{1}+n_2-1}) 即为我们需要的平稳分布对应的样本集

对采样过程的两点补充说明:1、第(3)步,每对 P(xxt)P(x|x_t) 采样一次就意味着模拟了一次 πi(x)=πi1(x)P\pi_i(x)=\pi_{i-1}(x)P 操作。在此过程中,状态转化矩阵 PP 没有变化。2、第(4)步, 注意的是采样的样本集的下标是 n1,n1+1,,n1+n21n_1,n_1+1,\dots,n_1+n_2-1 ,即状态稳定之后采样的样本才是我们需要的样本集。 即我们需要从真正趋于稳定时候的分布抽样样本。

如果假定我们可以得到我们需要采样样本的平稳分布所对应的马尔科夫链状态转移矩阵 PP ,那么我们就可以用马尔科夫链采样得到我们需要的样本集,进而进行蒙特卡罗模拟。但是一个重要的问题是,随意给定一个平稳分布 π\pi ,如何得到它所对应的马尔科夫链状态转移矩阵 PP 呢?这是个大问题。我们绕了一圈似乎还是没有解决任意概率分布采样样本集的问题。幸运的是,MCMC采样通过迂回的方式解决了上面这个大问题,以及它的使用改进版采样: M-H采样和Gibbs采样。

MCMC采样

在解决从平稳分布 π\pi ,找到对应的马尔可夫链状态转移矩阵 PP 之前,我们还需要先看看马尔科夫链的细致平稳条件,定义如下:

如果非周期马尔科夫链的状态转移矩阵 PP 和概率分布 π(x)\pi(x) 对于所有的 i,ji,j 满足:

π(i)P(i,j)=π(j)P(j,i)\pi(i)P(i,j)=\pi(j)P(j,i)

则称概率分布 π(x)\pi(x) 是状态转移矩阵 PP 的平稳分布。

证明很简单,由细致平稳条件有:

i=1π(i)P(i,j)=i=1π(j)P(j,i)=π(j)i=1P(j,i)=π(j)\sum\limits_{i=1}^\infty\pi(i)P(i,j)=\sum\limits_{i=1}^\infty\pi(j)P(j,i)=\pi(j)\sum\limits_{i=1}^\infty P(j,i)=\pi(j)

将上式用矩阵表示即为:

πP=π\pi P=\pi

即满足马尔可夫链的收敛性质。也就是说,只要我们找到了可以使概率分布 π(x)\pi(x) 满足细致平稳分布的矩阵 PP 即可。这给了我们寻找从平稳分布 π\pi ,找到对应的马尔可夫链状态转移矩阵 PP 的新思路。

不过不幸的是,仅仅从细致平稳条件还是很难找到合适的矩阵 PP 。比如我们的目标平稳分布是 π(x)\pi(x) ,随机找一个马尔可夫链状态转移矩阵 QQ ,它是很难满足细致平稳条件的,即:

π(i)Q(i,j)π(j)Q(j,i)\pi(i)Q(i,j)\neq\pi(j)Q(j,i)

MCMC采样是这样做的。 由于一般情况下,目标平稳分布 π(x)\pi(x) 和某一个马尔可夫链状态转移矩阵 QQ 不满足细致平稳条件,即上式。我们可以对上式做一个改造,使细致平稳条件成立。方法是引入一个 α(i,j)\alpha(i,j) ,使上式可以去等号,即:

π(i)Q(i,j)α(i,j)=π(j)Q(j,i)α(j,i)\pi(i)Q(i,j)\alpha(i,j)=\pi(j)Q(j,i)\alpha(j,i)

问题是什么样的 α(i,j)\alpha(i,j) 可以使等式成立呢?其实很简单,只要满足下两式即可:

α(i,j)=π(j)Q(j,i)\alpha(i,j)=\pi(j)Q(j,i)

α(j,i)=π(i)Q(i,j)\alpha(j,i)=\pi(i)Q(i,j)

这样,我们就得到了我们的分布 π(x)\pi(x) 对应的马尔可夫链状态转移矩阵 PP ,满足:

P(i,j)=Q(i,j)α(i,j)P(i,j)=Q(i,j)\alpha(i,j)

也就是说,我们的目标矩阵 PP 可以通过任意一个马尔可夫链状态转移矩阵 QQ 乘以 α(i,j)\alpha(i,j) 得到。 α(i,j)\alpha(i,j) 我们又一般称之为接受率,取值在 [0,1][0,1] 之间。物理意义可以理解为:在原来的马氏链上,从状态 iiQ(i,j)Q(i,j) 的概率转移到状态 jj 的时候,我们以 α(i,j)\alpha(i,j) 的概率接受这个转移。即目标矩阵 PP 可以通过任意一个马尔科夫链状态转移矩阵 QQ 以一定的接受率获得。这个很像拒绝采样的思想,那里是以一个常用分布通过一定的接受-拒绝概率得到一个非常见分布,这里是以一个常见的马尔科夫链状态转移矩阵 QQ 通过一定的接受-拒绝概率得到目标转移矩阵 PP ,两者的解决问题思路是类似的。

采样过程

(1)输入我们任意选定的马尔可夫链状态转移矩阵 QQ ,平稳分布 π(x)\pi(x) ,设定状态转移次数阈值 n1n_1 ,需要的样本个数 n2n_2

(2)从任意简单概率分布采样得到初始状态值 x0x_0

(3) for t=0 to (n1+n21)\text{for}\ t=0\ \text{to}\ (n_1+n_2-1)

(a)从条件概率分布 Q(xxt)Q(x|x_t) 中采样得到样本 xx_*

(b)从均匀分布采样 uuniform[0,1]u\sim \text{uniform}[0,1]

(c)如果 u<α(xt,x)=π(x)Q(x,xt)u<\alpha(x_t,x_*)=\pi(x_*)Q(x_*,x_t) ,则接受转移 xtxx_t\to x_* ,即 xt+1=xx_{t+1} = x_*

(d)否则不接受转移,即 xt+1=xtx_{t+1}=x_t

样本集 (xn1,xn1+1,,xn1+n21)(x_{n_1},x_{n_1+1},\dots,x_{n_1+n_2-1}) 即为我们需要的平稳分布对应的样本集。对于(3)的(d)步, xt+1=xtx_{t+1} =x_t 意味着,尽管不接受转移,但是也把它放入了样本序列。而在有些算法中,被拒绝的样本就不进入采样样本序列。尽管双方有些mismatch,不过对理解算法影响不大。

上面这个过程基本上就是MCMC采样的完整采样理论了,但是这个采样算法还是比较难在实际中应用,为什么呢?问题在上面第三步的(c)步骤,接受率这儿。由于 α(xt,x)\alpha(x_t,x_*) 可能非常的小,比如 0.10.1 ,导致我们大部分的采样值都被拒绝转移,采样效率很低。有可能我们采样了上百万次马尔可夫链还没有收敛,也就是上面这个 n1n_1 要非常非常的大,这让人难以接受,怎么办呢?这时就轮到我们的M-H采样出场了。

M-H采样

M-H采样是Metropolis-Hastings采样的简称,这个算法首先由Metropolis提出,被Hastings改进,因此被称之为Metropolis-Hastings采样或M-H采样。M-H采样解决了MCMC采样接受率过低的问题。

我们回到MCMC采样的细致平稳条件:

π(i)Q(i,j)α(i,j)=π(j)Q(j,i)α(j,i)\pi(i)Q(i,j)\alpha(i,j)=\pi(j)Q(j,i)\alpha(j,i)

我们采样效率低的原因是 α(i,j)\alpha(i,j) 太小了,比如为 0.10.1 ,而 α(j,i)\alpha(j,i)0.20.2 。即:

π(i)Q(i,j)×0.1=π(j)Q(j,i)×0.2\pi(i)Q(i,j)\times 0.1=\pi(j)Q(j,i)\times 0.2

我们可以看到,如果两边同时扩大五倍,接受率提高到了 0.50.5 ,但是细致平稳条件却仍然是满足的,即:

π(i)Q(i,j)×0.5=π(j)Q(j,i)×1\pi(i)Q(i,j)\times 0.5=\pi(j)Q(j,i)\times 1

这样我们的接受率可以做如下改进,即:

α(i,j)=min{π(j)Q(j,i)π(i)Q(i,j),1}\alpha(i,j)=\min\{\frac{\pi(j)Q(j,i)}{\pi(i)Q(i,j)},1\}

对上式多解释一下:此时 α(i,j)\alpha(i,j) 已经不是原来的意义,变为了 α(i,j)/α(j,i)\alpha(i,j)/\alpha(j,i) 的倍率关系,这里默认右边为 11 ,看左边的较大的那个接受率的值。即当原始的 α(i,j)<α(j,i)\alpha(i,j)<\alpha(j,i) 时, α(i,j)\alpha(i,j) 扩大了 1/α(j,i)1/\alpha(j,i) 倍; α(i,j)>α(j,i)\alpha(i,j)>\alpha(j,i) 时, α(i,j)\alpha(i,j) 直接变为 11

采样过程

(1)输入我们任意选定的马尔可夫链状态转移矩阵 QQ ,平稳分布 π(x)\pi(x) ,设定状态转移次数阈值 n1n_1 ,需要的样本个数 n2n_2

(2)从任意简单概率分布采样得到初始状态值 x0x_0

(3) for t=0 to (n1+n21)\text{for}\ t=0\ \text{to}\ (n_1+n_2-1)

(a)从条件概率分布 Q(xxt)Q(x|x_t) 中采样得到样本 xx_*

(b)从均匀分布采样 uuniform[0,1]u\sim \text{uniform}[0,1]

(c)如果 u<α(xt,x)=min{π(j)Q(j,i)π(i)Q(i,j),1}u<\alpha(x_t,x_*)=\min\{\frac{\pi(j)Q(j,i)}{\pi(i)Q(i,j)},1\} ,则接受转移 xtxx_t\to x_* ,即 xt+1=xx_{t+1} = x_*

(d)否则不接受转移,即 xt+1=xtx_{t+1}=x_t

样本集 (xn1,xn1+1,,xn1+n21)(x_{n_1},x_{n_1+1},\dots,x_{n_1+n_2-1}) 即为我们需要的平稳分布对应的样本集。

吉布斯采样(Gibbs sampling)

M-H采样已经可以很好的解决蒙特卡罗方法需要的任意概率分布的样本集的问题。但是M-H采样有两个缺点:一是需要计算接受率,在高维时计算量大。并且由于接受率的原因导致算法收敛时间变长。二是有些高维数据,特征的条件概率分布好求,但是特征的联合分布不好求。因此需要一个好的方法来改进M-H采样,这就是Gibbs采样。

从上面内容我们得到细致平稳条件:如果非周期马尔科夫链的状态转移矩阵 PP 和概率分布 π(x)\pi(x) 对所有的 i,ji,j 满足:

π(i)P(i,j)=π(j)P(j,i)\pi(i)P(i,j)=\pi(j)P(j,i)

则称概率分布 π(x)\pi(x) 是状态转移矩阵 PP 的平稳分布。

在M-H采样中我们通过引入接受率使细致平稳条件满足。现在我们换一个思路。

从二维的数据分布开始,假设 π(x1,x2)\pi(x_1,x_2) 是一个二维联合数据分布,观察第一个特征维度相同的两个点 A(x1(1),x2(1))A(x_1^{(1)},x_2^{(1)})B(x1(1),x2(1))B(x_1^{(1)},x_2^{(1)}) ,容易发现下面两式成立:

π(x1(1),x2(1))π(x2(2)x1(1))=π(x1(1))π(x2(1)x1(1))π(x2(2)x1(1))\pi(x_1^{(1)},x_2^{(1)})\pi(x_2^{(2)}|x_1^{(1)})=\pi(x_1^{(1)})\pi(x_2^{(1)}|x_1^{(1)})\pi(x_2^{(2)}|x_1^{(1)})

π(x1(1),x2(2))π(x2(1)x1(1))=π(x1(1))π(x2(2)x1(1))π(x2(1)x1(1))\pi(x_1^{(1)},x_2^{(2)})\pi(x_2^{(1)}|x_1^{(1)})=\pi(x_1^{(1)})\pi(x_2^{(2)}|x_1^{(1)})\pi(x_2^{(1)}|x_1^{(1)})

成立的原因是: π(x1(1),x2(1))=π(x1(1))π(x2(1)x1(1))\pi(x_1^{(1)},x_2^{(1)})=\pi(x_1^{(1)})\pi(x_2^{(1)}|x_1^{(1)})π(x1(1),x2(2))=π(x1(1))π(x2(2)x1(1))\pi(x_1^{(1)},x_2^{(2)})=\pi(x_1^{(1)})\pi(x_2^{(2)}|x_1^{(1)})

由于上面两式右边相等,因此我们有:

π(x1(1),x2(1))π(x2(2)x1(1))=π(x1(1),x2(2))π(x2(1)x1(1))\pi(x_1^{(1)},x_2^{(1)})\pi(x_2^{(2)}|x_1^{(1)}) = \pi(x_1^{(1)},x_2^{(2)})\pi(x_2^{(1)}|x_1^{(1)})

也就是

π(A)π(x2(2)x1(1))=π(B)π(x2(1)x1(1))\pi(A)\pi(x_2^{(2)}|x_1^{(1)}) =\pi(B)\pi(x_2^{(1)}|x_1^{(1)})

观察上式再观察细致平稳条件的公式,我们发现在 x1=x1(1)x_1 = x_1^{(1)} 这条直线上,如果用条件概率分布 π(x2x1(1))\pi(x_2|x_1^{(1)}) 作为马尔科夫链的状态转移概率,则任意两个点之间的转移满足细致平稳条件!这真是一个开心的发现,同样的道理,在 x2=x2(1)x_2 = x_2^{(1)} 这条直线上,如果用条件概率分布 π(x1x2(1))\pi(x_1|x_2^{(1)})作为马尔科夫链的状态转移概率,则任意两个点之间的转移也满足细致平稳条件。那是因为假如有一点 C(x1(2),x2(1))C(x_1^{(2)},x_2^{(1)}) ,我们可以得到:

π(A)π(x1(2)x2(1))=π(C)π(x1(1)x2(1))\pi(A)\pi(x_1^{(2)}|x_2^{(1)})=\pi(C)\pi(x_1^{(1)}|x_2^{(1)})

基于上面的发现,我们可以这样构造分布 π(x1,x2)\pi(x_1,x_2) 的马尔可夫链对应的状态转移矩阵 PP

P(AB)=π(x2(B)x1(1)),  if x1(A)=x1(B)=x1(1)P(A\to B)=\pi(x_2^{(B)}|x_1^{(1)}), \ \ \text{if}\ x_1^{(A)}=x_1^{(B)}=x_1^{(1)}

P(AC)=π(x2(C)x1(1)),  if x2(A)=x2(C)=x2(1)P(A\to C)=\pi(x_2^{(C)}|x_1^{(1)}), \ \ \text{if}\ x_2^{(A)}=x_2^{(C)}=x_2^{(1)}

P(AD)=0, elseP(A\to D) = 0,\ \text{else}

有了上面这样的状态转移矩阵 PP ,我们很容易验证平面上的任意两点 E,FE,F ,满足细致平稳条件:

π(E)P(EF)=π(F)P(FE)\pi(E)P(E\to F)=\pi(F)P(F\to E)

二维Gibbs采样

利用上一节找到的状态转移矩阵,我们就得到了二维Gibbs采样,这个采样必须要知道两个维度之间的条件概率。具体过程如下:

(1)输入平稳分布 π(x1,x2)\pi(x_1,x_2) ,设定状态转移次数阈值 n1n_1 ,需要的样本个数 n2n_2

(2)随机初始化初始状态值 x1(0)x_1^{(0)}x2(0)x_2^{(0)}

(3) for t=0 to (n1+n21)\text{for}\ t=0\ \text{to}\ (n_1+n_2-1)

(a)从条件概率分布 P(x2x1(t))P(x_2|x_1^{(t)}) 中采样得到样本 x2(t+1)x_2^{(t+1)}

(b)从条件概率分布 P(x1x2(t+1))P(x_1|x_2^{(t+1)}) 中采样得到样本 x1(t+1)x_1^{(t+1)}

样本 {(x1(n1),x2(n1)),(x1(n1+1),x2(n1+1)),,(x1(n1+n21),x2(n1+n21))}\{(x_1^{(n_1)},x_2^{(n_1)}),(x_1^{(n_1+1)},x_2^{(n_1+1)}),\dots,(x_1^{(n_1+n_2-1)},x_2^{(n_1+n_2-1)})\} 即为我们需要的平稳分布对应的样本集。整个采样过程中,我们通过轮换坐标轴,采样的过程为:

(x1(1),x2(1))(x1(1),x2(2))(x1(2),x2(2))(x1(n1+n21),x2(n1+n21))(x_1^{(1)},x_2^{(1)})\to (x_1^{(1)},x_2^{(2)})\to (x_1^{(2)},x_2^{(2)})\to \dots \to (x_1^{(n_1+n_2-1)},x_2^{(n_1+n_2-1)})

用下图可以很直观的看出,采样是在两个坐标轴上不停的轮换的。当然,坐标轴轮换不是必须的,我们也可以每次随机选择一个坐标轴进行采样。不过常用的Gibbs采样的实现都是基于坐标轴轮换的。

多维Gibbs采样

上面的这个算法推广到多维的时候也是成立的。比如一个 nn 维的概率分布 π(x1,x2,,xn)\pi(x_1,x_2,\dots,x_n) ,我们可以通过 nn 个坐标轴上轮换采样,来得到新的样本。对于轮换到的任意一个坐标轴 xix_i 上的转移,马尔科夫链的状态转移概率为 P(xix1,x2,,xi1,xi+1,,xn)P(x_i|x_1,x_2,\dots,x_{i-1},x_{i+1},\dots,x_n) ,即固定 n1n-1 个坐标轴,在某一个坐标轴上移动。

采样过程

(1)输入平稳分布 π(x1,x2,,xn)\pi(x_1,x_2,\dots,x_n) 或者对应的所有特征的条件概率分布,设定状态转移次数阈值 n1n_1 ,需要的样本个数 n2n_2

(2)随机初始化初始状态值 (x1(0),x2(0),,xn(0))(x_1^{(0)},x_2^{(0)},\dots,x_n^{(0)})

(3) for t=0 to (n1+n21)\text{for}\ t=0\ \text{to}\ (n_1+n_2-1)

(a)从条件概率分布 P(x1x2(t),x3(t),,xn(t))P(x_1|x_2^{(t)},x_3^{(t)},\dots,x_n^{(t)}) 中采样得到样本 x1(t+1)x_1^{(t+1)}

(b)从条件概率分布 P(x2x1(t+1),x3(t),x4(t),,xn(t))P(x_2|x_1^{(t+1)},x_3^{(t)},x_4^{(t)},\dots,x_n^{(t)}) 中采样得到样本 x2(t+1)x_2^{(t+1)}

(c)。。。

()从条件概率分布 P(xjx1(t+1),x2(t+1),,xj1t+1,xj+1t,,xn(t))P(x_j|x_1^{(t+1)},x_2^{(t+1)},\dots,x_{j-1}^{t+1},x_{j+1}^{t},\dots,x_n^{(t)}) 中采样得到样本 xj(t+1)x_j^{(t+1)}

(f)。。。

(g)从条件概率分布 P(xnx1(t+1),x2(t+1),,xn1(t+1))P(x_n|x_1^{(t+1)},x_2^{(t+1)},\dots,x_{n-1}^{(t+1)}) 中采样得到样本 xn(t+1)x_n^{(t+1)}

样本 {(x1(n1),x2(n1),,xn(n1)),(x1(n1+1),x2(n1+1)),,(x1(n1+n21),x2(n1+n21),,xn(n1+n21))}\{(x_1^{(n_1)},x_2^{(n_1)},\dots,x_n^{(n_1)}),(x_1^{(n_1+1)},x_2^{(n_1+1)}),\dots,(x_1^{(n_1+n_2-1)},x_2^{(n_1+n_2-1)},\dots,x_n^{(n_1+n_2-1)})\} 即为我们需要的平稳分布对应的样本集。整个采样过程和Lasso回归的坐标轴下降法算法非常类似,只不过Lasso回归是固定 n1n-1 个特征,对某一个特征求极值。而Gibbs采样是固定 n1n-1 个特征在某一个特征采样。

同样的,轮换坐标轴不是必须的,我们可以随机选择某一个坐标轴进行状态转移,只不过常用的Gibbs采样的实现都是基于坐标轴轮换的。

Source

Last updated