受限玻尔兹曼机(Restricted Boltzmann Machine,简称RBM)是由Hinton和Sejnowski于1986年提出的一种生成式随机神经网络(generative stochastic neural network),该网络由一些可见单元(visible unit,对应可见变量,亦即数据样本)和一些隐藏单元(hidden unit,对应隐藏变量)构成,可见变量和隐藏变量都是二元变量,亦即其状态取{0,1}。整个网络是一个二部图,只有可见单元和隐藏单元之间才会存在边,可见单元之间以及隐藏单元之间都不会有边连接,如下图所示:

受限玻尔兹曼机(Restricted Boltzmann Machine, RBM) 简介-LMLPHP

上图所示的RBM含有12个可见单元(构成一个向量v)和3个隐藏单元(构成一个向量h),W是一个12*3的矩阵,表示可见单元和隐藏单元之间的边的权重。

1. RBM的学习目标-最大化似然(Maximizing likelihood)

RBM是一种基于能量(Energy-based)的模型,其可见变量v和隐藏变量h的联合配置(joint configuration)的能量为:

受限玻尔兹曼机(Restricted Boltzmann Machine, RBM) 简介-LMLPHP(式子-1)

其中θ是RBM的参数{W, a, b}, W为可见单元和隐藏单元之间的边的权重,b和a分别为可见单元和隐藏单元的偏置(bias)。

有了v和h的联合配置的能量之后,我们就可以得到v和h的联合概率:

受限玻尔兹曼机(Restricted Boltzmann Machine, RBM) 简介-LMLPHP                  (式子-2)

其中Z(θ)是归一化因子,也称为配分函数(partition function)。根据式子-1,可以将上式写为:

受限玻尔兹曼机(Restricted Boltzmann Machine, RBM) 简介-LMLPHP  (式子-3)

我们希望最大化观测数据的似然函数P(v),P(v)可由式子-3求P(v,h)对h的边缘分布得到:

受限玻尔兹曼机(Restricted Boltzmann Machine, RBM) 简介-LMLPHP     (式子-4)

我们通过最大化P(v)来得到RBM的参数,最大化P(v)等同于最大化log(P(v))=L(θ):

受限玻尔兹曼机(Restricted Boltzmann Machine, RBM) 简介-LMLPHP                     (式子-5)

2. RBM的学习方法-CD(Contrastive Divergence,对比散列)

可以通过随机梯度下降(stichastic gradient descent)来最大化L(θ),首先需要求得L(θ)对W的导数:

受限玻尔兹曼机(Restricted Boltzmann Machine, RBM) 简介-LMLPHP     (式子-6)

经过简化可以得到:

受限玻尔兹曼机(Restricted Boltzmann Machine, RBM) 简介-LMLPHP   (式子-7)

后者等于

受限玻尔兹曼机(Restricted Boltzmann Machine, RBM) 简介-LMLPHP                         (式子-8)

式子-7中的前者比较好计算,只需要求vh在全部数据集上的平均值即可,而后者涉及到v,h的全部2种组合,计算量非常大(基本不可解)。

为了解决式子-8的计算问题,Hinton等人提出了一种高效的学习算法-CD(Contrastive Divergence),其基本思想如下图所示:

受限玻尔兹曼机(Restricted Boltzmann Machine, RBM) 简介-LMLPHP

首先根据数据v来得到h的状态,然后通过h来重构(Reconstruct)可见向量v,然后再根据v1来生成新的隐藏向量h。因为RBM的特殊结构(层内无连接,层间有连接), 所以在给定v时,各个隐藏单元h的激活状态之间是相互独立的,反之,在给定h时,各个可见单元的激活状态v也是相互独立的,亦即:

受限玻尔兹曼机(Restricted Boltzmann Machine, RBM) 简介-LMLPHP  (式子-9)

重构的可见向量v和隐藏向量h就是对P(v,h)的一次抽样,多次抽样得到的样本集合可以看做是对P(v,h)的一种近似,使得式子-7的计算变得可行。

RBM的权重的学习算法:

  1. 取一个样本数据,把可见变量的状态设置为这个样本数据。随机初始化W。
  2. 根据式子-9的第一个公式来更新隐藏变量的状态,亦即h以P(h=1|v)的概率设置为状态1,否则为0。然后对于每个边vh,计算P(vh)=v*h(注意,v和h的状态都是取{0,1})。
  3. 根据h的状态和式子-9的第二个公式来重构v,并且根据v1和式子-9的第一个公式来求得h,计算P(vh)=v*h。
  4. 更新边vh的权重W为W=W+L*(P(vh)=P(vh))。
  5. 取下一个数据样本,重复1-4的步骤。
  6. 以上过程迭代K次。

参考文献:

1. R. Salakhutdinov. Deep Learning Tutorial.

2. 张春霞, 姬楠楠, 王冠伟. 受限玻尔兹曼机简介.

3. Wikipedia: Restricted Boltzmann Machine

4. Edwin Chen: Introduction to Retricted Boltzmann Machine

05-11 13:49