SAF Improves Reinforcement Learning with On-Policy Distillation

Yifan Ding, Xincheng Wei, Yoshua Y. Li, Ziheng Li, Yuquan Lu, Siyu Zhang, Dongsheng Ma, Rongxiang Weng, Xunliang Cai, Yun Chen· August 3, 2026 View original

Key takeaways

  • Combining RLVR and OPD can cause entropy collapse due to advantage mismatches.
  • SAF resolves magnitude and temporal mismatches with a four-stage pipeline.
  • It prevents entropy collapse and improves training stability and performance.
  • SAF enhances AI agents for tasks like code generation and mathematical reasoning.

Who benefits

Software DevelopmentAI/ML ResearchEducationFinanceRobotics

Summary

Researchers propose SAF, a Stable Advantage Fusion framework, to combine Reinforcement Learning with Verifiable Rewards (RLVR) and On-Policy Distillation (OPD) without causing entropy collapse. SAF resolves magnitude and temporal mismatches between advantages, leading to more stable training and improved performance across benchmarks.

This paper introduces SAF (Stable Advantage Fusion), a novel framework designed to effectively combine Reinforcement Learning with Verifiable Rewards (RLVR) and On-Policy Distillation (OPD). While both techniques are complementary—RLVR provides response-level rewards and OPD offers dense token-level advantages from a stronger teacher—their direct fusion often leads to "entropy collapse," where the model's exploration capacity is severely limited. The core problem lies in two mismatches: a magnitude mismatch, where OPD advantages can overwhelm RLVR signals, and a temporal mismatch, where continuous strong OPD guidance hinders exploration. SAF addresses these issues through a lightweight, four-stage pipeline applied specifically to the OPD advantage. This pipeline includes a sparsify-then-compress mechanism for magnitude control and a warm-up-then-anneal mechanism for temporal control. Each stage is independently switchable and adds minimal overhead. Evaluated with GRPO (a type of RLVR) across seven mathematical reasoning and code generation benchmarks using Qwen3 models, SAF consistently avoids entropy collapse. It outperforms fixed-coefficient GRPO+OPD fusion, improving aggregate scores by 0.51-2.70% across various model-domain settings and achieving more stable training. This demonstrates SAF's effectiveness in leveraging the strengths of both RLVR and OPD for enhanced reinforcement learning.

Why it matters

Professionals developing advanced AI agents for tasks like code generation, mathematical reasoning, or complex decision-making can use SAF to train more stable and performant models by effectively combining different reinforcement learning signals.

How to implement this in your domain

  1. 1Review current reinforcement learning strategies for potential entropy collapse or instability issues.
  2. 2Investigate integrating SAF's advantage fusion pipeline into existing RL training frameworks.
  3. 3Experiment with SAF's sparsify-then-compress and warm-up-then-anneal mechanisms for optimal performance.
  4. 4Apply SAF to improve the training stability and performance of AI models in code generation or reasoning tasks.

Original post by Yifan Ding, Xincheng Wei, Yoshua Y. Li, Ziheng Li, Yuquan Lu, Siyu Zhang, Dongsheng Ma, Rongxiang Weng, Xunliang Cai, Yun Chen

"arXiv:2607.29209v1 Announce Type: new Abstract: Reinforcement learning with verifiable rewards (RLVR) broadcasts a single response-level reward to every token, while on-policy distillation (OPD) scores each token against a stronger teacher for a dense advantage but caps performan…"

View on X

Originally posted by Yifan Ding, Xincheng Wei, Yoshua Y. Li, Ziheng Li, Yuquan Lu, Siyu Zhang, Dongsheng Ma, Rongxiang Weng, Xunliang Cai, Yun Chen on X · view source

Want to go deeper?

Turn these trends into skills with Learnijoy's hands-on AI & tech courses.

Explore courses