Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

Understanding and Preventing Entropy Collapse in RLVR with On-Policy Entropy Flow Optimization

About

Reinforcement learning with verifiable rewards (RLVR) has become an effective paradigm for improving the reasoning ability of large language models. However, widely used RLVR algorithms, such as GRPO, often suffer from entropy collapse, leading to premature determinism and unstable optimization. Existing remedies, including entropy regularization and ratio-based clipping heuristics, either control entropy in a coarse-grained manner or rely on approximate on-policy training. In this paper, we revisit entropy collapse from a token-level entropy flow perspective. Our analysis reveals that entropy-decreasing tokens consistently outweigh entropy-increasing ones, resulting in a severely imbalanced entropy flow. This perspective provides a unified explanation of entropy collapse in existing RLVR algorithms and highlights the importance of balancing entropy dynamics. Motivated by this analysis, we propose On-Policy Entropy Flow Optimization (OPEFO), an adaptive entropy flow balancing mechanism that rescales entropy-increasing and entropy-decreasing updates according to their contributions to entropy change, while remaining strict on-policy. Experiments on six mathematical reasoning benchmarks demonstrate that OPEFO improves training stability and final performance. We will release the code and models upon publication.

Huimin Xu, Shuai Zhao, Xiaobao Wu, Anh Tuan Luu• 2026

Related benchmarks

TaskDatasetResultRank
Mathematical ReasoningAIME 2024
Accuracy34.5
479
Mathematical ReasoningAMC 2023
Accuracy82.2
144
Mathematical ReasoningCombined Mathematical Reasoning Benchmarks
Average Accuracy52.4
30
Mathematical ReasoningAIME 24
Pass@3262.4
7
Mathematical ReasoningAIME 25
Pass@3243.3
7
Mathematical ReasoningAMC 23
Pass@3295.6
7
Mathematical ReasoningMATH 500
Pass@320.941
7
Showing 7 of 7 rows

Other info

Follow for update