4.4 马尔可夫链蒙特卡洛方法
到目前为止,前面讲述的对分布采样的方法称为直接采样法,在这种情况下概率分布通常是已知的,有些时候也能很容易地计算出归一化常数。然而直接采样方法在某些情况下会遇到困难甚至不适用,因此本章介绍另一种(也是非常重要的一种)基于马尔可夫链的梅特罗波利斯算法。
直接采样算法的遇到的第一个困难是,当随机变量的维度变得越来越高时,其采样效率变得越来越低,因此不适合于对高维分布进行采样。考虑随机向量: ,其中为随机向量的维度,每个分量的定义域为:,则直接采样方法中取舍采样的效率为:
(式1)
其中,,如果值很大,或者维数很高,则分母值很大,取舍算法的接受概率很低,拒绝率很高,因此采样效率非常低。直观上看,这是由于维数越高,用于包围目标概率分布的建议概率分布与目标概率分布之间存在更多的间隙空间,这些间隙空间即是被拒绝的采样点。
直接采样方法的另一个困难来自于概率分布为不完全已知概率分布,此时直接采样方法完全不适应。不完全已知概率分布是指,虽然概率分布的形式已知,但是其中包含部分项或本身就是很难进行数值计算的,例如为很难进行积分计算的高维积 分,因此相当于是未知的。
一个直接的例子是渲染中每一帧图像的分布函数,我们可以将一张图像看成是某个值的概率分布,由于图像的每个像素的值是由RGB三个颜色值组成,因此这个值可以是这三个分量某种形式的函数,或者说我们可以以一个图像的亮度作为这个分布函数的值。在这个分布中,每一个像素位置的值是光线通过无数表面反射/折射传播的结果,因此每一个值本身就是一个高维积分计算,我们将在下一章看到渲染方程的路径形式,每一条路径都是一个高维积分。
本节要讨论的梅特罗波利斯算法是一种能够从任意概率分布中进行采样的方法,它提供一种机制使我们可以通过计算一个正比于目标分布函数的分布函数的值来获取采样值。由于只需要正比于目标分布,而不需要精确计算目标分布的归一化常数,因此梅特罗波利斯算法可以用于从任意分布采样。
马尔可夫链
梅特罗波利斯算法可以看做是取舍算法的一种推广,取舍算法每一步从建议分布产生一个全局的随机数,并与目标概率分布进行比较,以此来决定对新随机数的接受与拒绝,然而取舍算法的建议概率分布和目标概率分布之间的间隙随着随机变量维数的增加而增加,因此其效率极低。
而梅特罗波利斯算法以马尔可夫链为基础,整个取样的过程产生一条马尔可夫链,它将目标概率分布当做一个稳态的系统,每个新产生的随机数根据该系统的动态平衡关系进行取舍,所以最终整条马尔可夫链上的随机数序列就是服从目标概率分布的,并且梅特罗波利斯算法只需要能够计算目标概率分布每个定义域处的值,而不需要计算其归一化常数(注意:计算的每个值和计算的归一化常数是不一样的,后者实际上是积分的计算,这个积分计算可能非常困难。)。
所以为了理解梅特罗波利斯算法,我们需要首先了解马尔可夫链(Markov chain)的概念及其属性,这是推导和理解梅特罗波利斯算法的重要基础。基于马尔可夫链的蒙特卡洛方法称为马尔可夫蒙特卡洛方法(Markov chain Monte Carlo, MCMC),而梅特罗波利斯算法是最重要的一种马尔可夫蒙特卡洛方法。
设一个系统状态序列为,每个状态可以以一定的概率转移到另一个状态,假设我们从其中任意一个起始状态开始行走,每个时刻由一个状态转移到另一个状态(也可以转移到自身),如果对任何时刻,系统状态的转移条件概率为:
(式2)
则此转移过程形成的状态序列称为马尔可夫链(注意这里表示时刻产生的一个随机数,而表示时刻的一个系统状态值,其中两个相邻时刻的系统状态值可能是相同的。)。马尔可夫链的定义说明,其将来时刻的状态,只依赖于当前时刻的状态,与以前任何时刻的状态都无关。据此,系统状态序列发生的概率可分解为:
(式3)
式中,为初始状态的概率,通常是从初始状态出发,所以;式中条件概率称为一步转移概率,简称为转移概率,表示在时刻由状态转移到状态,转移概率通常与时刻有关。若转移概率与时刻无关,即,则称马尔可夫链是均匀的,或者时齐的(time-homogeneous),通常我们研究的都是时齐的马尔可夫链,因此后面的表述我们将去掉时间相关信息,到的状 态条件转移概率直接表述为: 。
马尔可夫链将一个系统的定义域当做一个状态空间,因此马尔可夫链又分为离散时间马尔可夫链(discrete-time Markov chain, DTMC)和连续时间马尔可夫链(continuous-time Markov chain, CTMC)。
马尔可夫链是在一个状态空间中从一个状态到另一个状态的不断转换的随机过程,这个过程中经过的每个状态值就是我们获得的服从这个状态空间概率分布的采样值。在一个平衡的系统状态中,每个状态到另一个状态的转移概率是固定的(我们可以将系统状态表述为一个状态图,如图(1)是一个具有两个状态的状态图),这些转移概率的结果就是使得状态空间的每个状态服从系统的分布,因此马尔可夫链是服从系统分布的一个随机采样序列。
初学者在理解马尔可夫链的时候往往比较吃力,这里尝试做一些解释,请读者细细体会。统计方法是关于用概率来解决确定性问题的方法,它的实际操作方式就是用大量的重复试验以获取足够的样本,而这些样本是满足目标概率分布的,所以能够用来解决确定性问题。具体地,以前面讲述的蒙特卡洛方法为例,当我 们从一个概率分布采样产生一个随机数序列时,这个随机数序列里的值并不是唯一的,而是包含大量重复的值,每个随机数对应的概率越高,则其重现的次数越多,最终每个唯一随机数在这个随机序列中出现的次数正比于它的概率。
所以在马尔可夫链中也是一样,给定一个初始状态,马尔可夫链按照各个状态的转移概率在系统状态中随机行走,由于这些转移概率是按照系统状态的概率决定的,因此马尔可夫链就是一条状态链,这里马尔可夫链里出现的每个状态就是一个随机值,那些具有较高概率的系统状态,则其在马尔可夫链中出现的次数越多。因此,马尔可夫链本质上和前面讲述的其他采样方法是一样的,其目的都是产生服从目标概率分布的采样,但是操作方法不一样。
图(1):一个具有两个转换状态的马尔可夫链(图片来自Wikipedia)
在一个给定平衡状态概率分布的系统中,如图(1)就是一个包含两个状态的系统,每个状态到其他状态的转移概率是固定的(实际上对于一个有限状态空间,整个系统状态的转移可以表述为一个转移矩阵,每个状态的矢量只需要乘以这个矩阵就能按照相关的概率进行转移,这里的例子参见Examples_of_Markov_chains;在本书后面的辐射度(radiosity)方法里,其思路就是将整个场景分成多个有限的面积块,然后由几何关系计算出每个面积块到其他面积块的转移比例,即是概率,然后给定任意一个光源,经过一定时间的反射,就能得到一个稳定的光照分布。),因此给定任意一个初始状态,例如晴天为,雨天,则等到一定时间后系统处于平衡状态时,期间产生的马尔可夫链中晴天和雨天的分布一定服从平衡状态下的概率分布,即:约的时间为晴天,剩下为雨天。
所以,利用马尔可夫链对目标概率分布进行采样的过程,实际上就是设一个状态系统服从目标概率分布,然后给定一个任意初始值,经历一定时间等系统趋于平衡之后,产生的马尔可夫链就是服从目标概率分布的,因此可作为服从该目标概率分布的采样值。由此也可以看出马尔可夫蒙特卡洛方法的一个问题,那就是开始一段趋近稳定状态的过程存在很大的方差,实践中往往需要通过某种方法省略掉这部分采样值。
而在马尔可夫蒙特卡洛方法中,梅特罗波利斯算法主要就是以一种简单的方式从已知概率分布中产生每个状态下正确的转移概率,要理解梅特罗波利斯算法的推导过程,我们还需要了解马尔可夫链的一些属性。
马尔可夫链属性
马尔可夫链有许多属性,为了简化概念,这里只讨论跟后面推导梅特罗波利斯算法相关的属性。
不可约性
如果一个系统从状态开始,存在非零的概率经过步之后可以达到状态,则称可达到(accessible),记为,表述为:
(式4)
如果和同时存在(与可能不相等),则称与是联通的(communicating)。一个联通集合(communicating class)是指存在一个最大状态集合,使得其中的每对状态相互都是联通的。
如果一个状态空间就是一个单一的联通集合,则称该状态空间内的马尔可夫链是不可约的(irreducible),也就是说,在该状态空间中,可以由任何一个状态出发,经过步之后,达到另外一个任意状态。
非周期性
假设从一个状态出发,经过一定的步数后回到状态自身,如果是的倍数,则称状态具有周期(period),称为周期态,一个状态的周期表述为:
(式5)
这里表示取最大公约数(注意 :即使状态拥有最大公约数,但并不意味着从出发经过步之后一定可以回到状态,例如可能回到原状态的步数为,这里为2,但是2并没有出现在返回步数列表中。)(greatest common divisor),如果,则称该状态是非周期态(aperiodic state),它意味着回到状态自身的步数可以是任意的;如果至少有一个状态是非周期态,则称为非周期的马尔可夫链;若状态的周期数,则称其为周期态。
回返性
假设从一个状态出发,经过一定步数之后回到状态的概率记为:
(式6)
概率又称为首达概率,如果平均返回时间,则称状态为正回返(positive recurrent)状态,如果,则称其为零回返状态,或者暂态(transient)。有限马尔可夫链的回返状态必为正回返状态。
遍历性
如果状态是正回返,不可约和非周期状态,则称状态是各态遍历(ergodic)状态,或遍历态。换句话说,如果一个状态具有周期为1,并且能够在有限的平均步数内回返,则为遍历态。如果一个不可约的马尔可夫链中所有的状态都是遍历的,则该马尔可夫链是遍历的。
更一般地,如果一个马尔可夫链是遍历的,则始终存在一个步数,使得从任意其他非状态经过步之后可以达到状态。也就是说,马尔可夫链达到状态的概率与初始状态无关,其实遍历性就是初始状态独立性。遍历性是马尔可夫链收敛到平稳分布的定量条件。
平稳分布
设马尔可夫链在一个状态空间中从某个初始状态出发经历一定的时间后,各个状态的离散分布概率为,如果对任意状态存在概率密度函数,并使得:
(式7)
即从所有其他状态转移到状态的转移概率之和为状态的概率分布,记为:
(式8)
则称马尔可夫链达到总体平衡,这里是一个转移矩阵,为所有状态的概率构成的矢量。一个正回返和非周期的马尔可夫链,当且仅当它是一个遍历链时,经历足够多的状态转移步数,这个马尔可夫链具有一个不变的概率分布,这时无论初始分布如何,最终都会趋向于平稳概率分布,于是有:
(式9)
这个唯一的近似不变的概率分布就是平稳分布(stationary distribution),它是一个极限分布。这就表明,最终状态不再发生变化,系统达到平稳状态。马尔可夫链具有平稳分布,是马尔可夫链的 一个重要特征,在马尔可夫链蒙特卡洛方法中具有重要意义。
我们可以以前面的天气预报的状态分布为例来分析马尔可夫链从任意一个状态出发,经历一定时间后逐渐趋近于平稳分布的过程。如图(1)所示,这里状态空间的转移矩阵为:
(式10)
设初始状态为晴天概率为,雨天为,用一个矢量表述为:
(式11)
则经过一次随机行走之后状态概率分布为:
(式12)
即依照这样的概率分布,第二天为晴天的概率为,为雨天的概率相应为,以此类推我们可以得到第天的状态概率分布,如果足够大,使系统趋近于稳定分布,则最终的天气预报的概率为:
(式13)
细致平衡
在一个状态空间中,如果存在概率分布,使得:
(式14)
该条件称为细致平衡(detailed balance),又称为局部平衡(local balance)。细致平衡是微观平衡,比上述的平稳分布(宏观平衡)具有更多的约束条件。
由细致平衡可以得到平稳分布,反之则不成立,平稳分布并不能保证细致平衡。满足细致平衡条件的马尔可夫链是可逆的(细致平衡本就是源于在一个平衡的动力学系统中,每个基础过程与其逆过程保持平衡。),称为可逆链,因此细致平衡与可逆性是等价的,可逆性保证了平稳分布。
梅特罗波利斯算法
有了平稳分布的条件,对于使用马尔可夫链的蒙特卡洛方法的思路则比较容易理解:它首先把目标概率分布的作用域看做一个状态空间,然后从任意一个状态出发随机行走以建立马尔可夫链,这个随机行走过程中的每一 步遵循状态空间的转移概率分布,则经过一定时间之后,马尔可夫链中的状态概率分布会趋向于目标概率分布。在这个过程中,最重要的是怎样根据已知的目标概率求出转移概率分布,梅特罗波利斯算法(Metropolis algorithm)的核心就是使用了一种简单求解转移概率的方法。
为了更好地理解梅特罗波利斯算法及其最关键的概念,我们有必要区分一下统计学中利用马尔可夫链解决积分问题的两种不同的思路:第一种是已知目标概率分布,使用蒙特卡洛方法对这个目标概率分布进行采样,然后计算积分,此种方法的关键是要根据已知目标概率求出状态转移概率,这样使得马尔可夫链能够逼近目标分布,这即是本节讨论的梅特罗波利斯算法;第二种是已知转移概率分布,然后给定任意一个初始概率分布,则马尔可夫链会根据转移概率逐渐趋向于稳定分布,这种思路在计算机图形学中被用于本书后面第(8)章会讨论的辐射度方法(radiosity)中;当然后者是一种迭代法(而不是蒙特卡洛方法),但是它与马尔可夫链性质的运用其实有很深的联系,在学习辐射度理论的时候可以细细体会这种联系。
梅特罗波利斯算法通过逐渐产生一序列的随机数的方式工作,这个随机数序列是以迭代的方式产生的,在这个迭代的过程中,下一个随机数的产生仅依赖于上一个随机数(因此整个随机数序列是一个马尔可夫链),随着产生越来越多的随机数,这个随机数序列会逐渐逼近于目标概率分布。
特别地,在每一次迭代中,该算法基于当前随机数产生一个新的候选随机数,然后这个候选随机数根据一定的概率被接受(这种情况下新的候选随机数会被用于下一个迭代中)或拒绝(这种情况下新的候选随机数被丢弃,当前随机数继续被用在下一个迭代中),这个接受概率被适当选取以使其能够近似目标概率的分布。
设为一个正比于目标概率分布的近似分布,梅特罗波利斯算法的基本步骤如下:
- 初始化阶段:选取一个任意点作为起始采样点,并选择一个任意概率密度函数用来根据当前采样随机数产生下一个候选随机数,在梅特罗波利斯算法中,必须是对称的,即满足。一个常用的选择是高斯分布(Gaussian distribution),也即正态分布(normal distribution),这种情况下,靠近的点会被以更高的概率成为候选采样点。被称为建议密度函数(proposal density function)或跳跃分布(jumping distribution)。
- 迭代阶段,对于每一次迭代: 3.1. 首先从建议密度函数产生一个候选采样点。 3.2 计算接受率,这个接受率用 于决定新的候选采样点被接受或拒绝。因为正比于目标概率密度函数,所以。 3.3 如果,新的候选点直接被接受:;否则,按照概率来接受新的候选点,如果新的候选点被拒绝,则设置:。
梅特罗波利斯算法的过程是通过在采样空间随机行走实现,在随机行走的每一步,新的随机采样点可能被接受,也可能被拒绝,这完全由接受率来决定。接受率表述的是根据目标概率分布,新产生的候选随机数相对于当前随机数有多大的概率被接受,例如如果,说明新的候选采样点在中处于更高概率区域,即是它比具有更高的概率,因此新的候选点总是被接受;如果,则说明新的候选点在中处于较低概率区域,此时它可能会被接受,可能会被拒绝,越小,新的候选点拥有更小的几率被接受。因此,通过这样的机制,马尔可夫链中的随机数倾向于停留在目标概率分布的高概率区域,而只有偶尔很少的机会会访问低概率区域。直观上看,这就是梅特罗波利斯算法能够工作的原因,也使它最终返回的采样点是符合目标概率分布的。
收敛性证明
梅特罗波利斯算法的目标是根据目标概率分布产生一个状态序列,为了达到这个目的,它使用一个马尔可夫过程来渐渐地逼近一个唯一的稳定分布,使得。那么怎样证明该算法是收敛于的呢?
马尔可夫链蒙特卡洛方法的各种算法的收敛性是基于马尔可夫链的收敛性理论。一个马尔可夫过程由其转移概率唯一决定,要使该马尔可夫过程存在一个唯一的稳定分布,必须满足以下两个条件:
- 存在稳定分布。
- 稳定分布是唯一的。
要使马尔可夫链存在稳定分布,由前面的内容可知,一个充分不必要条件是满足细致平衡(detailed balance)条件,细致平衡条件要求每个转移过程是可逆的,对于每一对状态和,处于状态并转移到状态的概率必须等于处于状态并转移到状态的概率,即:。
稳定分布的唯一性可以由遍历性(ergodicity)来保证,通过前面的马尔可夫链属性可知,马尔可夫链的遍历性必须要求非周期性和正回返性。非周期性其实暗指每个状态不是只能从某些特定路径返回,而是可以沿任意路径返回,其意义就是每个状态可以与任意状态联通;而正回返性保证这种联通的概率小于。由不可约性可知,如果状态空间内所有状态都是联通的,从每个状态出发经过一定步数之后都可以达到另一个状态,则马尔可夫链就是与初始状态无关的,所以遍历性保证了稳定分布的唯一性。
梅特罗波利斯算法就是通过设计一个满足上述两个条件的马尔可夫链,以使稳定分布逼近。该算法的推导首先从满足细致平衡条件开始:
(式15)
上式可以改写为:
(式16)
梅特罗波利斯算法将转移概率的计算分成两步:建议概率和接受率,并使得:
(式17)
将上式代入前面的细致平衡关系,得到:
(式18)
接下来就是怎么选择接受概率以满足上面的条件,由前面的梅特罗波利斯算法可知其选择为:
(式19)
上式后半部分的等式成立是因为梅特罗波利斯算法中建议概率是对称的,从此也可以看出,梅特罗波利斯算法中中的建议概率是可以任意选择的,因为它并不会参与到最终转移概率的计算上。梅特罗波利斯算法中的转移概率是满足细致平衡条件的,而细致平衡条件可以保证稳定分布,也就是总体平衡,因此梅特罗波利斯算法是收敛的。
对于好奇的读者可以由以下证明过程证明其满足细致平衡条件:
(式20)
梅特罗波利斯算法的不足
尽管梅特罗波利斯算法中是收敛的,但是马尔可夫链蒙特卡洛方法的各种算法收敛性证明只能证明其样本值遵从已知概率分布,但不能证明样本值是独立无关的,也没有给出收敛速度是多少。本节介绍梅特罗波利斯算法主要的两个方面的不足。
首先,尽管梅特罗波利斯算法的建议概率可以是任意选取的,并且能够保证收敛性,但是它的选取却对效率和方差由一定的影响。直观上,为了使马尔可夫链收敛的速度更快,以及更快地遍历整个状态空间,我们趋向于选取步幅更 大的建议概率。但是对于用于高频率分布的目标分布函数,过大的建议步幅在高频部分将导致,因此使 得接受率非常低,其结果就是大量的随机数停留在尖锐的区域,导致方差增大,并且收敛速度变慢(随机数被卡在高频部分很长时间),如图(2)所示;相反,过小的建议步幅则会直接导致更慢的收敛速度,所以合适地选取建议概率函数非常重要,我们将在后面第(7)章学习MLT算法的时候更具体地讨论一些建议概率函数选择的策略。
图(2):如果分布函数的某些区域比较尖锐,或者由于建议概率产生的候选随机数的步幅过大,则会导致大部分候选随机数由于接受率太低而被拒绝,因此在尖锐的部分聚集大量相同的随机数,导致方差增大
其次,虽然梅特罗波利斯算法最终会收敛于目标概率分布,但是在这个收敛的过程中,一些初始的随机数序列则可能完全不是服从目标概率分布的,这部分随机数通常需要从马尔可夫链中去除掉,然而并没有一些好的算法用于预测这个收敛的长度需要多少初始 随机数。
尽管如此,梅特罗波利斯算法能够轻松应对高维概率分布的采样是其他方法不可比拟的,有时候几乎梅特罗波利斯算法是唯一的选择。在本章中我们仅介绍梅特罗波利斯算法的一些基本概念,在第(7)章还会有更多更具体的内容来讨论梅特罗波利斯算法的一些取舍以及更多的实现细节。