“EM 算法将同一分解用于带待估参数的潜变量模型:精确 E 步使下界在旧参数处贴合,M 步提高该下界,因而提高观测似然。这项保证依赖贴合,不能从任意近似下界上升直接推出似然上升。”
形式陈述 ​
观测
当直接处理积分外的对数很难、给定
- E 步: 求当前后验
,并形成条件期望 - M 步: 选择
。只保证 的版本称广义 EM。
输入还应包括迭代上限和数值容差。每轮检查参数是否仍可行、后验是否可归一化、目标值是否有限;达到上限或似然增量小于给定容差时停止并报告当前参数、目标值及停止原因。小增量是数值停止规则,不是全局最优证书。
下述单调性要求 E 步确实使用当前后验,相关期望有限,且 M 步至少不降低该
直觉
EM 不把缺失值填成一个最可能值后假装它已观测。E 步保留潜变量的整个当前分布;M 步在这个分布下平均完整数据的对数似然。比如混合模型中的一个观测可以以
为什么似然不下降 ​
将证据下界用于固定参数的联合模型,记
这里
E 步令缺口在旧参数处为零;M 步保持
证明的关键是旧位置处下界贴住似然。任意近似
例子与边界
三个观测的一轮更新 ​
拟合两成分高斯混合,两方差固定为
故
对均值求导、对权重在
把新旧参数代回包含正态归一化常数的观测对数似然,得到
计算成本 ​
一般 EM 的成本取决于后验计算和 M 步优化,不能仅凭“两步交替”给出统一复杂度。对
单调不等于解决全部问题 ​
若似然有有限上界,单调性只保证似然值收敛;参数序列收敛、极限为驻点仍需额外正则条件,更不能推出全局最大。对称初值可能使两个成分持续重合;不同初值可能落入不同局部解。
若高斯方差也自由更新,方差趋零可能使似然无界。此时似然不断上升恰可能是在走向退化。某成分有效样本数为零时,均值更新还会出现除零;实现必须报告退化或采用明示的约束、重启规则,不能悄悄赋一个任意均值。
推论与应用
EM 适合“完整数据容易,观测数据困难”的结构,潜变量可以是缺失数值,也可以是从未观测过的类别。若后验计算本身困难,可转向变分推断等近似路线,但应重新说明优化目标和保证。
判断一次实现是否遵循 EM,可保存每轮观测对数似然,并在小型离散模型上直接枚举潜状态核对 E 步。只画
自测 ​
上述初值下,若把中间观测
参考资料
- A. P. Dempster, N. M. Laird, and D. B. Rubin, “Maximum Likelihood from Incomplete Data via the EM Algorithm”, Journal of the Royal Statistical Society, Series B 39(1), 1977,§3,EM 与广义 EM 的单调性。
- Christopher M. Bishop, Pattern Recognition and Machine Learning, Springer, 2006,§9.2.2 与 §9.4,高斯混合更新及 EM 的下界解释。