修复强化学习中因期望回报导致的解多样性崩溃问题
Expected Return Causes Outcome-Level Mode Collapse in Reinforcement Learning and How to Fix It with Inverse Probability Scaling
- 提出反概率缩放修正目标函数,消除结果频率放大效应
- 理论证明可实现奖励比例分布,避免多模态解坍缩
- 无需额外模型或结构改动,适配多种生成任务
许多强化学习问题存在多个质量相当的终态解,目标是生成多样化的高质量结果而非单一最优解。然而,标准期望回报最大化训练出的策略常坍缩至少数解,传统解释归因于探索不足或正则化弱。我们发现该现象本质源于期望回报目标本身的结构性缺陷:在理想学习动态下,任意两解的对数概率比随奖励差线性演化,导致概率比指数级发散,坍缩不可避免,与探索策略、熵正则化或优化算法无关。问题根源在于期望内部的概率乘子,我们提出最小修正——反概率缩放(IPS),移除学习信号中的结果频率放大,从根本上改变学习动态,可证明实现奖励成比例的终态分布,有效防止多模态场景下的坍缩。我们将该原则应用于组相对策略优化(GRPO)中,形成即插即用的IPS-GRPO,无需辅助模型或架构修改。在不同推理和分子生成任务中,IPS-GRPO持续降低结果层面模式坍缩,同时性能匹配或超越基线,表明修正目标而非增加探索启发式才是可靠多模态策略优化的关键。
原文摘要 · Abstract (English)
Many reinforcement learning (RL) problems admit multiple terminal solutions of comparable quality, where the goal is not to identify a single optimum but to represent a diverse set of high-quality outcomes. Nevertheless, policies trained by standard expected return maximization routinely collapse onto a small subset of outcomes, a phenomenon commonly attributed to insufficient exploration or weak regularization. We show that this explanation is incomplete: outcome level mode collapse is a structural consequence of the expected-return objective itself. Under idealized learning dynamics, the log-probability ratio between any two outcomes evolves linearly in their reward difference, implying exponential ratio divergence and inevitable collapse independent of the exploration strategy, entropy regularization, or optimization algorithm. We identify the source of this pathology as the probability multiplier inside the expectation and propose a minimal correction: inverse probability scaling, which removes outcome-frequency amplification from the learning signal, fundamentally changes the learning dynamics, and provably yields reward-proportional terminal distributions, preventing collapse in multimodal settings. We instantiate this principle in Group Relative Policy Optimization (GRPO) as a drop-in modification, IPS-GRPO, requiring no auxiliary models or architectural changes. Across different reasoning and molecular generation tasks, IPS-GRPO consistently reduces outcome-level mode collapse while matching or exceeding baseline performance, suggesting that correcting the objective rather than adding exploration heuristics is key to reliable multimodal policy optimization.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。