本文内容来自 Deep Learning: Foundations and Concepts 一书的第十六章——连续潜在变量。
正文的数学公式会尽可能详细易懂,但是某些公式的详细推导太长,因此其过程在 Appendix 中给出。
应当注意的是本文只是博主在学习过程中对于原书内容的摘要性记录,并不能完全代替原书内容。
CH16 Continuous Latent Variables
什么是潜在变量(Latent Variables)?
这个概念主要是相对于高维数据而言的,对于很多高维数据,我们可以认为它们实际上是由一些低维的潜在变量控制。举个例子,对于手写数字图片(见下图),一般而言我们会认为每一个像素点都是一个维度,假如整个图片有 100x100 个像素,那么每个图片都是一个高达 10000 维的数据。但是换个角度思考,我们可以认为这些图片实际上是由第一张图片经过一些变化,例如平移和旋转得到的。让我们暂时忽略数字部分在像素级别的细微差异,我们是否可以认为,以下五张图片实际上是由三个变量(纵向平移,横向平移,中心旋转)所控制的呢?完全可以,此时我们可以认为这几张图片是由这三个潜在变量控制生成的,因为你只要拿到第一张图片,然后通过修改这三个变量,就可以得到后续所有的图片。而在观测数据中我们只能看到像素,而看不到控制生成过程的三个变量,所以这些变量就称之为潜在变量。
上面的例子也许不是很严谨,例如控制生成过程的也许还有好多个其他的潜在变量,但是我们主要关注这种思想。实际上现实中绝大多数数据都符合这种特点,即高维的数据往往集中在某些低维流形上,且其维度远小于原始数据维度 。此时我们就可以通过这些低维流形所代表在潜在变量来生成一些数据,但是生成的数据往往和现实不太一致,因为现实数据往往带有噪音。因此我们在生成的数据上添加一些噪音,从而使其更加接近观测数据。
这个过程很自然的启发了一些生成式模型。
而要从观测数据中找到控制其生成过程的潜在变量,我们就需要想办法抛弃无用的维度,将高维的观测数据降维到真正的低维流形上。
16.1 Principal Component Analysis
主成分分析(PCA)是一种非常常用的降维手段,也被称之为 Kosambi–Karhunen–Loeve transform。PCA 有两种定义,一是最大化投影后数据方差的线性投影;二是最小化投影代价的线性投影,投影代价通常定义为投影前后数据的均方距离。我们会从两个角度分别尝试推导出 PCA 的数学过程。
所有推导过程以单变量为基础,多变量降维请参考 canonical correlation analysis。
我们有 N N N 个观测数据 { x n } = { x 1 , … , x N } \{\mathbf{x}_n\}=\{\mathbf{x}_1,\dots,\mathbf{x}_N\} { x n } = { x 1 , … , x N } ,每个数据由一个 D 维向量表示,即 x i ∈ R D \mathbf{x}_i\in \mathbb{R}^D x i ∈ R D 。我们希望将其降维到一个 M M M 维空间中(M < D M<D M < D ),即每个数据点使用 M M M 维向量表示。通常而言 M M M 是给定的,后面我们会讨论一些选择 M M M 的技巧。
现在考虑一个简单情况,即 M = 1 M=1 M = 1 。我们希望将数据投影到一条线上,我们可以使用一个向量 u t \mathbf{u}_t u t 来表示,而且我们关注的是向量的方向,而不在乎其长度,因为这个向量代表的是我们所要投影到的一维空间方向,至于这个向量到底多长,我们并不在乎,所以我们规定 u t T u t = 1 \mathbf{u}_t^T\mathbf{u}_t=1 u t T u t = 1 。投影后的向量就可以表示为 u t x n \mathbf{u}_t\mathbf{x}_n u t x n ,此时我们不难求出投影后数据的均值为 u t T x ˉ \mathbf{u}_t^T\bar{\mathbf{x}} u t T x ˉ ,方差为 u t T S u t \mathbf{u}_t^T \mathbf{S}\mathbf{u}_t u t T S u t 。
均值和方差的推导见 Appendix A1.1~A1.2。
我们要最大化投影后数据的方差,即 max u t T S u t \max \mathbf{u}_t^T \mathbf{S} \mathbf{u}_t max u t T S u t ,但是可能出现 u t T S u t = a \mathbf{u}_t^T \mathbf{S} \mathbf{u}_t =a u t T S u t = a ,有 u ^ t = k u t \hat{\mathbf{u}}_t=k\mathbf{u_t} u ^ t = k u t , 令 u ^ t T S u ^ t = k 2 u t T S u t = k 2 a \hat{\mathbf{u}}_t^T \mathbf{S} \hat{\mathbf{u}}_t=k^2\mathbf{u}_t^T \mathbf{S} \mathbf{u}_t=k^2 a u ^ t T S u ^ t = k 2 u t T S u t = k 2 a 。但我们不想关注向量的长度,因此我们要引入约束项 u t T u t = 1 \mathbf{u}_t^T\mathbf{u}_t=1 u t T u t = 1 。
此时使用拉格朗日乘子法,构建带约束的最大化方程:
L ( x , λ 1 ) = u t T S u t + λ 1 ( 1 − u t T u t ) (1.1) \begin{align*}\mathcal{L}(\mathbf{x},\lambda_1)=\mathbf{u}_t^T\mathbf{S}\mathbf{u}_t + \lambda_1(1-\mathbf{u}_t^T\mathbf{u}_t) \end{align*}\tag{1.1} L ( x , λ 1 ) = u t T S u t + λ 1 ( 1 − u t T u t ) ( 1.1 )
该方程对 u t \mathbf{u}_t u t 求偏导得:
S u t = λ 1 u t (1.2) \mathbf{S}\mathbf{u}_t=\lambda_1\mathbf{u}_t \tag{1.2} S u t = λ 1 u t ( 1.2 )
求导过程见 Appendix A1.3。
显然投影向量 u t \mathbf{u}_t u t 是协方差矩阵 S \mathbf{S} S 的一个特征向量,此时两边左乘 x t T \mathbf{x}_t^T x t T 得到投影后的方差 u t T S u t = λ 1 \mathbf{u}_t^T \mathbf{S}\mathbf{u}_t=\lambda_1 u t T S u t = λ 1 。当 M > 1 M>1 M > 1 时,令投影后方差最大化的投影向量就是协方差矩阵的前 M M M 个最大特征值所对应的特征向量。由此我们就推导出了 PCA 降维的数学过程,即对协方差矩阵做特征值分解,从最大的特征值开始选择对应的特征向量,就是降维所需要的投影向量。
但是这种方法的计算量太大,对于维度为 D D D 的数据,其计算成本为 O ( D 3 ) O(D^3) O ( D 3 ) 。
有 D D D 维基向量 { u n } = { u 1 , … , u N } \{\mathbf{u}_n\}=\{\mathbf{u}_1,\dots,\mathbf{u}_N\} { u n } = { u 1 , … , u N } ,满足 u i T u f = δ _ i j \mathbf{u}_i^T\mathbf{u}_f = \delta\_{ij} u i T u f = δ _ ij ,意思是 i = j i=j i = j 时 u i T u f = 1 \mathbf{u}_i^T\mathbf{u}_f=1 u i T u f = 1 ,否则 u i T u f = 0 \mathbf{u}_i^T\mathbf{u}_f=0 u i T u f = 0 ,这意味着 { u n } \{\mathbf{u}_n\} { u n } 是一组完全正交的基向量。
对于 D D D 维的观测数据 { x n } = { x 1 , … , x N } \{\mathbf{x}_n\}=\{\mathbf{x}_1,\dots,\mathbf{x}_N\} { x n } = { x 1 , … , x N } ,我们可以使用新的基向量做如下转换:
x n = ∑ i = 1 D α n i u i (1.3) \begin{align*}
\mathbf{x}_n=\sum_{i=1}^{D}\alpha_{ni}\mathbf{u}_i
\end{align*}\tag{1.3} x n = i = 1 ∑ D α ni u i ( 1.3 )
这相当于在新的基向量张成的空间中表示观测数据,原始数据 n n \mathbf{n}_n n n 的分量 x n = { x 1 , … , x D } \mathbf{x}_n=\{x_1,\dots,x_D\} x n = { x 1 , … , x D } ,被替换为 { α 1 , … , α D } \{\alpha_1,\dots,\alpha_D\} { α 1 , … , α D } 。此时我们在上式基础上同时左乘某个基向量 u j T \mathbf{u}_j^T u j T ,同时应用基向量之间的正交性,可以得到:
u j T x n = ∑ i = 1 D α n i u j T u i = α n j α n j = u j T x n = x n T u j (1.4) \begin{align*}
\mathbf{u}_j^T\mathbf{x}_n
&= \sum_{i=1}^D \alpha_{ni} \mathbf{u}_j^T \mathbf{u}_i
= \alpha_{nj} \\\\
\alpha_{nj}
&= \mathbf{u}_j^T \mathbf{x}_n
= \mathbf{x}_n^T \mathbf{u}_j
\end{align*}
\tag{1.4} u j T x n α nj = i = 1 ∑ D α ni u j T u i = α nj = u j T x n = x n T u j ( 1.4 )
我们将公式(1.4)的结果带入公式(1.3),可以改写为:
x n = ∑ i = 1 D ( x n T u i ) u i (1.5) \begin{align*}
\mathbf{x}_n = \sum_{i=1}^D \left( \mathbf{x}_n^T \mathbf{u}_i \right) \mathbf{u}_i
\end{align*}\tag{1.5} x n = i = 1 ∑ D ( x n T u i ) u i ( 1.5 )
此时我们成功使用了自定义的完全正交基向量来表示观测数据,但是我们的目标是对数据进行降维,这意味着我们不能使用全部的 D D D 个基向量,如果要降到 M M M 维,那么我们就只能使用其中的 M M M 个基向量来重新表示,但是选择哪些基向量比较好呢?
假设我们选择了前 M M M 个基向量,我们可以将观测的数据近似的表示为:
x ~ n = ∑ i = 1 M z n i u i + ∑ i = M + 1 D b i u i (1.6) \begin{align*}
\tilde{\mathbf{x}}_n = \sum_{i=1}^{M} z_{ni} \mathbf{u}_i + \sum_{i=M+1}^{D} b_i \mathbf{u}_i
\end{align*}\tag{1.6} x ~ n = i = 1 ∑ M z ni u i + i = M + 1 ∑ D b i u i ( 1.6 )
其中 { z n i } \{z_{ni}\} { z ni } 依赖于特定的观测数据点,而 { b i } \{b_i\} { b i } 则是对于所有数据点都一致的常量。这种表示下观测数据只有 M M M 维基向量描述,而公式中的第二项用来补偿降维所失去维度,从而使我们可以在 D D D 维中表示降维后的数据。
现在我们只需要确定 z n i , u i , b i z_{ni},\mathbf{u}_i,b_i z ni , u i , b i 就可以确定降维过程了。该怎么确定呢?答案就在这一小节的标题中,我们要最小化投影前后的数据的投影代价,使用投影前数据和投影后近似数据的距离来描述投影代价。因此我们需要建立如下代价函数:
J = 1 N ∑ n = 1 N ∣ ∣ x n − x ~ n ∣ ∣ 2 (1.7) \begin{align*}
J=\frac{1}{N}\sum_{n=1}^N\mid\mid x_n-\tilde{x}_n\mid\mid^2
\end{align*}\tag{1.7} J = N 1 n = 1 ∑ N ∣∣ x n − x ~ n ∣ ∣ 2 ( 1.7 )
结合公式(1.6)和(1.7),我们将代价函数分别对每个变量求偏导,令偏导为零从而求出使代价函数最小的变量值。首先考虑对 z n i z_{ni} z ni 求偏导,我们可以得到:
z n j = u j T x n , where j = 1 , … , M (1.8) \begin{align*}
z_{nj}=\mathbf{u}_j^T\mathbf{x}_n,\quad\text{where } j&=1,\dots,M
\end{align*}\tag{1.8} z nj = u j T x n , where j = 1 , … , M ( 1.8 )
其中 j j j 是为了避免在推导过程中和 i i i 混用引起歧义,其含义和 i i i 类似,表示某个特定基向量,推导过程见 Appendix A1.4~A1.9。
同理我们对 b j b_j b j 求偏导可以得到:
b j = x ˉ T u j , where j = M + 1 , … , D (1.9) \begin{align*}
b_j=\bar{\mathbf{x}}^T\mathbf{u}_j,\quad \text{where }j=M+1,\dots,D
\end{align*}\tag{1.9} b j = x ˉ T u j , where j = M + 1 , … , D ( 1.9 )
推导过程见 Appendix A1.10~A1.12。
我们将公式 1.8 和 1.9 代回 1.6,得到
x ~ n = ∑ i = 1 M ( u i T x n ) x n + ∑ i = M + 1 D ( x ˉ n T u i ) u i (1.10) \begin{align*}
\tilde{\mathbf{x}}_n=\sum_{i=1}^M(\mathbf{u}_i^T\mathbf{x}_n)\mathbf{x}_n+\sum_{i=M+1}^D(\bar{\mathbf{x}}_n^T\mathbf{u}_i)\mathbf{u}_i
\end{align*}\tag{1.10} x ~ n = i = 1 ∑ M ( u i T x n ) x n + i = M + 1 ∑ D ( x ˉ n T u i ) u i ( 1.10 )
此时我们得到了降维后近似变量的表达式,然后我们只要做相减就可以求出原始变量和降维后近似变量的距离。注意我们用公式 1.5 来表示原始变量,这样整个计算都在我们新定义的基向量空间中进行。
x n − x ~ n = ∑ i = M + 1 D { ( x n − x ˉ ) T u i } u i (1.11) \begin{align*}
\mathbf{x}_n-\tilde{\mathbf{x}}_n=\sum_{i=M+1}^D\{(\mathbf{x}_n-\bar{\mathbf{x}})^T\mathbf{u}_i\}\mathbf{u}_i
\end{align*}\tag{1.11} x n − x ~ n = i = M + 1 ∑ D {( x n − x ˉ ) T u i } u i ( 1.11 )
详细推导过程见 Appendix A1.13。
这个公式说明了降维前后数据向量的差距都落在了我们选择的 M M M 维空间中之外,根据该公式我们可以定义 N N N 个数据点的总投影代价。
J = 1 N ∑ n = 1 N ∑ i = M + 1 D ( x n T u i − x ˉ T u i ) 2 = ∑ i = M + 1 D u i T S u i (1.12) \begin{align*}
J=\frac{1}{N}\sum_{n=1}^N\sum_{i=M+1}^D(\mathbf{x}_n^T\mathbf{u}_i-\bar{\mathbf{x}}^T\mathbf{u}_i)^2=\sum_{i=M+1}^D\mathbf{u}_i^T\mathbf{S}\mathbf{u}_i
\end{align*}\tag{1.12} J = N 1 n = 1 ∑ N i = M + 1 ∑ D ( x n T u i − x ˉ T u i ) 2 = i = M + 1 ∑ D u i T S u i ( 1.12 )
推导过程见 Appendix 1.14。
此时我们最小化这个代价函数,同时带有约束 u i T u i = 1 \mathbf{u}_i^T\mathbf{u}_i=1 u i T u i = 1 。我们以 D = 2 , M = 1 D=2,M=1 D = 2 , M = 1 为例使用拉普拉斯乘子法。
J = u 2 T S u 2 J ~ = u 2 T S u 2 + λ 2 ( 1 − u 2 T u 2 ) ∂ J ∂ u 2 = ∂ u 2 T S u 2 ∂ u 2 + ∂ λ 2 ∂ u 2 − ∂ λ 2 u 2 T u 2 ∂ u 2 = 2 S u 2 − 2 λ 2 u 2 = 0 S u 2 = λ 2 u 2 \begin{align*}
J&=\mathbf{u}_2^T\mathbf{S}\mathbf{u}_2\\\\
\tilde{J}&=\mathbf{u}_2^T\mathbf{S}\mathbf{u}_2+\lambda_2(1-\mathbf{u}_2^T\mathbf{u}_2)\\\\
\frac{\partial J}{\partial \mathbf{u}_2}&=\frac{\partial \mathbf{u}_2^T\mathbf{S}\mathbf{u}_2}{\partial \mathbf{u}_2}+\frac{\partial\lambda_2}{\partial \mathbf{u}_2}-\frac{\partial \lambda_2\mathbf{u}_2^T\mathbf{u}_2}{\partial \mathbf{u}_2}\\\\
&=2\mathbf{S}\mathbf{u}_2-2\lambda_2\mathbf{u}_2=0\\\\
\mathbf{S}\mathbf{u}_2&=\lambda_2\mathbf{u}_2
\end{align*} J J ~ ∂ u 2 ∂ J S u 2 = u 2 T S u 2 = u 2 T S u 2 + λ 2 ( 1 − u 2 T u 2 ) = ∂ u 2 ∂ u 2 T S u 2 + ∂ u 2 ∂ λ 2 − ∂ u 2 ∂ λ 2 u 2 T u 2 = 2 S u 2 − 2 λ 2 u 2 = 0 = λ 2 u 2
由此可见,降维后的丢弃基向量就是协方差矩阵的特征向量的一部分,我们将其推广到 M M M 维可以得到 J = ∑ i = M + 1 D λ i J=\sum_{i=M+1}^D\lambda_i J = ∑ i = M + 1 D λ i ,因此我们只需要抛弃最小的特征值所代表的基向量,就可以令投影代价最小化。
16.1.3 Data compression
这一章节简要描述了 PCA 在数据有损压缩中的一些应用,并对如何看待降维后的近似变量提供了一种新的看法,即降维后的数据实际上是原始数据的均值加上我们给定的 M M M 个自由度来描述压缩后的数据,因此压缩后数据的信息量完全取决于这 M M M 个维度的信息,所以 M M M 越大,保留的信息就越多,就越像原始数据。
16.1.4 Data whitening
数据白化其实不属于降维,而是数据预处理的一种,其效果类似于标准化或者归一化,即把数据的某些数量级差距过大的属性转为数据量近似的表示方法。
16.1.5 High-dimensional Data
这一节概述了如何对高维数据做 PCA,因为做 PCA 的复杂度和数据维度 D D D 的三次方成正比,因此直接对高维数据进行 PCA 的成本是难接受的,这一章节描述了一种将 PCA 转入原始数据的某一个低维空间中进行处理的方法。
我平常不使用 PCA,因此这一节就大概看了一下。
16.2 Probabilistic Latent Variables
上一节中我们介绍了 PCA,一种通过线性映射将高维数据变化到低维空间中的方法。更详细一点说是使用一组低维基向量的线性组合来近似表示高维数据,而线性组合的系数 z n i z_{ni} z ni 被看作是一组确定的潜在变量。一组潜在变量可以描述对应的高维的观测数据,也可以理解为所有的高维观测数据都是从不同的潜在变量组合中生成的,由此引入了生成式模型的一些概念。但是现在这些潜在变量都是确定的值,因为他们是基向量的系数。但是如果潜在变量是一个联系概率分布呢?
如果我们使用最大似然来表示一个使用概率潜在变量的 PCA,那就叫 Probabilistic PCA。它有以下几个有点:
是一种受限高斯分布,限制了自由参数的数量,但是模型仍然可以捕捉数据中的主要相关性。
可以通过生成的方式运行,从分布中提供样本。
为 PCA 的贝叶斯处理奠定了基础,使得主子空间的维度可以自动从数据中确定。
实际上书中给出了八条优点,但我觉得这三条最重要,可惜这三条我也没有完全理解。
16.2.1 Generative model
我们首先定义 M M M 维的潜在变量 z \mathbf{z} z ,然后定义高斯先验 p ( z ) p(\mathbf{z}) p ( z ) 和对应的以 D D D 维观测数据 x \mathbf{x} x 作为条件的高斯条件分布 p ( z ∣ x ) p(\mathbf{z}|\mathbf{x}) p ( z ∣ x ) 。
具体而言,我们将高斯先验定义为标准高斯分布。
p ( z ) = N ( z ∣ 0 , I ) (2.1) \begin{align*}
p(\mathbf{z})=\mathcal{N}(\mathbf{z}\mid\mathbf{0},\mathbf{I})
\end{align*}\tag{2.1} p ( z ) = N ( z ∣ 0 , I ) ( 2.1 )
同样的,观测数据是潜在变量的线性变化,因此有
p ( x ∣ z ) = N ( x ∣ W z + μ , σ 2 I ) (2.2) \begin{align*}
p(\mathbf{x}\mid\mathbf{z})=\mathcal{N}(\mathbf{x}\mid\mathbf{Wz}+\boldsymbol{\mu},\sigma^2\mathbf{I})
\end{align*}\tag{2.2} p ( x ∣ z ) = N ( x ∣ Wz + μ , σ 2 I ) ( 2.2 )
其中 W \mathbf{W} W 是一个 D × M D\times M D × M 的矩阵,表示线性变换,实际上其列向量所张成的空间就代表了主子空间,而 μ \boldsymbol{\mu} μ 则对均值进行了偏移。σ \sigma σ 对条件分布的方差进行了缩放。我们可以生成式模型的角度来看待 probabilistic PCA,即首先从标准高斯分布中采样潜在变量 z \mathbf{z} z ,然后在该潜在变量控制的条件分布上采样生成观测数据 x \mathbf{x} x 。具体而言,观测数据是潜在变量的线性变化然后加上了一点高斯噪声,数学描述为:
x = W z + μ + ϵ (2.3) \begin{align*}
\mathbf{x}=\mathbf{Wz}+\boldsymbol{\mu}+\boldsymbol{\epsilon}
\end{align*}\tag{2.3} x = Wz + μ + ϵ ( 2.3 )
其中 z \mathbf{z} z 是 M M M 维潜在变量,ϵ \boldsymbol{\epsilon} ϵ 是 D D D 维均值为零,方差为 σ 2 I \sigma^2\mathbf{I} σ 2 I 的高斯噪声。
16.2.2 Likelihood function
根据公式 2.3 的描述,如果我们想要生成数据,那么我们就需要确定 W , μ , σ \mathbf{W},\boldsymbol{\mu},\sigma W , μ , σ 这几个量。通过最大似然法,我们首先写出 x \mathbf{x} x 的边缘概率分布。
p ( x ) = ∫ p ( x ∣ z ) p ( z ) d z (2.4) \begin{align*}
p(\mathbf{x})=\int p(\mathbf{x}\mid\mathbf{z})p(\mathbf{z})d\mathbf{z}
\end{align*}\tag{2.4} p ( x ) = ∫ p ( x ∣ z ) p ( z ) d z ( 2.4 )
高斯分布的线性变换依然是高斯分布,因此我们可以定义
p ( x ) = N ( x ∣ μ , C ) C = W W T + σ 2 I \begin{align*}
p(\mathbf{x})&=\mathcal{N}(\mathbf{x}\mid\boldsymbol{\mu},\mathbf{C})\tag{2.5}\\\\
\mathbf{C}&=\mathbf{WW^T}+\sigma^2\mathbf{I}\tag{2.6}
\end{align*} p ( x ) C = N ( x ∣ μ , C ) = W W T + σ 2 I ( 2.5 ) ( 2.6 )
这个分布的均值和协方差矩阵推导如下
E [ x ] = E [ W z + μ + ϵ ] = μ cov [ x ] = cov [ W z + ϵ ] = E [ ( W z + ϵ ) T ( W z + ϵ ) ] = E [ W z z W T ] + E ( ϵ ϵ T ) = W W T + σ 2 I \begin{align*}
\mathbb{E}[\mathbf{x}]&=\mathbb{E}[\mathbf{Wz}+\boldsymbol{\mu}+\boldsymbol{\epsilon}]=\boldsymbol{\mu}\tag{2.7}\\\\
\text{cov}[\mathbf{x}]&=\text{cov}[\mathbf{Wz}+\boldsymbol{\epsilon}]\\\\
&=\mathbb{E}[(\mathbf{Wz}+\boldsymbol{\epsilon})^\mathbf{T}(\mathbf{Wz}+\boldsymbol{\epsilon})]\\\\
&=\mathbb{E}[\mathbf{WzzW}^\text{T}]+\mathbb{E}(\boldsymbol{\epsilon\epsilon}^\text{T})\\\\
&=\mathbf{WW}^\text{T}+\sigma^2\mathbf{I}\tag{2.8}
\end{align*} E [ x ] cov [ x ] = E [ Wz + μ + ϵ ] = μ = cov [ Wz + ϵ ] = E [( Wz + ϵ ) T ( Wz + ϵ )] = E [ WzzW T ] + E ( ϵϵ T ) = WW T + σ 2 I ( 2.7 ) ( 2.8 )
推导过程中使用了 z \mathbf{z} z 和 ϵ \boldsymbol{\epsilon} ϵ 相互独立的特性。
你也许发现一开始定义的边缘概率积分似乎不需要,我也这么觉得。
现在我们使用了三个变量 W , μ , σ \mathbf{W},\boldsymbol{\mu},\sigma W , μ , σ 来表示了观测数据分布,但其实这些变量仍然有一些冗余。为了说明这个问题,我们一个矩阵 W ~ = W R \tilde{\mathbf{W}}=\mathbf{WR} W ~ = WR ,其中 R \mathbf{R} R 是一个正交阵,使用正交阵的性质,我们可以得到如下结果
W ~ W ~ T = W R R T W T = W W T (2.9) \begin{align*}
\tilde{\mathbf{W}}\tilde{\mathbf{W}}^\text{T}=\mathbf{WRR^T W^T}=\mathbf{WW^T}
\end{align*}\tag{2.9} W ~ W ~ T = WR R T W T = W W T ( 2.9 )
而上面这个公式的结果实际上出现在了观测数据 x \mathbf{x} x 的协方差矩阵中。这里我们其实对线性变换矩阵 W \mathbf{W} W 施加了一个新的线性变化,而由于变化矩阵 R \mathbf{R} R 是正交阵,因此可以更具体说我们做了旋转。但是旋转后变换矩阵 W ~ \tilde{\mathbf{W}} W ~ 对数据观测数据的方差分布没有任何影响。这意味着即使潜变量空间发生了旋转,模型的协方差结构是保持不变的,因此我们的变量中对于旋转的描述是冗余的,无论怎么旋转,最终结果都是一样的。
此外,如果我们要写出 p ( x ) p(\mathbf{x}) p ( x ) 的表达式,我们需要求 C − 1 \mathbf{C}^{-1} C − 1 ,这是因为对于多维高斯分布而言,其表达式形式如下:
p ( x ) = 1 ( 2 π ) D / 2 ∣ C ∣ exp ( − 1 2 ( x − μ ) T C − 1 ( x − μ ) ) (2.10) \begin{align*}
p(\mathbf{x})=\frac{1}{(2\pi)^{D/2}\mid \mathbf{C}\mid}\exp\left(-\frac{1}{2}(\mathbf{x}-\boldsymbol{\mu})^\text{T}\mathbf{C}^{-1}(\mathbf{x}-\boldsymbol{\mu})\right)
\end{align*}\tag{2.10} p ( x ) = ( 2 π ) D /2 ∣ C ∣ 1 exp ( − 2 1 ( x − μ ) T C − 1 ( x − μ ) ) ( 2.10 )
这涉及到对于一个 D × D D\times D D × D 维的矩阵求逆,其可以通过矩阵求逆引理来降低计算量:
C − 1 = σ − 2 I − σ − 2 W M − 1 W T M = W T W + σ 2 I \begin{align*}
C^{-1} = \sigma^{-2}\mathbf{I} - \sigma^{-2}\mathbf{W}\mathbf{M}^{-1}\mathbf{W}^{T}\tag{2.11} \\\\
\mathbf{M} = \mathbf{W}^\text{T}\mathbf{W} + \sigma^2 \mathbf{I}\tag{2.12}
\end{align*} C − 1 = σ − 2 I − σ − 2 W M − 1 W T M = W T W + σ 2 I ( 2.11 ) ( 2.12 )
其中 M \mathbf{M} M 是一个 M × M M\times M M × M 维的矩阵,从而降低了求逆维度。
矩阵求逆引理见 Appendix A2.1
我们将观测数据 x \mathbf{x} x 的边缘概率分布称之为预测分布,除此之外我们还需要后验分布
p ( z ∣ x ) = N ( z ∣ M − 1 W T ( x − μ ) , σ 2 M − 1 ) (2.13) \begin{align*}
p(\mathbf{z} \mid \mathbf{x}) = \mathcal{N} \left(\mathbf{z} \mid \mathbf{M}^{-1}\mathbf{W}^\text{T}(\mathbf{x} - \boldsymbol{\mu}), \sigma^2 \mathbf{M}^{-1} \right)
\end{align*}\tag{2.13} p ( z ∣ x ) = N ( z ∣ M − 1 W T ( x − μ ) , σ 2 M − 1 ) ( 2.13 )
该公式的推导来自一个定理,见 Appendix A 2.2。
需要注意的是后验分布的均值依赖于观测数据 x \mathbf{x} x ,而方差则与观测数据无关。
16.2.3 Maximum likelihood
现在我们考虑最大化似然。我们有观测数据 X = { x n } \mathbf{X}=\{\mathbf{x}_n\} X = { x n } ,根据公式 2.5 和 2.6 和 2.10 我们可以写出似然函数
ln p ( X ∣ μ , W , σ 2 ) = ∑ n = 1 N ln p ( x n ∣ W , μ , σ 2 ) = − N D 2 ln ( 2 π ) − N 2 ln ∣ C ∣ − 1 2 ∑ n = 1 N ( x n − μ ) T C − 1 ( x n − μ ) . (2.14) \begin{align*}
\ln p(\mathbf{X} \mid \boldsymbol{\mu}, \mathbf{W}, \sigma^2) = \sum_{n=1}^{N} \ln p(\mathbf{x}_n \mid \mathbf{W}, \boldsymbol{\mu}, \sigma^2) = -\frac{ND}{2} \ln(2\pi) - \frac{N}{2} \ln |\mathbf{C}| - \frac{1}{2} \sum_{n=1}^{N} (\mathbf{x}_n - \boldsymbol{\mu})^T \mathbf{C}^{-1} (\mathbf{x}_n - \boldsymbol{\mu}).
\end{align*}\tag{2.14} ln p ( X ∣ μ , W , σ 2 ) = n = 1 ∑ N ln p ( x n ∣ W , μ , σ 2 ) = − 2 N D ln ( 2 π ) − 2 N ln ∣ C ∣ − 2 1 n = 1 ∑ N ( x n − μ ) T C − 1 ( x n − μ ) . ( 2.14 )
对数似然对 μ \boldsymbol{\mu} μ 求偏导,令偏导等于零可以得到 μ = x ˉ \boldsymbol{\mu}=\bar{\mathbf{x}} μ = x ˉ 。求导过程见 Appendix A2.3 和 A2.4。
对 W \mathbf{W} W 和 σ \sigma σ 求偏导过于复杂,推导过程就不写了,但是依然存在闭式解。
W ML = U M ( L M − σ 2 I ) 1 / 2 R (2.15) \begin{align*}
\mathbf{W}_{\text{ML}} = \mathbf{U}_M (\mathbf{L}_M - \sigma^2 \mathbf{I})^{1/2} \mathbf{R}
\end{align*}\tag{2.15} W ML = U M ( L M − σ 2 I ) 1/2 R ( 2.15 )
其中 U M \mathbf{U}_{M} U M 是一个 D × M D\times M D × M 维矩阵,包含了观测数据的协方差矩阵 S \mathbf{S} S 的特征向量的前 M M M 列,对应了最大的 M M M 个特征值。 L M \mathbf{L}_M L M 表示一个 M × M M\times M M × M 的对角阵,元素为特征向量所对应的特征值。R \mathbf{R} R 表示一个任意的正交阵,其代表了主子空间旋转,由于前面提到的旋转不变性存在,因此其解不唯一。
除了特征值最大的 M M M 个特征向量之外,选择其他特征向量会导致最大似然估计落在鞍点上,这对最大化没有意义。
σ ML 2 = 1 D − M ∑ i = M + 1 D λ i (2.16) \begin{align*}
\sigma^2_{\text{ML}} = \frac{1}{D-M} \sum_{i=M+1}^{D} \lambda_i
\end{align*}\tag{2.16} σ ML 2 = D − M 1 i = M + 1 ∑ D λ i ( 2.16 )
σ M L 2 \sigma^2_{ML} σ M L 2 则描述了被丢弃的部分所解构出的噪声方差。
我们将公式 2.15 代入 公式 2.6,得到
C = U M ( L M − σ 2 I ) 1 / 2 R ( R T ( L M − σ 2 I ) 1 / 2 U M T ) + σ 2 I = U M ( L M − σ 2 I ) U M T + σ 2 I \begin{align*}
\mathbf{C} &= \mathbf{U}_M (\mathbf{L}_M - \sigma^2 \mathbf{I})^{1/2} \mathbf{R} (\mathbf{R}^T (\mathbf{L}_M - \sigma^2 \mathbf{I})^{1/2} \mathbf{U}_M^T) + \sigma^2 \mathbf{I}\tag{2.17}\\\\
&= \mathbf{U}_M (\mathbf{L}_M - \sigma^2 \mathbf{I}) \mathbf{U}_M^T + \sigma^2 \mathbf{I}\tag{2.18}
\end{align*} C = U M ( L M − σ 2 I ) 1/2 R ( R T ( L M − σ 2 I ) 1/2 U M T ) + σ 2 I = U M ( L M − σ 2 I ) U M T + σ 2 I ( 2.17 ) ( 2.18 )
前面提到 R \mathbf{R} R 是一个任意的正交阵,但是如果我们将其作为单位正交阵,那么就可以从公式 2.17 简化为 2.18。此时其中 L M − σ 2 I \mathbf{L}_M - \sigma^2 \mathbf{I} L M − σ 2 I 描述了对于主子空间特征向量矩阵 U M \mathbf{U}_M U M 中的每个特征向量 u i \mathbf{u_i} u i 进行 λ i − σ 2 \sqrt{\lambda_i - \sigma^2} λ i − σ 2 缩放。
观测数据被认为是由潜在变量和噪声的结合产生的。其中,潜在变量通过矩阵 W \mathbf{W} W 投影到数据空间,并加上独立、同分布的高斯噪声。协方差矩阵 C \mathbf{C} C 由两部分组成:由 W \mathbf{W} W 投影所引入的变化和噪声的方差 σ 2 \sigma^2 σ 2 。当我们希望找到表征数据主要结构的投影,即主成分,在引入噪声参数 σ 2 \sigma^2 σ 2 后,这些主要的方差特征(即特征值)要减去噪声的方差贡献,以得到在没有噪声干扰下的数据空间方差。这个就是( λ i − σ 2 ) (\lambda_i - \sigma^2) ( λ i − σ 2 ) ,其中λ i \lambda_i λ i 是特征值(公式 2.18)。最后,对应的实际投影是特征向量u i \mathbf{u}_i u i 乘以方差的平方根( λ i − σ 2 ) \sqrt{(\lambda_i - \sigma^2)} ( λ i − σ 2 ) (公式 2.15),这样做是因为在高斯分布中,若从一个分布映射到另一个分布中,方差的变换是平方根关系的,保持数据的尺度不变。
后面的内容说实话我没有完全读懂,因此只对大概的意思做一个简要总结。
首先说明了 Probabilisic PCA 可以完全捕捉到主子空间各个方向上的方差,同时使用特征值的平均来近似其他方向的方差。
其次说明了对于 R \mathbf{R} R 可以通过某些约束来限制其为正交阵。
和传统 PCA 将数据空间映射到潜在空间不同,Probabilisic PCA 通过公式 2.3 将潜在空间映射到数据空间。通过贝叶斯定理,我们可以使用后验分布 2.13 来构建将数据映射到潜在空间的过程,同时计算出其均值和方差。当我们令方差趋于零时,实际上就回到了传统 PCA,因为映射不再具有不确定性,而是精确的映射到某一个点。
最后描述了一些关于参数量的计算,说实话我没看懂。
之后的小节是一些扩展内容,讲述了几种衍生方法。
16.2.4 Factor analysis
因子分析和 Probabilisic PCA 的区别在于,其观测数据的条件分布具有对角协方差矩阵。
回顾 Probabilisic PCA,其定义观测数据定义为 p ( x ∣ z ) = N ( x ∣ W z + μ , σ 2 I ) p(\mathbf{x}\mid\mathbf{z})=\mathcal{N}(\mathbf{x}\mid\mathbf{Wz}+\boldsymbol{\mu},\sigma^2\mathbf{I}) p ( x ∣ z ) = N ( x ∣ Wz + μ , σ 2 I ) ,其协方差矩阵表明了对于所有方向而言,方差都是一样的,这也叫各向同性高斯分布。但是因子分析将协方差定义为一个对角阵,即各个方向可以有自己的方差,这使得因子分析更加灵活,允许观测数据有自己独有的方差信息。
16.2.5 Independent Component Analysis
前文提到的旋转不变性是由于潜变量遵循高斯分布导致的,为了解决这个问题就要允许潜变量可以是非高斯分布。这就是独立成分分析。
16.2.6 Kalman Filters
前面提到的所有方法都假设了观测数据是独立同分布的,那如果观测数据不是独立同分布呢?这就是卡尔曼滤波或者叫线性动态系统。
对于这种数据而言,潜变量之间不再是相互独立,而是呈现出马尔科夫链的依赖关系。
16.3 Evidence Lower Bound
首先介绍一下 ELBO 推导过程。
对于模型 p ( x , z ∣ w ) p(\mathbf{x},\mathbf{z}\mid\mathbf{w}) p ( x , z ∣ w ) ,我们已经定义观测数据 x \mathbf{x} x 是由潜在变量 z \mathbf{z} z 控制生成的,相关公式在 2.1~2.6 中定义,这里不再赘述,需要注意的是我们此时将 w \mathbf{w} w 作为一个可学习参数列了出来。现在我们希望最大化对数似然 p ( x ∣ w ) p(\mathbf{x\mid\mathbf{w}}) p ( x ∣ w ) 从而达到生成观测数据 x \mathbf{x} x 的目的。
ln p ( x ∣ w ) = ln ∫ p ( x , z ∣ w ) d z (3.1) \begin{align*}
\ln p(\mathbf{x}\mid\mathbf{w})&=\ln \int p(\mathbf{x},\mathbf{z}\mid\mathbf{w})d\mathbf{z}
\end{align*}\tag{3.1} ln p ( x ∣ w ) = ln ∫ p ( x , z ∣ w ) d z ( 3.1 )
这个公式中涉及到对 z \mathbf{z} z 的积分,但是通常对于高维潜在变量的积分很难求解,因此我们采用变分推断的思想,引入变分分布 q ( z ) q(\mathbf{z}) q ( z ) 。
ln p ( x ∣ w ) = ln ∫ q ( z ) p ( x , z ∣ w ) q ( z ) d z ≥ ∫ q ( z ) ln ( p ( x , z ∣ w ) q ( z ) ) d z = E q ( z ) [ ln p ( x , z ∣ w ) − ln q ( z ) ] (3.2) \begin{align*}
\ln p(\mathbf{x}|\mathbf{w}) &= \ln \int q(\mathbf{z}) \frac{p(\mathbf{x}, \mathbf{z}|\mathbf{w})}{q(\mathbf{z})} \, d\mathbf{z}\\\\
&\geq \int q(\mathbf{z}) \ln \left(\frac{p(\mathbf{x},\mathbf{z}|\mathbf{w})}{q(\mathbf{z})}\right) \, d\mathbf{z} = \mathbb{E}_{q(\mathbf{z})}\left[\ln p(\mathbf{x}, \mathbf{z}|\mathbf{w}) - \ln q(\mathbf{z})\right]
\end{align*}\tag{3.2} ln p ( x ∣ w ) = ln ∫ q ( z ) q ( z ) p ( x , z ∣ w ) d z ≥ ∫ q ( z ) ln ( q ( z ) p ( x , z ∣ w ) ) d z = E q ( z ) [ ln p ( x , z ∣ w ) − ln q ( z ) ] ( 3.2 )
这个 E q ( z ) [ ln p ( x , z ∣ w ) − ln q ( z ) ] \mathbb{E}_{q(\mathbf{z})}\left[\ln p(\mathbf{x}, \mathbf{z}|\mathbf{w}) - \ln q(\mathbf{z})\right] E q ( z ) [ ln p ( x , z ∣ w ) − ln q ( z ) ] 就是证据下界 ELBO。接下来我们使用一个技巧来进一步变化 ELBO,将 p ( x , z ∣ w ) p(\mathbf{x},\mathbf{z}∣\mathbf{w}) p ( x , z ∣ w ) 分解为 p ( z ∣ x , w ) p(\mathbf{z}|\mathbf{x}, \mathbf{w}) p ( z ∣ x , w ) 和 p ( x ∣ w ) p(\mathbf{x}∣\mathbf{w}) p ( x ∣ w ) 。
∫ q ( z ) ln ( p ( x , z ∣ w ) q ( z ) ) d z = ∫ q ( z ) ln ( p ( x ∣ w ) p ( z ∣ x , w ) q ( z ) ) d z = ∫ q ( z ) ln p ( z ∣ x , w ) q ( z ) d z + ∫ q ( z ) ln p ( x ∣ w ) d z = E q ( z ) [ ln p ( z ∣ x , w ) q ( z ) ] + E q ( z ) [ ln p ( x ∣ w ) ] E q ( z ) [ ln p ( x , z ∣ w ) − ln q ( z ) ] = E q ( z ) [ ln p ( z ∣ x , w ) q ( z ) ] + E q ( z ) [ ln p ( x ∣ w ) ] ln p ( x ∣ w ) = E q ( z ) [ ln p ( x , z ∣ w ) − ln q ( z ) ] − E q ( z ) [ ln p ( z ∣ x , w ) q ( z ) ] = L ( q , w ) + KL ( q ( z ) ∣ ∣ p ( z ∣ x , w ) ) \begin{align*}
\int q(\mathbf{z}) \ln \left(\frac{p(\mathbf{x},\mathbf{z}|\mathbf{w})}{q(\mathbf{z})}\right) \, d\mathbf{z}
&=\int q(\mathbf{z}) \ln \left(\frac{p(\mathbf{x}|\mathbf{w})p(\mathbf{z}|\mathbf{x}, \mathbf{w})}{q(\mathbf{z})}\right) \, d\mathbf{z}\\\\
&=\int q(\mathbf{z})\ln \frac{p(\mathbf{z}|\mathbf{x},\mathbf{w})}{q(\mathbf{z})}d\mathbf{z}+\int q(\mathbf{z})\ln p(\mathbf{x\mid\mathbf{w}})d\mathbf{z}\\\\
&=\mathbb{E}_{q(\mathbf{z})}\left[\ln \frac{p(\mathbf{z}\mid\mathbf{x},\mathbf{w})}{q(\mathbf{z})}\right]+\mathbb{E}_{q(\mathbf{z})}\left[\ln p(\mathbf{x\mid\mathbf{w}})\right]\\\\
\mathbb{E}_{q(\mathbf{z})}\left[\ln p(\mathbf{x}, \mathbf{z}|\mathbf{w}) - \ln q(\mathbf{z})\right]
&=\mathbb{E}_{q(\mathbf{z})}\left[\ln \frac{p(\mathbf{z}\mid\mathbf{x},\mathbf{w})}{q(\mathbf{z})}\right]+\mathbb{E}_{q(\mathbf{z})}\left[\ln p(\mathbf{x\mid\mathbf{w}})\right]\\\\
\ln p(\mathbf{x\mid\mathbf{w}})
&=\mathbb{E}_{q(\mathbf{z})}\left[\ln p(\mathbf{x}, \mathbf{z}|\mathbf{w}) - \ln q(\mathbf{z})\right]-\mathbb{E}_{q(\mathbf{z})}\left[\ln \frac{p(\mathbf{z}\mid\mathbf{x},\mathbf{w})}{q(\mathbf{z})}\right]\\\\
&=\mathcal{L}(q,\mathbf{w})+\text{KL}(q(\mathbf{z})\mid\mid p(\mathbf{z}\mid\mathbf{x},\mathbf{w}))\tag{3.3}
\end{align*} ∫ q ( z ) ln ( q ( z ) p ( x , z ∣ w ) ) d z E q ( z ) [ ln p ( x , z ∣ w ) − ln q ( z ) ] ln p ( x ∣ w ) = ∫ q ( z ) ln ( q ( z ) p ( x ∣ w ) p ( z ∣ x , w ) ) d z = ∫ q ( z ) ln q ( z ) p ( z ∣ x , w ) d z + ∫ q ( z ) ln p ( x ∣ w ) d z = E q ( z ) [ ln q ( z ) p ( z ∣ x , w ) ] + E q ( z ) [ ln p ( x ∣ w ) ] = E q ( z ) [ ln q ( z ) p ( z ∣ x , w ) ] + E q ( z ) [ ln p ( x ∣ w ) ] = E q ( z ) [ ln p ( x , z ∣ w ) − ln q ( z ) ] − E q ( z ) [ ln q ( z ) p ( z ∣ x , w ) ] = L ( q , w ) + KL ( q ( z ) ∣∣ p ( z ∣ x , w )) ( 3.3 )
其中
L ( q , w ) = ∫ q ( z ) ln { p ( x , z ∣ w ) q ( z ) } d z = E q ( z ) [ ln p ( z ∣ x , w ) q ( z ) ] KL ( q ( z ) ∥ p ( z ∣ x , w ) ) = − ∫ q ( z ) ln { p ( z ∣ x , w ) q ( z ) } d z = E q ( z ) [ ln p ( x ∣ w ) ] \begin{align*}
\mathcal L(q, \mathbf{w}) &= \int q(\mathbf{z}) \ln \left\{ \frac{p(\mathbf{x}, \mathbf{z}|\mathbf{w})}{q(\mathbf{z})} \right\} \, d\mathbf{z}=\mathbb{E}_{q(\mathbf{z})}\left[\ln \frac{p(\mathbf{z}\mid\mathbf{x},\mathbf{w})}{q(\mathbf{z})}\right]\tag{3.4}\\\\
\text{KL}(q(\mathbf{z}) \| p(\mathbf{z}|\mathbf{x}, \mathbf{w})) &= - \int q(\mathbf{z}) \ln \left\{ \frac{p(\mathbf{z}|\mathbf{x}, \mathbf{w})}{q(\mathbf{z})} \right\} \, d\mathbf{z}=\mathbb{E}_{q(\mathbf{z})}\left[\ln p(\mathbf{x\mid\mathbf{w}})\right]\tag{3.5}
\end{align*} L ( q , w ) KL ( q ( z ) ∥ p ( z ∣ x , w )) = ∫ q ( z ) ln { q ( z ) p ( x , z ∣ w ) } d z = E q ( z ) [ ln q ( z ) p ( z ∣ x , w ) ] = − ∫ q ( z ) ln { q ( z ) p ( z ∣ x , w ) } d z = E q ( z ) [ ln p ( x ∣ w ) ] ( 3.4 ) ( 3.5 )
由于可知,对数似然 p ( x ∣ w ) p(\mathbf{x\mid\mathbf{w}}) p ( x ∣ w ) 在使用变分推断后构建出的 ELBO 其实就是缺少了 KL ( q ( z ) ∥ p ( z ∣ x , w ) ) \text{KL}(\mathbf{q}(\mathbf{z}) \| p(\mathbf{z}|\mathbf{x}, \mathbf{w})) KL ( q ( z ) ∥ p ( z ∣ x , w )) 这一部分,由于 KL 散度始终大于零,这个地方为了将对数似然变为两项之和,所以将负号移到了 KL 散度的内部。因此 3.5 式恒小于等于零,从而得到对数似然大于等于 ELBO 的结果。
写出了对数似然接下来我们就要最大化对数似然,我们在这里使用一种叫做期望最大化(Expectation Maximum;EM)的方法。
实际上 PPCA 我们前面已经写出了闭式解,通过特征分解就可以写出 p ( x ) p(x) p ( x ) ,但是在实际使用时由于观测数据维度过高,对协方差矩阵进行特征分解往往很困z难,此时 EM 算法的迭代策略就可以通过多步迭代计算进行求解。
EM 算法分为两步,在 E 步中我们改变 q ( z ) q(\mathbf{z}) q ( z ) 最大化 L ( q , w ) \mathcal L(\mathbf{q}, \mathbf{w}) L ( q , w ) ,在 M 步我们改变 w \mathbf{w} w 最大化 L ( q , w ) \mathcal L(\mathbf{q}, \mathbf{w}) L ( q , w ) 。具体而言过程如下。
首先我们随机初始化参数 w o l d \mathbf{w}^{old} w o l d ,然后进行 E 步,令参数 w \mathbf{w} w 固定,通过更新 q ( z ) q(\mathbf{z}) q ( z ) 来最大化下界。根据公式 3.3 可知,此时由于我们固定了 w \mathbf{w} w ,那么最大化下界就等同于最小化 KL 散度,也就是令变分分布 q ( z ) q(\mathbf{z}) q ( z ) 等于后验分布 p ( z ∥ p ( z ∣ x , w o l d ) ) p(\mathbf{z}\|p(\mathbf{z}\mid\mathbf{x},\mathbf{w}^{old})) p ( z ∥ p ( z ∣ x , w o l d )) 。
接下来在 M 步,我们首先代入 E 步结果 q ( z ) = p ( z ∥ p ( z ∣ x , w o l d ) ) q(\mathbf{z})=p(\mathbf{z}\|p(\mathbf{z}\mid\mathbf{x},\mathbf{w}^{old})) q ( z ) = p ( z ∥ p ( z ∣ x , w o l d )) ,然后固定 q ( z ) q(\mathbf{z}) q ( z ) ,更新 w \mathbf{w} w 来最大化下界,此时的下界公式如下:
L ( q , w ) = ∫ p ( z ∣ x , w ( old ) ) ln p ( x , z ∣ w ) d z − ∫ p ( z ∣ x , w ( old ) ) ln p ( z ∣ x , w ( old ) ) d z (3.6) \begin{align*}
\mathcal L(q, \mathbf{w}) = \int p( \mathbf{z} \mid \mathbf{x}, \mathbf{w}^{(\text{old})}) \ln p( \mathbf{x}, \mathbf{z} \mid \mathbf{w}) \, d \mathbf{z} - \int p( \mathbf{z} \mid \mathbf{x}, \mathbf{w}^{(\text{old})}) \ln p( \mathbf{z} \mid \mathbf{x}, \mathbf{w}^{(\text{old})}) \, d \mathbf{z}
\end{align*}\tag{3.6} L ( q , w ) = ∫ p ( z ∣ x , w ( old ) ) ln p ( x , z ∣ w ) d z − ∫ p ( z ∣ x , w ( old ) ) ln p ( z ∣ x , w ( old ) ) d z ( 3.6 )
上面这个公式是将 E 步结果代入公式 3.4 得到,需要注意其中带有 w o l d \mathbf{w}^{old} w o l d 的项都是 E 步得到的 q ( z ) = p ( z ∥ p ( z ∣ x , w o l d ) ) q(\mathbf{z})=p(\mathbf{z}\|p(\mathbf{z}\mid\mathbf{x},\mathbf{w}^{old})) q ( z ) = p ( z ∥ p ( z ∣ x , w o l d )) 代换得到的,而在 M 步我们要固定 q ( z ) q(\mathbf{z}) q ( z ) 不变,因此这一步中实际更新的参数只有 w \mathbf{w} w 这一个。
从公式 3.6 不难看出,等号右边第二项在 M 步最大化时和参数 w \mathbf{w} w 完全无关,而第一项则是完整数据对数似然的期望,该期望是相对于 q ( z ) = p ( z ∥ p ( z ∣ x , w o l d ) ) q(\mathbf{z})=p(\mathbf{z}\|p(\mathbf{z}\mid\mathbf{x},\mathbf{w}^{old})) q ( z ) = p ( z ∥ p ( z ∣ x , w o l d )) 而言的。
通过不断进行 E 步和 M 步迭代,我们会逐渐逼近模型的真实似然,通常会设定一个最大迭代步数或者两次迭代之间的变化最小值,然后停止迭代得到结果。
当我们有观测数据 X = { x 1 , … , x 2 } \mathbf{X}=\{ \mathbf{x}_1,\dots,\mathbf{x}_2\} X = { x 1 , … , x 2 } 。我们定义 ELBO 为所有观测变量的和。
L ( q , w ) = ∑ n = 1 N ∫ q ( z n ) ln { p ( x n , z n ∣ w ) q ( z n ) } d z n (3.7) \begin{align*}
\mathcal L(q, \mathbf{w}) = \sum_{n=1}^{N} \int q(\mathbf{z}_n) \ln \left\{ \frac{p(\mathbf{x}_n, \mathbf{z}_n|\mathbf{w})}{q(\mathbf{z}_n)} \right\} \, d\mathbf{z}_n
\end{align*}\tag{3.7} L ( q , w ) = n = 1 ∑ N ∫ q ( z n ) ln { q ( z n ) p ( x n , z n ∣ w ) } d z n ( 3.7 )
16.3.1 Expectation Maximization
上面我们使用 EM 算法最大化 ELBO,但是我们也可以遵循 EM 算法的一般形式,直接对完整数据的对数似然进行迭代。
首先我们写出完整数据的对数似然:
ln p ( X , Z ∣ μ , W , σ 2 ) = ∑ n = 1 N { ln p ( x n ∣ z n ) + ln p ( z n ) } (3.8) \begin{align*}
\ln p(\mathbf{X}, \mathbf{Z} \mid \boldsymbol{\mu}, \mathbf{W}, \sigma^2) = \sum_{n=1}^{N} \left\{ \ln p(\mathbf{x}_n \mid \mathbf{z}_n) + \ln p(\mathbf{z}_n) \right\}
\end{align*}\tag{3.8} ln p ( X , Z ∣ μ , W , σ 2 ) = n = 1 ∑ N { ln p ( x n ∣ z n ) + ln p ( z n ) } ( 3.8 )
其中 Z \mathbf{Z} Z 的每一行表示一个潜在变量 z n \mathbf{z}_n z n 。通过公式 2.14,A2.3 和 A2.4,我们已经知道 μ \boldsymbol{\mu} μ 的闭式解 μ = x ˉ \boldsymbol{\mu}=\bar{\mathbf{x}} μ = x ˉ ,因此这里可以直接代入。通过代入公式 2.1 和 2.2 后将分布展开为表达式,我们可以得到对数似然的期望。
E [ ln p ( X , Z ∣ μ , W , σ 2 ) ] = ∑ n = 1 N { − D 2 ln ( 2 π σ 2 ) − M 2 ln ( 2 π ) = − 1 2 σ 2 ∥ x n − μ ∥ 2 + 1 σ 2 E [ z n ] T W T ( x n − μ ) = − 1 2 σ 2 Tr ( W T W E [ z n z n T ] ) − 1 2 Tr ( E [ z n z n T ] ) } (3.9) \begin{align*}
\mathbb{E}\left[\ln p(\mathbf{X}, \mathbf{Z} \mid \boldsymbol{\mu}, \mathbf{W}, \sigma^2)\right]
&= \sum_{n=1}^{N} \left\{-\frac{D}{2} \ln(2\pi\sigma^2) - \frac{M}{2} \ln(2\pi) \right. \\\\
&\phantom{=} \quad
- \frac{1}{2\sigma^2} \|\mathbf{x}_n - \boldsymbol{\mu}\|^2
+ \frac{1}{\sigma^2} \mathbb{E}[\mathbf{z}_n]^T \mathbf{W}^T (\mathbf{x_n} - \boldsymbol{\mu}) \\\\
&\phantom{=} \quad
- \frac{1}{2\sigma^2} \text{Tr}(\mathbf{W}^T \mathbf{W} \mathbb{E}[\mathbf{z}_n \mathbf{z}_n^T])
- \frac{1}{2} \text{Tr}(\mathbb{E}[\mathbf{z}_n \mathbf{z}_n^T])
\left.\right\}
\end{align*}\tag{3.9} E [ ln p ( X , Z ∣ μ , W , σ 2 ) ] = n = 1 ∑ N { − 2 D ln ( 2 π σ 2 ) − 2 M ln ( 2 π ) = − 2 σ 2 1 ∥ x n − μ ∥ 2 + σ 2 1 E [ z n ] T W T ( x n − μ ) = − 2 σ 2 1 Tr ( W T W E [ z n z n T ]) − 2 1 Tr ( E [ z n z n T ]) } ( 3.9 )
具体推导过程见 Appendix A3。
在 E 步中,我们固定 μ = x ˉ , W , σ 2 \boldsymbol{\mu}=\bar{\mathbf{x}},\mathbf{W},\sigma^2 μ = x ˉ , W , σ 2 ,计算潜在变量的期望,通过公式 2.13 和 E [ z n z n T ] = cov [ z n ] + E [ z n ] E [ z n ] T \mathbb{E}[\mathbf{z}_n \mathbf{z}_n^T] = \text{cov}[\mathbf{z}_n] + \mathbb{E}[\mathbf{z}_n] \mathbb{E}[\mathbf{z}_n]^T E [ z n z n T ] = cov [ z n ] + E [ z n ] E [ z n ] T ,可以得到
E [ z n ] = M − 1 W T ( x n − x ) E [ z n z n T ] = σ 2 M − 1 + E [ z n ] E [ z n ] T \begin{align*}
\mathbb{E}[\mathbf{z}_n] = \mathbf{M}^{-1} \mathbf{W}^T (\mathbf{x}_n - \mathbf{x}) \tag{3.10}\\\\
\mathbb{E}[\mathbf{z}_n \mathbf{z}_n^T] = \sigma^2 \mathbf{M}^{-1} + \mathbb{E}[\mathbf{z}_n] \mathbb{E}[\mathbf{z}_n]^T \tag{3.11}
\end{align*} E [ z n ] = M − 1 W T ( x n − x ) E [ z n z n T ] = σ 2 M − 1 + E [ z n ] E [ z n ] T ( 3.10 ) ( 3.11 )
在 M 步,我们通过以下公式更新 W , σ 2 \mathbf{W},\boldsymbol{\sigma}^2 W , σ 2 来最大化期望。
W new = [ ∑ n = 1 N ( x n − x ) E [ z n ] T ] [ ∑ n = 1 N E [ z n z n T ] ] − 1 σ new 2 = 1 N D ∑ n = 1 N { ∥ x n − x ∥ 2 − 2 E [ z n ] T W new T ( x n − x ) + Tr ( E [ z n z n T ] W new T W new ) } \begin{align*}
\mathbf{W}_{\text{new}} &= \left[ \sum_{n=1}^{N} (\mathbf{x}_n - \mathbf{x}) \mathbb{E}[\mathbf{z}_n]^T \right] \left[ \sum_{n=1}^{N} \mathbb{E}[\mathbf{z}_n \mathbf{z}_n^T] \right]^{-1} \tag{3.12}\\\\
\sigma^2_{\text{new}} &= \frac{1}{ND} \sum_{n=1}^{N} \left\{ \|\mathbf{x}_n - \mathbf{x}\|^2 - 2 \mathbb{E}[\mathbf{z}_n]^T \mathbf{W}_{\text{new}}^T (\mathbf{x}_n - \mathbf{x}) + \text{Tr} \left( \mathbb{E}[\mathbf{z}_n \mathbf{z}_n^T] \mathbf{W}_{\text{new}}^T \mathbf{W}_{\text{new}} \right) \right\} \tag{3.13}
\end{align*} W new σ new 2 = [ n = 1 ∑ N ( x n − x ) E [ z n ] T ] [ n = 1 ∑ N E [ z n z n T ] ] − 1 = N D 1 n = 1 ∑ N { ∥ x n − x ∥ 2 − 2 E [ z n ] T W new T ( x n − x ) + Tr ( E [ z n z n T ] W new T W new ) } ( 3.12 ) ( 3.13 )
具体推导过程就不写了,说真的写公式写的我头疼。
之后还有两个小节分别介绍了 EM 算法在 PCA 和因子分析中的应用,我就跳过了。
16.4 Nonlinear Latent Variable Models
前面讨论的都是线性模型,即观测数据和潜在变量是线性变换的关系。之后我们会讨论从潜在空间到数据空间的非线性映射,这里先简单的了解一下。
我们首先定义一个服从标准正态分布的潜在变量 z \mathbf{z} z
p z ( z ) = N ( z ∣ 0 , I ) (4.1) \begin{align*}
p_{\mathbf{z}}(\mathbf{z})=\mathcal{N}(z|\mathbf{0},\mathbf{I})
\end{align*}\tag{4.1} p z ( z ) = N ( z ∣ 0 , I ) ( 4.1 )
然后我们定一个从潜在空间到数据空间的映射 x = g ( z , w ) \mathbf{x}=g(\mathbf{z}, \mathbf{w}) x = g ( z , w ) ,这个映射实际由一个神经网络实现。
为了可以从数据中学习这个映射,我们定义似然函数
p x ( x ) = p z ( z ( x ) ) ∣ det J ( x ) ∣ (4.2) \begin{align}
p_{\mathbf{x}}(\mathbf{x})=p_{\mathbf{z}}(\mathbf{z}(\mathbf{x}))|\det \mathbf{J}(\mathbf{x})|
\end{align}\tag{4.2} p x ( x ) = p z ( z ( x )) ∣ det J ( x ) ∣ ( 4.2 )
这个似然来自变量变化公式,描述见 Appendix A4。
其中
J i j ( x ) = ∂ z i ∂ x j (4.3) \begin{align*}
J_{ij}(x) = \frac{\partial z_i}{\partial x_j}
\end{align*}\tag{4.3} J ij ( x ) = ∂ x j ∂ z i ( 4.3 )
为了求解雅各比行列式,我们需要 z = g − 1 ( x , w ) \mathbf{z}=g^{-1}(\mathbf{x},\mathbf{w}) z = g − 1 ( x , w ) 。但是这个由神经网络定义的映射往往是不可逆的。
一种方法是对神经网络做出限制,要求潜在变量空间和数据空间维度一致,从而使映射可逆,这是归一化流(normalizing flows)的思路。但是这种要求对于潜在空间而言是一种很强的限制。正如前文曾提到的,潜在变量往往落在数据空间的一个低维流形上。
16.4.1 Nolinear Manifolds
如果我们考虑潜在空间就是数据空间的一个低维流形,那么此时会面临另一个问题。当数据点没有落在这个低维流形上时,其概率密度将为零。但是实际中由于噪声等因素的存在,数据点往往不会精确的落在低维流形上,而是有一定的偏移。这个问题会导致模型难以学习,因此我们定义了一个条件分布。
p ( x ∣ z , w ) = N ( x ∣ g ( z , w ) , σ 2 I ) (4.4) \begin{align*}
\mathbf{p}(\mathbf{x}|\mathbf{z}, \mathbf{w}) = \mathcal{N}(\mathbf{x}|\mathbf{g}(\mathbf{z}, \mathbf{w}), \sigma^2 \mathbf{I})
\end{align*}\tag{4.4} p ( x ∣ z , w ) = N ( x ∣ g ( z , w ) , σ 2 I ) ( 4.4 )
这个条件分布以潜在空间到数据空间的映射作为均值,允许数据点在一定范围内偏离低维流形。
此时整个数据生成过程就已经被定义了,我们先从潜在空间中采样,将采样得到的潜在变量输入神经网络,使用神经网络的输出作为均值,以 σ 2 \sigma^2 σ 2 作为方差采样生成数据。
16.2 Likelihood Function
现在我们希望根据观测数据来最大化似然,从而得到这个模型。
p ( x ∣ w ) = ∫ p ( x ∣ z , w ) p ( z ) d z = ∫ N ( x ∣ g ( z , w ) , σ 2 I ) N ( z ∣ 0 , I ) d z (4.5) \begin{align*}
p(\mathbf{x}|\mathbf{w}) = \int p(\mathbf{x}|\mathbf{z}, \mathbf{w})p(\mathbf{z}) \, dz = \int \mathcal{N}(\mathbf{x}|\mathbf{g}(\mathbf{z}, \mathbf{w}), \sigma^2 \mathbf{I}) \mathcal{N}(\mathbf{z}|\mathbf{0}, \mathbf{I}) \, dz
\end{align*}\tag{4.5} p ( x ∣ w ) = ∫ p ( x ∣ z , w ) p ( z ) d z = ∫ N ( x ∣ g ( z , w ) , σ 2 I ) N ( z ∣ 0 , I ) d z ( 4.5 )
但是由神经网络定义的映射 g g g 往往是高度非线性的,这很难积分。一种方法是通过不断地从潜在空间中采样,并使用很多样本来近似上面这个似然,但是实际使用时需要采样的数量庞大到不可接受。
讨论基本到这里就结束了,后文提到之后会介绍四种基于上述过程的生成式模型,他们各自采用了不同的方法来解决上面的问题。分别是生成对抗网络、归一化流、自编码器和扩散模型。
Appandix
Appendix A1
A1 是 CH16.1 中的公式推导过程。
1 N ∑ n = 1 N u t T x n = u t T 1 N ∑ n = 1 N x n = u t T x ˉ (A1.1) \begin{align*}
\frac{1}{N}\sum_{n=1}^{N}\mathbf{u}_t^T\mathbf{x}_n=\mathbf{u}_t^T\frac{1}{N}\sum_{n=1}^{N}\mathbf{x}_n=\mathbf{u}_t^T\bar{\mathbf{x}}
\end{align*}\tag{A1.1} N 1 n = 1 ∑ N u t T x n = u t T N 1 n = 1 ∑ N x n = u t T x ˉ ( A1.1 )
1 N ∑ n = 1 N { u t T x n − u t T x ˉ } 2 = 1 N ∑ n = 1 N [ u t T ( x n − x ˉ ) ] 2 = 1 N ∑ n = 1 N ( x n − x ˉ ) T u t u t T ( x n − x ˉ ) = u t T ( 1 N ∑ n = 1 N ( x n − x ˉ ) ( x n − x ˉ ) T ) u t = u t T S u t (A1.2) \begin{align*}
\frac{1}{N}\sum_{n=1}^{N}\{\mathbf{u}_t^T\mathbf{x}_n-\mathbf{u}_t^T\bar{\mathbf{x}}\}^2
&=\frac{1}{N}\sum_{n=1}^{N}\left[\mathbf{u}_t^T (\mathbf{x}_n - \bar{\mathbf{x}})\right]^2\\\\
&=\frac{1}{N}\sum_{n=1}^N (\mathbf{x}_n-\bar{\mathbf{x}})^T \mathbf{u}_t \mathbf{u}_t^T (\mathbf{x}_n-\bar{\mathbf{x}})\\\\
&=\mathbf{u}_t^T \left( \frac{1}{N}\sum_{n=1}^N (\mathbf{x}_n-\bar{\mathbf{x}})(\mathbf{x}_n-\bar{\mathbf{x}})^T \right) \mathbf{u}_t\\\\
&= \mathbf{u}_t^T \mathbf{S} \mathbf{u}_t
\end{align*}\tag{A1.2} N 1 n = 1 ∑ N { u t T x n − u t T x ˉ } 2 = N 1 n = 1 ∑ N [ u t T ( x n − x ˉ ) ] 2 = N 1 n = 1 ∑ N ( x n − x ˉ ) T u t u t T ( x n − x ˉ ) = u t T ( N 1 n = 1 ∑ N ( x n − x ˉ ) ( x n − x ˉ ) T ) u t = u t T S u t ( A1.2 )
L ( x , λ 1 ) = u t T S u t + λ 1 ( 1 − u t T u t ) ∂ L ∂ u t = ∂ ( u t T S u t ) ∂ u t + ∂ λ 1 ∂ u t − ∂ ( λ 1 u t T u t ) ∂ u t = 2 S u t − 2 λ 1 u t = 0 S u t = λ 1 u t (A1.3) \begin{align*}
\mathcal{L}(\mathbf{x},\lambda_1)&=\mathbf{u}_t^T\mathbf{S}\mathbf{u}_t + \lambda_1(1-\mathbf{u}_t^T\mathbf{u}_t) \\\\
\frac{\partial \mathcal{L}}{\partial \mathbf{u}_t}&=\frac{\partial (\mathbf{u}_t^T\mathbf{S}\mathbf{u}_t)}{\partial \mathbf{u}_t}+
\frac{\partial \lambda_1}{\partial \mathbf{u}_t} - \frac{\partial (\lambda_1\mathbf{u}_t^T\mathbf{u}_t)}{\partial \mathbf{u}_t}=
2\mathbf{S}\mathbf{u_t}-2\lambda_1\mathbf{u_t}=0\\\\
\mathbf{S}\mathbf{u}_t&=\lambda_1\mathbf{u}_t
\end{align*}\tag{A1.3} L ( x , λ 1 ) ∂ u t ∂ L S u t = u t T S u t + λ 1 ( 1 − u t T u t ) = ∂ u t ∂ ( u t T S u t ) + ∂ u t ∂ λ 1 − ∂ u t ∂ ( λ 1 u t T u t ) = 2 S u t − 2 λ 1 u t = 0 = λ 1 u t ( A1.3 )
J = 1 N ∑ n = 1 N ∣ ∣ x n − x ~ n ∣ ∣ 2 = 1 N ∑ n = 1 N ( x n − x ~ n ) T ( x n − x ~ n ) = 1 N ∑ n = 1 N ( x n − ∑ i = 1 M z n i u i − ∑ i = M + 1 D b i u i ) T ( x n − ∑ i = 1 M z n i u i − ∑ i = M + 1 D b i u i ) (A1.4) \begin{align*}
J&=\frac{1}{N}\sum_{n=1}^N\mid\mid \mathbf{x}_n-\tilde{\mathbf{x}}_n\mid\mid^2\\\\
&=\frac{1}{N}\sum_{n=1}^N(\mathbf{x}_n-\tilde{\mathbf{x}}_n)^T(\mathbf{x}_n-\tilde{\mathbf{x}}_n)\\\\
&=\frac{1}{N}\sum_{n=1}^N\left(\mathbf{x}_n-\sum_{i=1}^M z_{ni}\mathbf{u}_i-\sum_{i=M+1}^D b_i\mathbf{u}_i \right)^T\left(\mathbf{x}_n-\sum_{i=1}^M z_{ni}\mathbf{u}_i-\sum_{i=M+1}^D b_i\mathbf{u}_i \right)
\end{align*}\tag{A1.4} J = N 1 n = 1 ∑ N ∣∣ x n − x ~ n ∣ ∣ 2 = N 1 n = 1 ∑ N ( x n − x ~ n ) T ( x n − x ~ n ) = N 1 n = 1 ∑ N ( x n − i = 1 ∑ M z ni u i − i = M + 1 ∑ D b i u i ) T ( x n − i = 1 ∑ M z ni u i − i = M + 1 ∑ D b i u i ) ( A1.4 )
对于某个特定的数据点 x n x_n x n 的偏导可以不失一般性地推广到上式的求和中,因此下面我们以特定数据点的 x n x_n x n 的偏导过程为例。
首先给出向量微积分的基本公式,注意这里是对变量的求导,实际上等价于二次函数求导。后面会有对向量的求导,形式略有不同。
∂ ∂ α ( a − α u ) T ( a − α u ) = − 2 u T ( a − α u ) (A1.5) \begin{align*}
\frac{\partial}{\partial \alpha}(\mathbf{a}-\alpha\mathbf{u})^T(\mathbf{a}-\alpha\mathbf{u})=-2\mathbf{u}^T(\mathbf{a}-\alpha\mathbf{u})
\end{align*}\tag{A1.5} ∂ α ∂ ( a − α u ) T ( a − α u ) = − 2 u T ( a − α u ) ( A1.5 )
对于某个数据点的投影代价
J = ( x n − ∑ i = 1 M z n i u i − ∑ i = M + 1 D b i u i ) T ( x n − ∑ i = 1 M z n i u i − ∑ i = M + 1 D b i u i ) (A1.6) \begin{align*}
J=\left(\mathbf{x}_n-\sum_{i=1}^M z_{ni}\mathbf{u}_i-\sum_{i=M+1}^D b_i\mathbf{u}_i \right)^T\left(\mathbf{x}_n-\sum_{i=1}^M z_{ni}\mathbf{u}_i-\sum_{i=M+1}^D b_i\mathbf{u}_i \right)
\end{align*}\tag{A1.6} J = ( x n − i = 1 ∑ M z ni u i − i = M + 1 ∑ D b i u i ) T ( x n − i = 1 ∑ M z ni u i − i = M + 1 ∑ D b i u i ) ( A1.6 )
我们结合 A3.2 和 A3.3,对某个特定的 z n j z_{nj} z nj 求偏导,因此我们将 j j j 项单独拆出来
a = ( x n − ∑ i = 1 , i ≠ j M z n i u i − ∑ i = M + 1 D b i u i ) α u = z n j u j (A1.7) \begin{align*}
\mathbf{a}&=\left(\mathbf{x}_n-\sum_{i=1,i\ne j}^M z_{ni}\mathbf{u}_i-\sum_{i=M+1}^D b_i\mathbf{u}_i \right)\\\\
\alpha\mathbf{u}&=z_{nj}\mathbf{u}_j
\end{align*}\tag{A1.7} a α u = x n − i = 1 , i = j ∑ M z ni u i − i = M + 1 ∑ D b i u i = z nj u j ( A1.7 )
注意一点,j j j 项是我们希望求得降维后用以表示降维数据的基向量,因此它一定在 M M M 维中。我们结合 A3.2, A3.4 得到如下解。
∂ ∂ z n j = − 2 u j T ( x n − ∑ i = 1 , i ≠ j M z n i u i − ∑ i = M + 1 D b i u i − z n j u j ) = − 2 u j T x n + 2 u j T ∑ i = 1 , i ≠ j M z n i u i + 2 u j T ∑ i = M + 1 D b i u i + 2 u j T z n j u j = − 2 u j T x n + 2 u j T z n j u j (A1.8) \begin{align*}
\frac{\partial}{\partial z_{nj}}&=-2\mathbf{u}_j^T\left(\mathbf{x}_n-\sum_{i=1,i\ne j}^M z_{ni}\mathbf{u}_i-\sum_{i=M+1}^D b_i\mathbf{u}_i-z_{nj}\mathbf{u}_j\right)\\\\
&=-2\mathbf{u}_j^T\mathbf{x}_n+2\mathbf{u}_j^T\sum_{i=1,i\ne j}^M z_{ni}\mathbf{u}_i+2\mathbf{u}_j^T\sum_{i=M+1}^D b_i\mathbf{u}_i+2\mathbf{u}_j^Tz_{nj}\mathbf{u}_j\\\\
&=-2\mathbf{u}_j^T\mathbf{x}_n+2\mathbf{u}_j^Tz_{nj}\mathbf{u}_j
\end{align*}\tag{A1.8} ∂ z nj ∂ = − 2 u j T x n − i = 1 , i = j ∑ M z ni u i − i = M + 1 ∑ D b i u i − z nj u j = − 2 u j T x n + 2 u j T i = 1 , i = j ∑ M z ni u i + 2 u j T i = M + 1 ∑ D b i u i + 2 u j T z nj u j = − 2 u j T x n + 2 u j T z nj u j ( A1.8 )
中间两个求和项由正交向量内积为零而消去了,此时令导数为零就可以得到
∂ ∂ z n j = − 2 u j T x n + 2 u j T z n j u j = 0 z n j = u j T x n = x n T u j (A1.9) \begin{align*}
\frac{\partial}{\partial z_{nj}}&=-2\mathbf{u}_j^T\mathbf{x}_n+2\mathbf{u}_j^Tz_{nj}\mathbf{u}_j=0\\\\
z_{nj}&=\mathbf{u}_j^T\mathbf{x}_n=\mathbf{x}_n^T\mathbf{u}_j
\end{align*}\tag{A1.9} ∂ z nj ∂ z nj = − 2 u j T x n + 2 u j T z nj u j = 0 = u j T x n = x n T u j ( A1.9 )
结合公式 A3.2 和 A3.3,我们将 b j b_j b j 项单独拆出来。
a = ( x n − ∑ i = 1 M z n i u i − ∑ i = M + 1 , i ≠ j D b i u i ) α u = b j u j (A1.10) \begin{align*}
\mathbf{a}&=\left(\mathbf{x}_n-\sum_{i=1}^M z_{ni}\mathbf{u}_i-\sum_{i=M+1,i\ne j}^D b_i\mathbf{u}_i \right)\\\\
\alpha\mathbf{u}&=b_{j}\mathbf{u}_j
\end{align*}\tag{A1.10} a α u = x n − i = 1 ∑ M z ni u i − i = M + 1 , i = j ∑ D b i u i = b j u j ( A1.10 )
对 b j b_j b j 求偏导的过程如下:
∂ ∂ b j = − 2 u j T ( x n − ∑ i = 1 M z n i u i − ∑ i = M + 1 , i ≠ j D b i u i − b j u j ) = − 2 u j T x n + 2 u j T ∑ i = 1 M z n i u i + 2 u j T ∑ i = M + 1 , i ≠ j D b i u i + 2 u j T b j u j = − 2 u j T x n + 2 u j T b j u j = 0 b j = u j T x (A1.11) \begin{align*}
\frac{\partial}{\partial b_j}&=-2\mathbf{u}_j^T\left(\mathbf{x}_n-\sum_{i=1}^M z_{ni}\mathbf{u}_i-\sum_{i=M+1,i\ne j}^D b_i\mathbf{u}_i-b_j\mathbf{u}_j\right)\\\\
&=-2\mathbf{u}_j^T\mathbf{x}_n+2\mathbf{u}_j^T\sum_{i=1}^M z_{ni}\mathbf{u}_i+2\mathbf{u}_j^T\sum_{i=M+1,i\ne j}^D b_i\mathbf{u}_i+2\mathbf{u}_j^Tb_j\mathbf{u}_j\\\\
&=-2\mathbf{u}_j^T\mathbf{x}_n+2\mathbf{u}_j^Tb_j\mathbf{u}_j=0\\\\
b_j&=\mathbf{u}_j^T\mathbf{x}
\end{align*}\tag{A1.11} ∂ b j ∂ b j = − 2 u j T x n − i = 1 ∑ M z ni u i − i = M + 1 , i = j ∑ D b i u i − b j u j = − 2 u j T x n + 2 u j T i = 1 ∑ M z ni u i + 2 u j T i = M + 1 , i = j ∑ D b i u i + 2 u j T b j u j = − 2 u j T x n + 2 u j T b j u j = 0 = u j T x ( A1.11 )
这里我们依然是对某个特定数据 x n \mathbf{x}_n x n 做的计算,但是 b j b_j b j 实际上对所有数据点都是一致的,因此我们可以将所有数据的求和加到公式中,得到
b j = 1 N ∑ n = 1 N u j T x n = u j T x ˉ = x ˉ T u j (A1.12) \begin{align*}
b_j&=\frac{1}{N}\sum_{n=1}^N\mathbf{u}_j^T\mathbf{x}_n=\mathbf{u}_j^T\bar{\mathbf{x}}=\bar{\mathbf{x}}^T\mathbf{u}_j
\end{align*}\tag{A1.12} b j = N 1 n = 1 ∑ N u j T x n = u j T x ˉ = x ˉ T u j ( A1.12 )
x n − x ~ n = ∑ i = 1 D ( x n T u i ) u i − ∑ i = 1 M ( x n T u i ) u i − ∑ i = M + 1 D ( x ˉ T u i ) u i = ∑ M + 1 D ( x n T u i ) u i − ∑ i = M + 1 D ( x ˉ T u i ) u i = ∑ i = M + 1 D { ( x n − x ˉ ) T u i } u i (A1.13) \begin{align*}
\mathbf{x}_n-\tilde{\mathbf{x}}_n&=\sum_{i=1}^D(\mathbf{x}_n^T\mathbf{u}_i)\mathbf{u}_i-\sum_{i=1}^M(\mathbf{x}_n^T\mathbf{u}_i)\mathbf{u}_i-\sum_{i=M+1}^D(\bar{\mathbf{x}}^T\mathbf{u}_i)\mathbf{u}_i\\\\
&=\sum_{M+1}^D(\mathbf{x}_n^T\mathbf{u}_i)\mathbf{u}_i-\sum_{i=M+1}^D(\bar{\mathbf{x}}^T\mathbf{u}_i)\mathbf{u}_i\\\\
&=\sum_{i=M+1}^D\{(\mathbf{x}_n-\bar{\mathbf{x}})^T\mathbf{u}_i\}\mathbf{u}_i
\end{align*}\tag{A1.13} x n − x ~ n = i = 1 ∑ D ( x n T u i ) u i − i = 1 ∑ M ( x n T u i ) u i − i = M + 1 ∑ D ( x ˉ T u i ) u i = M + 1 ∑ D ( x n T u i ) u i − i = M + 1 ∑ D ( x ˉ T u i ) u i = i = M + 1 ∑ D {( x n − x ˉ ) T u i } u i ( A1.13 )
J = 1 N ∑ n = 1 N ∥ x n − x ~ n ∥ 2 x n − x ~ n = ∑ i = M + 1 D a n i u i , where a n i = ( x n − x ˉ ) T u i ∥ x n − x ~ n ∥ 2 = ( ∑ i = M + 1 D a n i u i ) T ( ∑ j = M + 1 D a n j u j ) = ∑ i , j = M + 1 D a n i a n j u i T u j u i T u j = δ i j ∥ x n − x ~ n ∥ 2 = ∑ i = M + 1 D [ ( x n − x ˉ ) T u i ] 2 J = 1 N ∑ n = 1 N ∑ i = M + 1 D [ ( x n − x ˉ ) T u i ] 2 = 1 N ∑ i = M + 1 D ∑ n = 1 N [ ( x n − x ˉ ) T u i ] 2 S = 1 N ∑ n = 1 N ( x n − x ˉ ) ( x n − x ˉ ) T 1 N ∑ n = 1 N [ ( x n − x ˉ ) T u i ] 2 = 1 N ∑ n = 1 N u i T ( x n − x ˉ ) ( x n − x ˉ ) T u i = u i T ( 1 N ∑ n = 1 N ( x n − x ˉ ) ( x n − x ˉ ) T ) u i = u i T S u i J = ∑ i = M + 1 D u i T S u i (A1.14) \begin{align*}
J &= \frac{1}{N} \sum_{n=1}^N \|\mathbf{x}_n - \tilde{\mathbf{x}}_n\|^2\\\\
\mathbf{x}_n - \tilde{\mathbf{x}}_n &= \sum_{i=M+1}^D a_{ni} \mathbf{u}_i,
\quad \text{where } a_{ni} = (\mathbf{x}_n - \bar{\mathbf{x}})^T\mathbf{u}_i\\\\
\|\mathbf{x}_n - \tilde{\mathbf{x}}_n\|^2
&= \left( \sum_{i=M+1}^D a_{ni} \mathbf{u}_i \right)^T \left( \sum_{j=M+1}^D a_{nj} \mathbf{u}_j \right)
= \sum_{i,j=M+1}^D a_{ni} a_{nj} \mathbf{u}_i^T \mathbf{u}_j\\\\
\mathbf{u}_i^T \mathbf{u}_j& = \delta_{ij}\\\\
\|\mathbf{x}_n - \tilde{\mathbf{x}}_n\|^2 &= \sum_{i=M+1}^D [(\mathbf{x}_n - \bar{\mathbf{x}})^T \mathbf{u}_i]^2\\\\
J &= \frac{1}{N} \sum_{n=1}^N \sum_{i=M+1}^D [(\mathbf{x}_n - \bar{\mathbf{x}})^T \mathbf{u}_i]^2
= \frac{1}{N} \sum_{i=M+1}^D \sum_{n=1}^N [(\mathbf{x}_n - \bar{\mathbf{x}})^T \mathbf{u}_i]^2\\\\
S &= \frac{1}{N} \sum_{n=1}^N (\mathbf{x}_n - \bar{\mathbf{x}})(\mathbf{x}_n - \bar{\mathbf{x}})^T\\\\
\frac{1}{N} \sum_{n=1}^N [(\mathbf{x}_n - \bar{\mathbf{x}})^T \mathbf{u}_i]^2
&= \frac{1}{N} \sum_{n=1}^N \mathbf{u}_i^T (\mathbf{x}_n - \bar{\mathbf{x}})(\mathbf{x}_n - \bar{\mathbf{x}})^T \mathbf{u}_i
= \mathbf{u}_i^T \left( \frac{1}{N} \sum_{n=1}^N (\mathbf{x}_n - \bar{\mathbf{x}})(\mathbf{x}_n - \bar{\mathbf{x}})^T \right) \mathbf{u}_i
= \mathbf{u}_i^T S \mathbf{u}_i\\\\
J &= \sum_{i=M+1}^D \mathbf{u}_i^T S \mathbf{u}_i
\end{align*}\tag{A1.14} J x n − x ~ n ∥ x n − x ~ n ∥ 2 u i T u j ∥ x n − x ~ n ∥ 2 J S N 1 n = 1 ∑ N [( x n − x ˉ ) T u i ] 2 J = N 1 n = 1 ∑ N ∥ x n − x ~ n ∥ 2 = i = M + 1 ∑ D a ni u i , where a ni = ( x n − x ˉ ) T u i = ( i = M + 1 ∑ D a ni u i ) T j = M + 1 ∑ D a nj u j = i , j = M + 1 ∑ D a ni a nj u i T u j = δ ij = i = M + 1 ∑ D [( x n − x ˉ ) T u i ] 2 = N 1 n = 1 ∑ N i = M + 1 ∑ D [( x n − x ˉ ) T u i ] 2 = N 1 i = M + 1 ∑ D n = 1 ∑ N [( x n − x ˉ ) T u i ] 2 = N 1 n = 1 ∑ N ( x n − x ˉ ) ( x n − x ˉ ) T = N 1 n = 1 ∑ N u i T ( x n − x ˉ ) ( x n − x ˉ ) T u i = u i T ( N 1 n = 1 ∑ N ( x n − x ˉ ) ( x n − x ˉ ) T ) u i = u i T S u i = i = M + 1 ∑ D u i T S u i ( A1.14 )
Appendix A2
矩阵求逆引理公式
( A + U C V ) − 1 = A − 1 − A − 1 U ( C − 1 + V A − 1 U ) − 1 V A − 1 \begin{align*}
(A + UCV)^{-1} = A^{-1} - A^{-1}U(C^{-1} + VA^{-1}U)^{-1}VA^{-1}\tag{A2.1}
\end{align*} ( A + U C V ) − 1 = A − 1 − A − 1 U ( C − 1 + V A − 1 U ) − 1 V A − 1 ( A2.1 )
求解后验分布的定理
given p ( x ) = N ( x ∣ μ , Λ − 1 ) p ( y ∣ x ) = N ( y ∣ A x + b , L − 1 ) have p ( y ) = N ( y ∣ A μ + b , L − 1 + A Λ − 1 A T ) p ( x ∣ y ) = N ( x ∣ μ + Σ A T L ( y − A μ − b ) , Σ ) Σ = ( Λ + A T L A ) − 1 (A2.2) \begin{align*}
\text{given}\quad p(\mathbf{x})&=\mathcal{N}(\mathbf{x}\mid\boldsymbol{\mu},\Lambda^{-1})\\\\
p(\mathbf{y}\mid\mathbf{x})&=\mathcal{N}(\mathbf{y}\mid\mathbf{Ax}+\mathbf{b},\mathbf{L}^{-1})\\\\
\text{have}\quad p(\mathbf{y})&=\mathcal{N}(\mathbf{y}|\mathbf{A}\boldsymbol{\mu}+\mathbf{b},\mathbf{L}^{-1}+\mathbf{A}\Lambda^{-1}\mathbf{A}^\text{T})\\\\
p(\mathbf{x} \mid \mathbf{y}) &= \mathcal{N}(\mathbf{x} \mid \boldsymbol{\mu} + \Sigma \mathbf{A}^T \mathbf{L} (\mathbf{y} - \mathbf{A}\boldsymbol{\mu} - \mathbf{b}), \Sigma)\\\\
\Sigma &= (\Lambda + \mathbf{A}^T \mathbf{L} \mathbf{A})^{-1}
\end{align*}\tag{A2.2} given p ( x ) p ( y ∣ x ) have p ( y ) p ( x ∣ y ) Σ = N ( x ∣ μ , Λ − 1 ) = N ( y ∣ Ax + b , L − 1 ) = N ( y ∣ A μ + b , L − 1 + A Λ − 1 A T ) = N ( x ∣ μ + Σ A T L ( y − A μ − b ) , Σ ) = ( Λ + A T LA ) − 1 ( A2.2 )
首先给出对向量偏导的一个公式
∂ ∂ μ [ ( x n − μ ) T C − 1 ( x n − μ ) ] = − 2 C − 1 ( x n − μ ) (A2.3) \begin{align*}
\frac{\partial}{\partial \boldsymbol{\mu}} [(\mathbf{x}_n - \boldsymbol{\mu})^T \mathbf{C}^{-1} (\mathbf{x}_n - \boldsymbol{\mu})] = -2\mathbf{C}^{-1}(\mathbf{x}_n - \boldsymbol{\mu})
\end{align*}\tag{A2.3} ∂ μ ∂ [( x n − μ ) T C − 1 ( x n − μ )] = − 2 C − 1 ( x n − μ ) ( A2.3 )
∂ ∂ μ ln p ( X ∣ μ , W , σ 2 ) = ∂ ∂ μ ( − 1 2 ∑ n = 1 N ( x n − μ ) T C − 1 ( x n − μ ) ) = ∑ n = 1 N C − 1 ( x n − μ ) = 0 ∑ n = 1 N C − 1 ( x n − μ ) = C − 1 ∑ n = 1 N ( x n − μ ) = 0 μ = 1 N ∑ n = 1 N x n = x ˉ (A2.4) \begin{align*}
\frac{\partial}{\partial \boldsymbol{\mu}} \ln p(\mathbf{X} \mid \boldsymbol{\mu}, \mathbf{W}, \sigma^2)&=
\frac{\partial}{\partial \boldsymbol{\mu}} \left(-\frac{1}{2} \sum_{n=1}^{N} (\mathbf{x}_n - \boldsymbol{\mu})^T \mathbf{C}^{-1} (\mathbf{x}_n - \boldsymbol{\mu})\right) = \sum_{n=1}^{N} \mathbf{C}^{-1} (\mathbf{x}_n - \boldsymbol{\mu})=0\\\\
\sum_{n=1}^{N} \mathbf{C}^{-1} (\mathbf{x}_n - \boldsymbol{\mu})&=\mathbf{C}^{-1}\sum_{n=1}^{N}(\mathbf{x}_n -\boldsymbol{\mu})=0\\\\
\boldsymbol{\mu}&= \frac{1}{N}\sum_{n=1}^N\mathbf{x}_n=\bar{\mathbf{x}}
\end{align*}\tag{A2.4} ∂ μ ∂ ln p ( X ∣ μ , W , σ 2 ) n = 1 ∑ N C − 1 ( x n − μ ) μ = ∂ μ ∂ ( − 2 1 n = 1 ∑ N ( x n − μ ) T C − 1 ( x n − μ ) ) = n = 1 ∑ N C − 1 ( x n − μ ) = 0 = C − 1 n = 1 ∑ N ( x n − μ ) = 0 = N 1 n = 1 ∑ N x n = x ˉ ( A2.4 )
Appendix A3
完整数据对数似然
将联合概率 p ( x n , z n ∣ μ , W , σ 2 ) p(\mathbf{x}_n, \mathbf{z}_n | \boldsymbol{\mu}, \mathbf{W}, \sigma^2) p ( x n , z n ∣ μ , W , σ 2 ) 写成条件概率和先验的乘积:
ln p ( x n , z n ∣ μ , W , σ 2 ) = ln p ( x n ∣ z n , μ , W , σ 2 ) + ln p ( z n ) \ln p(\mathbf{x}_n, \mathbf{z}_n | \boldsymbol{\mu}, \mathbf{W}, \sigma^2) = \ln p(\mathbf{x}_n | \mathbf{z}_n, \boldsymbol{\mu}, \mathbf{W}, \sigma^2) + \ln p(\mathbf{z}_n) ln p ( x n , z n ∣ μ , W , σ 2 ) = ln p ( x n ∣ z n , μ , W , σ 2 ) + ln p ( z n )
条件概率
条件概率:
p ( x n ∣ z n , μ , W , σ 2 ) = 1 ( 2 π σ 2 ) D / 2 exp ( − 1 2 σ 2 ∥ x n − μ − W z n ∥ 2 ) p(\mathbf{x}_n | \mathbf{z}_n, \boldsymbol{\mu}, \mathbf{W}, \sigma^2) = \frac{1}{(2\pi\sigma^2)^{D/2}} \exp\left(-\frac{1}{2\sigma^2}\|\mathbf{x}_n - \boldsymbol{\mu} - \mathbf{Wz}_n\|^2\right) p ( x n ∣ z n , μ , W , σ 2 ) = ( 2 π σ 2 ) D /2 1 exp ( − 2 σ 2 1 ∥ x n − μ − Wz n ∥ 2 )
对数形式:
ln p ( x n ∣ z n , μ , W , σ 2 ) = − D 2 ln ( 2 π σ 2 ) − 1 2 σ 2 ( x n − μ − W z n ) T ( x n − μ − W z n ) \ln p(\mathbf{x}_n | \mathbf{z}_n, \boldsymbol{\mu}, \mathbf{W}, \sigma^2) = -\frac{D}{2} \ln(2\pi\sigma^2) - \frac{1}{2\sigma^2} (\mathbf{x}_n - \boldsymbol{\mu} - \mathbf{Wz}_n)^T (\mathbf{x}_n - \boldsymbol{\mu} - \mathbf{Wz}_n) ln p ( x n ∣ z n , μ , W , σ 2 ) = − 2 D ln ( 2 π σ 2 ) − 2 σ 2 1 ( x n − μ − Wz n ) T ( x n − μ − Wz n )
展开平方项:
∥ x n − μ − W z n ∥ 2 = ( x n − μ ) T ( x n − μ ) − 2 ( x n − μ ) T W z n + z n T W T W z n \|\mathbf{x}_n - \boldsymbol{\mu} - \mathbf{Wz}_n\|^2 = (\mathbf{x}_n - \boldsymbol{\mu})^T(\mathbf{x}_n - \boldsymbol{\mu}) - 2(\mathbf{x}_n - \boldsymbol{\mu})^T \mathbf{Wz}_n + \mathbf{z}_n^T \mathbf{W}^T \mathbf{Wz}_n ∥ x n − μ − Wz n ∥ 2 = ( x n − μ ) T ( x n − μ ) − 2 ( x n − μ ) T Wz n + z n T W T Wz n
放入对数形式:
ln p ( x n ∣ z n , μ , W , σ 2 ) = − D 2 ln ( 2 π σ 2 ) − 1 2 σ 2 ( ( x n − μ ) T ( x n − μ ) − 2 ( x n − μ ) T W z n + z n T W T W z n ) \ln p(\mathbf{x}_n | \mathbf{z}_n, \boldsymbol{\mu}, \mathbf{W}, \sigma^2) = -\frac{D}{2} \ln(2\pi\sigma^2) - \frac{1}{2\sigma^2} ((\mathbf{x}_n - \boldsymbol{\mu})^T(\mathbf{x}_n - \boldsymbol{\mu}) - 2(\mathbf{x}_n - \boldsymbol{\mu})^T \mathbf{Wz}_n + \mathbf{z}_n^T \mathbf{W}^T \mathbf{Wz}_n) ln p ( x n ∣ z n , μ , W , σ 2 ) = − 2 D ln ( 2 π σ 2 ) − 2 σ 2 1 (( x n − μ ) T ( x n − μ ) − 2 ( x n − μ ) T Wz n + z n T W T Wz n )
先验概率
潜在变量的先验概率:
p ( z n ) = 1 ( 2 π ) M / 2 exp ( − 1 2 z n T z n ) p(\mathbf{z}_n) = \frac{1}{(2\pi)^{M/2}} \exp\left(-\frac{1}{2} \mathbf{z}_n^T \mathbf{z}_n \right) p ( z n ) = ( 2 π ) M /2 1 exp ( − 2 1 z n T z n )
对数形式:
ln p ( z n ) = − M 2 ln ( 2 π ) − 1 2 z n T z n \ln p(\mathbf{z}_n) = -\frac{M}{2} \ln(2\pi) - \frac{1}{2} \mathbf{z}_n^T \mathbf{z}_n ln p ( z n ) = − 2 M ln ( 2 π ) − 2 1 z n T z n
期望计算
我们需要计算:
E [ ln p ( x n , z n ∣ μ , W , σ 2 ) ] = E [ ln p ( x n ∣ z n , μ , W , σ 2 ) ] + E [ ln p ( z n ) ] E[\ln p(\mathbf{x}_n, \mathbf{z}_n | \boldsymbol{\mu}, \mathbf{W}, \sigma^2)] = E[\ln p(\mathbf{x}_n | \mathbf{z}_n, \boldsymbol{\mu}, \mathbf{W}, \sigma^2)] + E[\ln p(\mathbf{z}_n)] E [ ln p ( x n , z n ∣ μ , W , σ 2 )] = E [ ln p ( x n ∣ z n , μ , W , σ 2 )] + E [ ln p ( z n )]
计算每项的期望
条件概率对数的期望:
E [ ln p ( x n ∣ z n , μ , W , σ 2 ) ] = − D 2 ln ( 2 π σ 2 ) − 1 2 σ 2 ( ∥ x n − μ ∥ 2 − 2 ( x n − μ ) T W E [ z n ] + Tr ( W T W E [ z n z n T ] ) ) E[\ln p(\mathbf{x}_n | \mathbf{z}_n, \boldsymbol{\mu}, \mathbf{W}, \sigma^2)] = -\frac{D}{2} \ln(2\pi\sigma^2) - \frac{1}{2\sigma^2} \left(\|\mathbf{x}_n - \boldsymbol{\mu}\|^2 - 2(\mathbf{x}_n - \boldsymbol{\mu})^T \mathbf{W}E[\mathbf{z}_n] + \text{Tr}(\mathbf{W}^T \mathbf{W} E[\mathbf{z}_n\mathbf{z}_n^T]) \right) E [ ln p ( x n ∣ z n , μ , W , σ 2 )] = − 2 D ln ( 2 π σ 2 ) − 2 σ 2 1 ( ∥ x n − μ ∥ 2 − 2 ( x n − μ ) T W E [ z n ] + Tr ( W T W E [ z n z n T ]) )
先验对数的期望:
E [ ln p ( z n ) ] = − M 2 ln ( 2 π ) − 1 2 Tr ( E [ z n z n T ] ) E[\ln p(\mathbf{z}_n)] = -\frac{M}{2} \ln(2\pi) - \frac{1}{2} \text{Tr}(E[\mathbf{z}_n\mathbf{z}_n^T]) E [ ln p ( z n )] = − 2 M ln ( 2 π ) − 2 1 Tr ( E [ z n z n T ])
总和公式
最后,把它们结合起来得到:
E [ ln p ( x n , z n ∣ μ , W , σ 2 ) ] = − D 2 ln ( 2 π σ 2 ) − M 2 ln ( 2 π ) − 1 2 σ 2 ∥ x n − μ ∥ 2 + 1 σ 2 E [ z n ] T W T ( x n − μ ) − 1 2 σ 2 Tr ( W T W E [ z n z n T ] ) − 1 2 Tr ( E [ z n z n T ] ) E[\ln p(\mathbf{x}_n, \mathbf{z}_n | \boldsymbol{\mu}, \mathbf{W}, \sigma^2)] = -\frac{D}{2} \ln(2\pi\sigma^2) - \frac{M}{2} \ln(2\pi) - \frac{1}{2\sigma^2}\|\mathbf{x}_n - \boldsymbol{\mu}\|^2 + \frac{1}{\sigma^2} E[\mathbf{z}_n]^T \mathbf{W}^T (\mathbf{x}_n - \boldsymbol{\mu}) - \frac{1}{2\sigma^2} \text{Tr}(\mathbf{W}^T \mathbf{W} E[\mathbf{z}_n \mathbf{z}_n^T]) - \frac{1}{2} \text{Tr}(E[\mathbf{z}_n \mathbf{z}_n^T]) E [ ln p ( x n , z n ∣ μ , W , σ 2 )] = − 2 D ln ( 2 π σ 2 ) − 2 M ln ( 2 π ) − 2 σ 2 1 ∥ x n − μ ∥ 2 + σ 2 1 E [ z n ] T W T ( x n − μ ) − 2 σ 2 1 Tr ( W T W E [ z n z n T ]) − 2 1 Tr ( E [ z n z n T ])
对所有数据点 (n = 1) 到 (N) 求和,我们得到完整的期望:
E [ ln p ( X , Z ∣ μ , W , σ 2 ) ] = ∑ n = 1 N { − D 2 ln ( 2 π σ 2 ) − M 2 ln ( 2 π ) − 1 2 σ 2 ∥ x n − μ ∥ 2 + 1 σ 2 E [ z n ] T W T ( x n − μ ) − 1 2 σ 2 Tr ( W T W E [ z n z n T ] ) − 1 2 Tr ( E [ z n z n T ] ) } E\left[\ln p(\mathbf{X}, \mathbf{Z} | \boldsymbol{\mu}, \mathbf{W}, \sigma^2)\right] = \sum_{n=1}^{N} \left\{-\frac{D}{2} \ln(2\pi\sigma^2) - \frac{M}{2} \ln(2\pi) - \frac{1}{2\sigma^2} \|\mathbf{x}_n - \boldsymbol{\mu}\|^2 + \frac{1}{\sigma^2} E[\mathbf{z}_n]^T \mathbf{W}^T (\mathbf{x}_n - \boldsymbol{\mu}) - \frac{1}{2\sigma^2} \text{Tr}(\mathbf{W}^T \mathbf{W} E[\mathbf{z}_n \mathbf{z}_n^T]) - \frac{1}{2} \text{Tr}(E[\mathbf{z}_n \mathbf{z}_n^T]) \right\} E [ ln p ( X , Z ∣ μ , W , σ 2 ) ] = n = 1 ∑ N { − 2 D ln ( 2 π σ 2 ) − 2 M ln ( 2 π ) − 2 σ 2 1 ∥ x n − μ ∥ 2 + σ 2 1 E [ z n ] T W T ( x n − μ ) − 2 σ 2 1 Tr ( W T W E [ z n z n T ]) − 2 1 Tr ( E [ z n z n T ]) }
Appendix A4
在变量变换问题中,对于从随机变量 ( Z ) 到 ( X ) 的变换,假设 ( Z ) 的概率密度函数为 ( p_Z(z) ),我们希望计算 ( X ) 的密度 ( p_X(x) ),其中 ( x = g(z) )。变量变换公式可以写作:
p X ( x ) = p Z ( z ( x ) ) ⋅ ∣ det ∂ z ∂ x ∣ p_X(x) = p_Z(z(x)) \cdot \left|\det \frac{\partial z}{\partial x}\right| p X ( x ) = p Z ( z ( x )) ⋅ det ∂ x ∂ z
其中:
p X ( x ) p_X(x) p X ( x ) 是 X X X 的概率密度。
p Z ( z ( x ) ) p_Z(z(x)) p Z ( z ( x )) 是从 X X X 的反向变换得到的Z Z Z 的概率密度。
det ∂ z ∂ x \det \frac{\partial z}{\partial x} det ∂ x ∂ z 是雅可比行列式,描述变化过程中体积元素的缩放。
∣ ⋅ ∣ |\cdot| ∣ ⋅ ∣ 表示取绝对值,保证密度函数的非负性。
解释
反向变换 z ( x ) z(x) z ( x ) : 为了找到 p Z ( z ) p_Z(z) p Z ( z ) ,我们需要知道 x x x 在原始变量z z z 下的位置。
雅可比行列式 : det ∂ z ∂ x \det \frac{\partial z}{\partial x} det ∂ x ∂ z 度量了从 z z z 空间到 x x x 空间的缩放效应,是多变量微积分中度量非线性变换时的一种工具。
绝对值的意义 : 确保任何转换后得到的概率密度函数都是非负的,符合概率密度的定义。