高斯混合模型|机器学习推导系列（十三）

一、概述

以一维数据为例，我们可以看到下图通过将多个单一的高斯模型加权叠加到一起就可以获得一个高斯混合模型，这个混合模型显然具备比单个高斯模型更强的拟合能力：

再举一个二维数据的例子，在下图中可以看到有两个数据密集区域，对应的概率分布也就会有两个峰。高斯混合模型可以看做生成模型，其数据生成过程可以认为先选择一个高斯分布，再从被选择的高斯分布中生成数据：

综合上述两种描述，我们可以从两种角度来描述高斯混合模型：

1. 几何角度：加权平均

可以认为高斯混合模型是将多个高斯分布加权平均而成的模型：

$$p(x)=\sum _{k=1}^K{\alpha }_{k}N({\mu }_k,{\Sigma }_k),\sum _{k=1}^K{\alpha }_k=1$$

2. 混合模型（或者生成模型）角度

可以认为高斯混合模型是一种含有隐变量的生成模型：

$x$:observed variable
$z$:latent variable

$z$是隐变量，表示对应的样本$x$属于哪一个高斯分布，其概率分为如下表：

$z$	$C_1$	$C_2$	$\cdots$	$C_k$
$p$	$p_1$	$p_2$	$\cdots$	$p_k$

可以认为这里的概率$p_k$就是几何角度加权平均中权重，两种角度的解释其实是一个意思。

我们可以画出高斯混合模型的概率图：

实心点代表模型的参数，右下角的$N$代表样本个数。

二、尝试用极大似然估计来求解

$X$:observed data$\to X=(x_1,x_2,\cdots ,x_N)$
$(X,Z)$:comlete data
$\theta$:parameter → θ = { p 1 , p 2 , ⋯   , p k , μ 1 , μ 2 , ⋯   , μ k , Σ 1 , Σ 2 , ⋯   , Σ k } , ∑ i = 1 K p k = 1 \rightarrow \theta =\left \{p_{1},p_{2},\cdots ,p_{k},\mu _{1},\mu _{2},\cdots ,\mu _{k},\Sigma _{1},\Sigma _{2},\cdots ,\Sigma _{k}\right \},\sum_{i=1}^{K}p_{k}=1 →θ={p1​,p2​,⋯,pk​,μ1​,μ2​,⋯,μk​,Σ1​,Σ2​,⋯,Σk​},∑i=1K​pk​=1

以上为我们的数据以及需要求解的参数。接下来我们表示一下概率$p(x)$:

$$\begin{gathered} p(x)=\sum _{z}p(x,z) \\ =\sum _{k=1}^{K}p(x,z=C_k) \\ =\sum _{k=1}^{K}p(z=C_k)\cdot p(x|z=C_k) \\ =\sum _{k=1}^K{p}_k\cdot N(x|{\mu }_k,{\Sigma }_k) \end{gathered}$$

然后我们使用极大似然估计法求解这个参数估计问题。首先告知结论：极大似然估计法无法求解含有隐变量的参数估计问题，或者说不能得到解析解。接下来来看为什么不能用极大似然估计法来求解：

$$\begin{gathered} {\hat{\theta }}_{MLE}=\underset{\theta }{\mathrm{argmax}}logp(X) \\ =\underset{\theta }{\mathrm{argmax}}log\prod _{i=1}^{N}p(x_i) \\ =\underset{\theta }{\mathrm{argmax}}\sum _{i=1}^{N}logp(x_i) \\ =\underset{\theta }{\mathrm{argmax}}\sum _{i=1}^{N}log\sum _{k=1}^K{p}_k\cdot N(x_i|{\mu }_k,{\Sigma }_k) \end{gathered}$$

极大似然估计法不能得到解析解的原因为$log$函数内部出现了求和符号。当然我们可以使用梯度下降法来进行求解，但是对于含有隐变量的模型来说使用EM算法是更为合适的。

三、使用EM算法求解

由于使用EM算法需要用到联合概率$p(x,z)$和后验$p(z|x)$，所有我们首先写出这两个概率的表示：

