arXiv:2602.20132cs.LG2026-02ACL被引 2

让大模型推理更聪明:用优势分布提升多样性和准确性

LAD: Learning Advantage Distribution for Reasoning

  • 用优势分布替代传统奖励最大化,避免单一路径过拟合
  • 在数学与代码任务中同时提升准确率和生成多样性
  • 无需额外计算成本,适合大模型后训练阶段使用

当前大型模型推理的强化学习目标主要聚焦于最大化期望奖励,这可能导致对主导奖励信号的过拟合,忽略其他有效推理路径,从而限制多样性与探索。为此,我们提出学习优势分布(LAD),一种基于分布匹配的框架,将优势最大化替换为学习优势诱导的分布。通过建立最优策略更新与基于优势的目标分布之间的等价性,推导出一个实际可操作的LAD目标,即最小化策略诱导分布与优势诱导分布之间的f-散度。该方法生成的梯度能提高高优势响应的可能性,同时抑制过度自信的概率增长,防止分布坍缩,且无需辅助熵正则化。LAD与GRPO相比不增加额外训练开销,可自然扩展至大模型后训练阶段。在受控老虎机环境中,LAD能准确恢复多模态优势分布,验证了理论设计。在多个大模型基座上的数学与代码推理任务实验表明,LAD能稳定提升准确率与生成多样性。

原文摘要 · Abstract (English)

Current reinforcement learning objectives for large-model reasoning primarily focus on maximizing expected rewards. This paradigm can lead to overfitting to dominant reward signals, while neglecting alternative yet valid reasoning trajectories, thereby limiting diversity and exploration. To address this issue, we introduce Learning Advantage Distributions (LAD), a distribution-matching framework that replaces advantage maximization with learning the advantage-induced distribution. By establishing the equivalence between the optimal policy update and an advantage-based target distribution, we derive a practical LAD objective formulated as minimizing an $f$-divergence between the policy-induced and advantage-induced distributions. This yields a gradient update that increases likelihood for high-advantage responses while suppressing over-confident probability growth, preventing collapse without requiring auxiliary entropy regularization. LAD incurs no extra training cost compared to GRPO and scales naturally to LLM post-training. In a controlled bandit setting, LAD faithfully recovers the multimodal advantage distribution, validating the theoretical formulation. Experiments on math and code reasoning tasks across several LLM backbones show that LAD reliably improves both accuracy and generative diversity.

强化学习大模型推理多样性

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