用期望最大化框架简化大模型推理强化学习,效果更优且更稳定。
An Expectation-Maximization Perspective on Reinforcement Learning for LLM Reasoning
- 将强化学习建模为期望最大化问题,避免复杂权重修正和裁剪。
- 在GSM8K和MATH Hard上表现优于或媲美现有方法,优化更简单。
- 生成更简洁推理路径,具备分解、验证、反思等结构化能力。
强化学习已成为提升大语言模型推理能力的有效方法,如OpenAI的O1和DeepSeek-R1系统所示。然而,广泛使用的PPO和GRPO等算法依赖似然比修正与启发式裁剪,增加算法复杂性并可能引入额外方差与优化偏差。本文提出 extbf{EM Policy Gradient (EMPG)},一种简洁且理论严谨的框架,将推理轨迹上的强化学习建模为期望最大化(EM)问题。E步构建奖励加权的推理轨迹分布,M步通过最大化加权对数似然来优化语言模型。该框架导出一种无需轨迹级重要性权重或似然比裁剪的实用离策略策略梯度算法。在精确的E步与改进的M步下,EMPG保证期望奖励目标单调提升。在多个Qwen2.5基模型上对GSM8K和MATH Hard的实验表明,EMPG性能可媲美或超越GRPO,同时优化过程显著简化。此外,EMPG生成更简洁的推理路径,并表现出子问题分解、自我验证、反思与回溯等结构化推理行为。结果表明,EM视角为推理型语言模型的强化学习提供了有效的理论与算法基础。
原文摘要 · Abstract (English)
Reinforcement learning has emerged as a powerful approach for improving the reasoning capabilities of large language models, as demonstrated by systems such as OpenAI's O1~\cite{o1} and DeepSeek-R1~\cite{r1}. However, widely used algorithms such as PPO~\cite{ppo} and GRPO~\cite{grpo} rely on likelihood-ratio correction and heuristic clipping, which increase algorithmic complexity and may introduce additional variance and optimization bias. We present \textbf{EM Policy Gradient (EMPG)}, a simple and principled framework that formulates reinforcement learning over reasoning trajectories as an Expectation-Maximization (EM) problem~\cite{em, dempster1977em}. The E-step constructs a reward-weighted distribution over rationale trajectories, while the M-step improves the language model by maximizing their weighted log-likelihood. This formulation leads to a practical off-policy policy-gradient algorithm that directly optimizes the EM surrogate objective without trajectory-level importance weighting or likelihood-ratio clipping. Moreover, under exact E- and improving M-steps, EMPG guarantees monotonic improvement of the expected-reward objective. Experiments on GSM8K and MATH Hard with multiple Qwen2.5 base models show that EMPG achieves performance comparable to or better than GRPO while using a substantially simpler optimization procedure. EMPG also produces more concise reasoning trajectories and exhibits structured reasoning behaviors, including subproblem decomposition, self-verification, reflection, and backtracking. These results suggest that the EM perspective provides an effective theoretical and algorithmic foundation for reinforcement learning of reasoning language models.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。