arXiv:2606.10184cs.LGcs.AI2026-06

用结构化丢弃让连续隐状态推理模型跑通强化学习。

Dropout-GRPO: Variational Stochasticity for Continuous Latent Reasoning

论文配图:Dropout-GRPO: Variational Stochasticity for Continuous Latent Reasoning
图 1 · 摘自论文原文
  • 用固定掩码在隐状态中引入结构化随机性,生成轨迹差异。
  • 在GSM8K上将基准模型准确率从27.29%提升至29.01%。
  • 适合希望微调连续推理大模型的研究者使用。

组相对策略优化(GRPO)依赖每组内K条轨迹的多样性;否则组均优势 $A^{(k)} = r^{(k)} - μ_r$ 会坍缩为零。这对像Coconut这类以连续隐状态递归推理的模型构成结构性挑战——因隐状态在给定参数与提示下本质确定,多条轨迹完全相同,导致GRPO无法推进。为此,我们提出通过结构化丢弃引入必要随机性:对每条轨迹,施加一个在所有隐状态步骤中保持不变的伯努利掩码,从而生成轨迹差异。该共享掩码将每条轨迹视为参数变分分布下的后验样本,使GRPO能优化贝叶斯模型平均策略的期望奖励。我们提供了理论支持,包括无偏性、方差缩减及隐梯度合理性,并通过实证验证。在GSM8K上,dropout-GRPO将Coconut基准的pass@1从27.29%提升至29.01%,证明了该方法在连续隐状态推理中应用GRPO的可行性。本工作为后训练连续推理大模型提供了一种实用且理论扎实的方法。

原文摘要 · Abstract (English)

Group Relative Policy Optimization (GRPO) relies on the diversity of $K$ rollouts within each group; otherwise, the group-mean advantage $A^{(k)} = r^{(k)} - μ_r$ collapses to zero. This presents a structural challenge for latent-reasoning models like Coconut, which feed continuous hidden states recurrently in place of discrete chain-of-thought tokens. Because the latent phase is inherently deterministic given the parameters and prompt, multiple rollouts produce identical trajectories, stalling GRPO's progress. Consequently, applying group-relative reinforcement learning to continuous latent reasoning has proven difficult. To address this, we propose sourcing the necessary stochasticity through structured dropout. By applying a single Bernoulli mask held constant across all latent recurrence steps for a given rollout, we generate essential trajectory variance. This shared mask effectively treats each rollout as a posterior sample from a variational distribution over parameters, allowing GRPO to optimize the expected reward of a Bayesian model-average policy. We provide both theoretical justification for this method -- including unbiasedness, variance reduction, and the well-definedness of the latent gradient -- and empirical validation. On GSM8K, dropout-GRPO improves a Coconut baseline from $27.29\%$ to $29.01\%$ pass@1, demonstrating the viability of GRPO learning for latent-reasoning models. Our work positions this as a practical, theoretically grounded approach for post-training latent-reasoning LLMs.

强化学习连续推理大模型微调

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。