[Paper] FlowRL: Matching Reward Distributions for LLM Reasoning
FlowRL introduces a paradigm shift in reinforcement learning for LLMs, moving from reward maximization to reward distribution matching. This novel approach addresses the mode collapse problem prevalent in existing methods, promoting diverse exploration and preventing convergence to limited solution patterns. By aligning the policy model with the full reward distribution, FlowRL encourages coverage of multiple solution strategies, resulting in more generalizable reasoning capabilities. Experimental results demonstrate consistent outperformance across mathematical and code reasoning tasks, with impressive improvements over baseline methods. FlowRL’s ability to generate substantially more diverse rollouts opens new possibilities for training reasoning models that can adapt to diverse problem settings, representing a significant advancement in developing more robust and flexible AI systems.
[Paper] FlowRL: Matching Reward Distributions for LLM Reasoning Read More »









