形式陈述
观测 固定,潜变量为 ,模型联合密度为 。本页的对数似然与 KL 散度均用自然对数。最大似然估计理路最大似然估计Maximum likelihood estimation · MLE在参数空间内选择使已观测数据似然达到上确界的参数估计方法。要最大化
当直接处理积分外的对数很难、给定 后却容易优化时,EM 从允许参数 出发重复两步:
- E 步: 求当前后验 ,并形成条件期望理路条件期望Conditional expectation以信息分组的加权平均建立条件期望直觉,再连接测度定义、最小均方预测、塔式性质和可计算反例。
- M 步: 选择 。只保证 的版本称广义 EM。
输入还应包括迭代上限和数值容差。每轮检查参数是否仍可行、后验是否可归一化、目标值是否有限;达到上限或似然增量小于给定容差时停止并报告当前参数、目标值及停止原因。小增量是数值停止规则,不是全局最优证书。
下述单调性要求 E 步确实使用当前后验,相关期望有限,且 M 步至少不降低该 。近似后验、蒙特卡洛噪声或不受控的数值优化不自动继承结论。
直觉
EM 不把缺失值填成一个最可能值后假装它已观测。E 步保留潜变量的整个当前分布;M 步在这个分布下平均完整数据的对数似然。比如混合模型中的一个观测可以以 和 的权重参与两个成分更新,而不是被永久扔进其中一个簇。
为什么似然不下降
将证据下界理路证据下界Evidence lower bound · ELBO · Variational lower bound将对数边缘似然分解为任意辅助分布下的可计算期望与到真实后验的非负 KL 缺口。用于固定参数的联合模型,记
这里 是待优化参数,无须对它设先验。由KL 散度理路KL 散度Kullback–Leibler divergence · Relative entropy同一可测空间上分布 P 相对于 Q 的对数 Radon–Nikodym 导数在 P 下的积分。分解,
E 步令缺口在旧参数处为零;M 步保持 不变,因熵项不含 ,提高 就提高 。因此
证明的关键是旧位置处下界贴住似然。任意近似 没有这个等号时,即使下界上升,原似然仍可能下降。
例子与边界
三个观测的一轮更新
拟合两成分高斯混合理路有限混合模型Finite mixture model · 有限混合分布先抽取有限个潜在成分之一,再由该成分生成观测,并通过边缘化得到观测分布的统计模型。,两方差固定为 。数据为 ,初值 。记 ,E 步给出
故 。第一成分的有效样本数为 。忽略与参数无关的项,M 步最大化
对均值求导、对权重在 上求导,分别得到加权平均和有效比例:
把新旧参数代回包含正态归一化常数的观测对数似然,得到 。均值和权重确实变了,而单调性比较的是同一个观测目标,不是两个不同轮次的 值。
计算成本
一般 EM 的成本取决于后验计算和 M 步优化,不能仅凭“两步交替”给出统一复杂度。对 个一维观测、 个方差固定的高斯成分,每轮计算全部责任度并累计加权和需 时间,执行 轮需 时间。保存整张责任度表需 空间;若每轮固定旧参数,逐个观测计算责任度并累计各成分的有效样本数与加权和,最后统一更新参数,则辅助空间可降为 。这里不计数据和模型本身的存储,也不把达到指定精度所需的轮数 当作已知常数。
单调不等于解决全部问题
若似然有有限上界,单调性只保证似然值收敛;参数序列收敛、极限为驻点仍需额外正则条件,更不能推出全局最大。对称初值可能使两个成分持续重合;不同初值可能落入不同局部解。
若高斯方差也自由更新,方差趋零可能使似然无界。此时似然不断上升恰可能是在走向退化。某成分有效样本数为零时,均值更新还会出现除零;实现必须报告退化或采用明示的约束、重启规则,不能悄悄赋一个任意均值。
推论与应用
EM 适合“完整数据容易,观测数据困难”的结构,潜变量可以是缺失数值,也可以是从未观测过的类别。若后验计算本身困难,可转向变分推断理路变分 Bayesian 推断Variational Bayesian inference · Variational inference · VI在可计算分布族内优化与真实后验的散度,以确定性或随机优化换取受限族中的快速后验近似。等近似路线,但应重新说明优化目标和保证。
判断一次实现是否遵循 EM,可保存每轮观测对数似然,并在小型离散模型上直接枚举潜状态核对 E 步。只画 曲线不足以验证单调性,因为 的条件参数每轮都在改变。多起点有助于发现不同局部解,却不构成全局保证。
自测
上述初值下,若把中间观测 硬分给第一成分,是否仍是同一次 EM 的 E 步?不是:正确条件概率是 。硬指派改变了辅助分布,旧位置下界贴合条件不再成立,所以必须另行分析所得算法。
事件只知发生在两个检查时刻之间时,潜变量可以取其所属的候选支持区间。Turnbull 估计理路Turnbull 区间删失估计Turnbull estimator用区间概率建立非参数似然,再以自洽质量更新估计区间删失分布,不伪造精确事件日。按当前区间质量分摊每个人的条件概率,再平均更新,提供一个可逐轮核验似然增加的非参数 EM 例子。
参考资料