Expected Return Causes Outcome-Level Mode Collapse in Reinforcement Learning and How to Fix It with Inverse Probability Scaling
Not on the map. Why?
Abstract (arXiv)
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.
Key facts. Cited by 4 papers in the map (data through 7 Oct 2026). Builds on 35 of its 42 references that are in the map. First posted on arXiv 29 Jan 2026.
Builds on
35 of 42 references.
References · 35
- 2022Training language models to follow instructions with human feedbackcited by 15,712
- 2024DeepSeekMath: Pushing the Limits of Mathematical Reasoning in Open Language Modelscited by 8,851
- 2023Direct Preference Optimization: Your Language Model is Secretly a Reward Modelcited by 7,916
- 2022Generative Adversarial Networkscited by 5,603
- 2018Soft Actor-Critic: Off-Policy Maximum Entropy Deep Reinforcement Learning with a Stochastic Actorcited by 4,635
- 2022Constitutional AI: Harmlessness from AI Feedbackcited by 2,950
- 2019The Curious Case of Neural Text Degenerationcited by 2,881
- 2020Learning to summarize from human feedbackcited by 2,562
- 2019Fine-Tuning Language Models from Human Preferencescited by 2,124
- 2016Concrete Problems in AI Safetycited by 1,582
- 2022Scaling Laws for Reward Model Overoptimizationcited by 1,056
- 2025AlphaEvolve: A coding agent for scientific and algorithmic discoverycited by 763
23 more
- 2025Beyond the 80/20 Rule: High-Entropy Minority Tokens Drive Effective Reinforcement Learning for LLM Reasoningcited by 540
- 2025The Entropy Mechanism of Reinforcement Learning for Reasoning Language Modelscited by 463
- 2023Understanding the Effects of RLHF on LLM Generalisation and Diversitycited by 386
- 2021GFlowNet Foundationscited by 286
- 2025Reasoning with Exploration: An Entropy Perspectivecited by 277
- 2017Molecular De Novo Design through Deep Reinforcement Learningcited by 264
- 2023Does Writing with Language Models Reduce Content Diversity?cited by 190
- 2023Beyond Reverse KL: Generalizing Direct Preference Optimization with Diverse Divergence Constraintscited by 174
- 2018Understanding the impact of entropy on policy optimizationcited by 171
- 2022Sample Efficiency Matters: A Benchmark for Practical Molecular Optimizationcited by 151
- 2023Amortizing intractable inference in large language modelscited by 94
- 2025Outcome-based Exploration for LLM Reasoningcited by 79
- 2022Winner Takes It All: Training Performant RL Populations for Combinatorial Optimizationcited by 57
- 2025FlowRL: Matching Reward Distributions for LLM Reasoningcited by 38
- 2025KL-Regularized Reinforcement Learning is Designed to Mode Collapsecited by 29
- 2024Maximum Entropy Reinforcement Learning via Energy-Based Normalizing Flowcited by 25
- 2025Diverse Preference Learning for Capabilities and Alignmentcited by 25
- 2024Saturn: Sample-efficient Generative Molecular Design using Memory Manipulationcited by 14
- 2018Program Synthesis Through Reinforcement Learning Guided Tree Searchcited by 7
- 2024Mol-AIR: Molecular Reinforcement Learning with Adaptive Intrinsic Rewards for Goal-directed Molecular Generationcited by 7
- 2025Co-GRPO: Co-Optimized Group Relative Policy Optimization for Masked Diffusion Modelcited by 5
- 2025Generative Molecular Design with Steerable and Granular Synthesizability Controlcited by 4
- 2024PreND: Enhancing Intrinsic Motivation in Reinforcement Learning through Pre-trained Network Distillationcited by 1
Built on by
Citing papers per year: 2026: 4.
4 papers.
- 2026Beyond the Best Guess: Improving LLM Solution Coverage with Evolution Strategiescited by 2
- 2026The Past Is Not Past: Memory-Enhanced Dynamic Reward Shapingcited by 0
- 2026PIVOT: Pivot-Aware On Policy Self Distillation for Multi-Turn VLM Agentscited by 0
- 2026Measuring and Mitigating Solution Mode Collapse in RLVRcited by 0
- 2026Measuring and Mitigating Solution Mode Collapse in RLVR8 Oct 2026
- 2026PIVOT: Pivot-Aware On Policy Self Distillation for Multi-Turn VLM Agents28 Sep 2026
- 2026Beyond the Best Guess: Improving LLM Solution Coverage with Evolution Strategies13 Aug 2026
- 2026The Past Is Not Past: Memory-Enhanced Dynamic Reward Shaping13 Apr 2026
Similar, not cited
Close in wording, no citation either way
- 2026On Training in Imagination
- 2026A Lecture Note on Offline RL and IRL, Part II: Foundations of Inverse Reinforcement Learning and Dynamic Discrete Choice Models
- 2025Statistical analysis of Inverse Entropy-regularized Reinforcement Learning
- 2025Low-probability Tokens Sustain Exploration in Reinforcement Learning with Verifiable Reward
- 2026An Informational Curse of Horizon in Goal-Conditioned Policy Learning
- 2025Horizon Reduction as Information Loss in Offline Reinforcement Learning
- 2026Success Conditioning as Policy Improvement: The Optimization Problem Solved by Imitating Success
- 2026Entropy Polarity in Reinforcement Fine-Tuning: Direction, Asymmetry, and Control