$$\begin{gathered} p(x,z)=p(z)p(x|z)=p_z\cdot N(x|{\mu }_z,{\Sigma }_z) \\ p(z|x)=\frac{p(x,z)}{p(x)}=\frac{p_z\cdot N(x|{\mu }_z,{\Sigma }_z)}{{\sum }_{k=1}^K{p}_k\cdot N(x|{\mu }_k,{\Sigma }_k)} \end{gathered}$$

1. E step

$$\begin{gathered} Q(\theta ,{\theta }^t)=\sum _{Z}logp(X,Z|\theta )\cdot p(Z|X,{\theta }^t) \\ =\sum _{z_1{z}_2\cdots z_N}log\prod _{i=1}^{N}p(x_i,z_i|\theta )\cdot \prod _{i=1}^{N}p(z_i|x_i,{\theta }^t) \\ =\sum _{z_1{z}_2\cdots z_N}\sum _{i=1}^{N}logp(x_i,z_i|\theta )\cdot \prod _{i=1}^{N}p(z_i|x_i,{\theta }^t) \\ =\sum _{z_1{z}_2\cdots z_N}[logp(x_1,z_1|\theta )+logp(x_2,z_2|\theta )+\cdots +logp(x_i,z_i|\theta )]\cdot \prod _{i=1}^{N}p(z_i|x_i,{\theta }^t) \end{gathered}$$

对于上式展开的每一项，我们可以进行化简：

$$\begin{gathered} \sum _{z_1{z}_2\cdots z_N}logp(x_1,z_1|\theta )\prod _{i=1}^{N}p(z_i|x_i,{\theta }^t) \\ =\sum _{z_1{z}_2\cdots z_N}logp(x_1,z_1|\theta )\cdot p(z_1|x_1,{\theta }^t)\prod _{i=2}^{N}p(z_i|x_i,{\theta }^t) \\ =\sum _{z_1}logp(x_1,z_1|\theta )\cdot p(z_1|x_1,{\theta }^t)\sum _{z_2{z}_3\cdots z_N}\prod _{i=2}^{N}p(z_i|x_i,{\theta }^t) \\ =\sum _{z_1}logp(x_1,z_1|\theta )\cdot p(z_1|x_1,{\theta }^t)\sum _{z_2{z}_3\cdots z_N}p(z_2|x_2,{\theta }^t)p(z_3|x_3,{\theta }^t)\cdots p(z_N|x_N,{\theta }^t) \\ =\sum _{z_1}logp(x_1,z_1|\theta )\cdot p(z_1|x_1,{\theta }^t)\underset{=1}{\underbrace{\sum _{z_2}p(z_2|x_2,{\theta }^t)}}\underset{=1}{\underbrace{\sum _{z_3}p(z_3|x_3,{\theta }^t)}}\cdots \underset{=1}{\underbrace{\sum _{z_N}p(z_N|x_N,{\theta }^t)}} \\ =\sum _{z_1}logp(x_1,z_1|\theta )\cdot p(z_1|x_1,{\theta }^t) \\ 同样的我们可以得到\sum _{z_1{z}_2\cdots z_N}logp(x_1,z_1|\theta )\prod _{i=1}^{N}p(z_i|x_i,{\theta }^t)=\sum _{z_i}logp(x_i,z_i|\theta )p(z_i|x_i,{\theta }^t) \end{gathered}$$

继续对$Q(\theta ,{\theta }^t)$进行化简可以得到：

