梯度下降隐式执行期望最大化,解释神经网络中的概率行为
Gradient Descent as Implicit EM in Distance-Based Neural Models
- 用距离/能量的log-sum-exp目标函数,梯度直接对应后验责任
- 训练中梯度更新等价于隐式EM算法,无需额外计算责任值
- 统一了无监督聚类、注意力机制和分类任务的训练原理
以距离或能量为基础的神经网络目标函数,其梯度与对应分量的后验责任完全对应:∂L/∂dj = -rj。这一代数恒等式源于Fisher身份,仅需可微性,且标准训练目标天然满足。这意味着梯度下降在这些目标上自动执行广义期望最大化(EM),责任项由梯度生成而非额外计算。该结果统一了三种学习范式:无监督混合建模(责任全隐含)、注意力机制(责任依赖查询)和交叉熵分类(责任受监督约束)。本文聚焦训练过程:Transformer中近期观察到的责任加权梯度动态,源自目标函数的几何结构;推理时的上下文贝叶斯计算是训练动态的终点,而非每步的内在内容。
原文摘要 · Abstract (English)
Neural networks trained with standard objectives exhibit behaviors characteristic of probabilistic inference: soft clustering, prototype specialization, and Bayesian uncertainty tracking. These phenomena appear across architectures -- in attention mechanisms, classification heads, and energy-based models -- yet existing explanations often rely on loose analogies to mixture models or post-hoc architectural interpretation. We provide a direct explanation. For any objective with log-sum-exp structure over distances or energies, the gradient with respect to each distance is exactly the negative posterior responsibility of the corresponding component: $\partial L / \partial d_j = -r_j$. The identity is algebraic, requiring only differentiability; it is a specialization of Fisher's identity, and its significance here is its address: standard neural objectives instantiate it without modification. The consequence is that gradient descent on such objectives performs generalized expectation-maximization implicitly, with responsibilities arising as gradients to be applied rather than auxiliary variables to be computed. This result unifies three regimes of learning: unsupervised mixture modeling, where responsibilities are fully latent; attention, where responsibilities are conditioned on queries; and cross-entropy classification, where supervision clamps responsibilities to targets. Our claims live at training time: the responsibility-weighted gradient dynamics recently documented in transformers follow from the objective's geometry. The in-context Bayesian computation that trained transformers perform at inference time is the endpoint of these dynamics, not their per-step content.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。