$$\begin{gathered} Q(\theta ,{\theta }^t)=\sum _{z_1}logp(x_1,z_1|\theta )\cdot p(z_1|x_1,{\theta }^t)+\cdots +\sum _{z_i}logp(x_i,z_i|\theta )\cdot p(z_i|x_i,{\theta }^t) \\ =\sum _{i=1}^N\sum _{z_i}logp(x_i,z_i|\theta )\cdot p(z_i|x_i,{\theta }^t) \\ =\sum _{i=1}^N\sum _{z_i}log[p_{z_i}\cdot N(x_i|{\mu }_{z_i},{\Sigma }_{z_i})]\cdot \frac{p_{z_i}^t\cdot N(x_i|{\mu }_{z_i}^t,{\Sigma }_{z_i}^t)}{{\sum }_{k=1}^K{p}_k^t\cdot N(x_i|{\mu }_k^t,{\Sigma }_k^t)} \\ (\frac{p_{z_i}^t\cdot N(x_i|{\mu }_{z_i}^t,{\Sigma }_{z_i}^t)}{{\sum }_{k=1}^K{p}_k^t\cdot N(x_i|{\mu }_k^t,{\Sigma }_k^t)}与\theta 无关，暂时写作p(z_i|x_i,{\theta }^t)) \\ =\sum _{i=1}^N\sum _{z_i}log[p_{z_i}\cdot N(x_i|{\mu }_{z_i},{\Sigma }_{z_i})]\cdot p(z_i|x_i,{\theta }^t) \\ =\sum _{z_i}\sum _{i=1}^{N}log[p_{z_i}\cdot N(x_i|{\mu }_{z_i},{\Sigma }_{z_i})]\cdot p(z_i|x_i,{\theta }^t) \\ =\sum _{k=1}^K\sum _{i=1}^{N}log[p_k\cdot N(x_i|{\mu }_k,{\Sigma }_k)]\cdot p(z_i=C_k|x_i,{\theta }^t) \\ =\sum _{k=1}^K\sum _{i=1}^N[log{p}_k+logN(x_i|{\mu }_k,{\Sigma }_k)]\cdot p(z_i=C_k|x_i,{\theta }^t) \end{gathered}$$

2. M step

EM算法的迭代公式为：

$${\theta }^{t+1}=\underset{\theta }{\mathrm{argmax}}Q(\theta ,{\theta }^t)$$

下面以求解$p^{t+1}=(p_1^{t+1},p_2^{t+1},\cdots ,p_K^{t+1}{)}^T$为例，来看如何进行迭代求解，以下是求解的迭代公式：

$$p_k^{t+1}=\underset{p_k}{\mathrm{argmax}}\sum _{k=1}^K\sum _{i=1}^{N}log{p}_k\cdot p(z_i=C_k|x_i,{\theta }^t),s.t.\sum _{k=1}^K{p}_k=1$$

于是可以转化为以下约束优化问题：

$$\{\begin{array}{c}\underset{p}{\min }-{\sum }_{k=1}^K{\sum }_{i=1}^{N}log{p}_k\cdot p(z_i=C_k|x_i,{\theta }^t)\\ s.t.{\sum }_{k=1}^K{p}_k=1\end{array}$$

然后使用拉格朗日乘子法进行求解：

$$\begin{gathered} L(p,\lambda )=-\sum _{k=1}^K\sum _{i=1}^{N}log{p}_k\cdot p(z_i=C_k|x_i,{\theta }^t)+\lambda (\sum _{k=1}^K{p}_k-1) \\ \frac{\partial L}{\partial p_k}=-\sum _{i=1}^N\frac{1}{p_k}\cdot p(z_i=C_k|x_i,{\theta }^t)+\lambda \stackrel{△}{=}0 \\ \Rightarrow -\sum _{i=1}^{N}p(z_i=C_k|x_i,{\theta }^t)+p_k^{t+1}\lambda =0 \\ \stackrel{k=1,2,\cdots ,K}{\Rightarrow }-\underset{=N}{\underbrace{\sum _{i=1}^N\underset{=1}{\underbrace{\sum _{k=1}^{K}p(z_i=C_k|x_i,{\theta }^t)}}}}+\underset{=1}{\underbrace{\sum _{k=1}^K{p}_k^{t+1}}}\lambda =0 \\ \Rightarrow -N+\lambda =0 \\ \Rightarrow \lambda =N \\ 代入-\sum _{i=1}^{N}p(z_i=C_k|x_i,{\theta }^t)+p_k^{t+1}\lambda =0得 \\ -\sum _{i=1}^{N}p(z_i=C_k|x_i,{\theta }^t)+p_k^{t+1}N=0 \\ \Rightarrow p_k^{t+1}=\frac{{\sum }_{i=1}^{N}p(z_i=C_k|x_i,{\theta }^t)}{N} \end{gathered}$$

这里以求解$p^{t+1}$为例展示了M step的求解过程，其他参数也按照极大化Q函数的思路就可以求解，求得一轮参数后要继续迭代求解直至收敛。