Self-Distilled Policy Gradient
Paper · 2026 · Yifeng Liu, Shiyuan Zhang, Yifan Zhang et al.
Summary
RLVR methods such as GRPO give a language model a single correctness score per sequence, so individual reasoning steps receive no specific credit. On-policy self-distillation offers dense token-level guidance instead: the same model sees privileged context, such as a reference solution, and acts as its own teacher. Without a verifier, though, it may reinforce plausible tokens on wrong trajectories.
The authors show that, on a fixed sampled prefix, the student-side gradient of a full-vocabulary reverse KL toward a detached privileged teacher is equivalent to a policy-gradient step. In that step, each token's advantage is a centered teacher/student log ratio.
Building on this, SDPG combines three parts:
- group-normalized verifier advantages,
- exact full-vocabulary self-distillation,
- an unnormalized KL anchor to a fixed reference policy.
Distillation applies only to positive-advantage rollouts, and its weight follows a warmup-decay schedule. With Qwen3-4B, SDPG outperforms GRPO and RLSD on AIME and AMC benchmarks. This suggests that dense self-supervision and verifier rewards complement each other.
Key points
-
SDPG has two reference-KL variants, SDPG-URKL (unnormalized reverse) and SDPG-UFKL (unnormalized forward), building on the RPG framework.
-
Privileged context for the teacher includes dataset answers plus solutions generated by Gemini 2.5 Pro.
-
Setup: Qwen3-4B, 400 training steps, evaluated with pass@1 (mean@32) on AIME24, AIME25 and AMC23.
-
Final-step scores (AIME24 / AIME25 / AMC23):
Method AIME24 AIME25 AMC23 GRPO 0.280 0.242 0.714 RLSD 0.378 0.300 0.813 SDPG-URKL 0.380 0.307 0.863 SDPG-UFKL 0.380 0.327 0.858 The best peaks came from SDPG-UFKL: 0.408 / 0.335 / 0.870.
-
Code is released at github.com/lauyikfung/SDPG.
Institutions
Fields, topics and concepts
📚 Shelves 4The field of science this work belongs to (OpenAlex fields).Computer Science · 19,503 worksArtificial Intelligence · 18,805 worksArts and Humanities · 17,959 worksLanguage and Linguistics · 17,789 works
🧭 Topics 1Research subjects and languages the work is about.Efficient models & compression · 328 works
Cite
Cite as Yifeng Liu, Shiyuan Zhang, Yifan Zhang and Quanquan Gu. 2026. Self-Distilled Policy Gradient. arXiv:2606.04036.
@misc{liu2026self,
title = {{Self-Distilled Policy Gradient}},
author = {Yifeng Liu, Shiyuan Zhang, Yifan Zhang and Quanquan Gu},
year = {2026},
doi = {10.48550/arxiv.2606.04036},
eprint = {2606.04036},
archivePrefix = {arXiv},
url = {https://doi.org/10.48550/arxiv.2606.04036}
}References
- Agarwal, R., Vieillard, N., Zhou, Y., Stanczyk, P., Garea, S. R., Geist, M. and Bachem, O. (2024). On-policy distillation of language models: Learning from self-generated mistakes. In The twelfth international conference on learning representations.
- Chen, G., Liao, M., Li, C. and Fan, K. (2024). Step-level value preference optimization for mathematical reasoning. In Findings of the Association for Computational Linguistics: EMNLP 2024. find
- Chen, M., Chen, G., Wang, W. and Yang, Y. (2025). Seed-grpo: Semantic entropy enhanced grpo for uncertainty-aware policy optimization. arXiv preprint arXiv:2505.12346 . — [arXiv:2505.12346](https://arxiv.org/abs/2505.12346) find
- Cheng, D., Huang, S., Zhu, X., Dai, B., Zhao, X., Zhang, Z. and Wei, F. (2026). Reasoning with exploration: An entropy perspective. In Proceedings of the AAAI Conference on Artificial Intelligence, vol. 40.
- Comanici, G., Bieber, E., Schaekermann, M. et al. (2025). Gemini 2.5: Pushing the frontier with advanced reasoning, multimodality, long context, and next generation agentic capabilities. arXiv preprint arXiv:2507.06261 .
- Dai, M., Yang, C. and Si, Q. (2025). S-grpo: Early exit via reinforcement learning in reasoning models. In The Thirty-ninth Annual Conference on Neural Information Processing Systems. find
- Fu, Y., Huang, H., Jiang, K., Zhu, Y. and Zhao, D. (2026). Revisiting on-policy distillation: Empirical failure modes and simple fixes. arXiv preprint arXiv:2603.25562 .
- Gu, Y., Dong, L., Wei, F. and Huang, M. (2024). Minillm: Knowledge distillation of large language models. In The twelfth international conference on learning representations. find
- Hübotter, J., Lübeck, F., Behric, L., Baumann, A., Bagatella, M., Marta, D., Hakimi, I., Shenfeld, I., Buening, T. K., Guestrin, C. et al. (2026). Reinforcement learning via self-distillation. arXiv preprint arXiv:2601.20802 .
- Kazemnejad, A., Aghajohari, M., Portelance, E., Sordoni, A., Reddy, S., Courville, A. and Roux, N. L. (2024). Vineppo: Refining credit assignment in rl training of llms. arXiv preprint arXiv:2410.01679 .
- Kwon, W. (2025). vLLM: An Efficient Inference Engine for Large Language Models. Ph.D. thesis, UC Berkeley. find
- Lee, B. W., Cho, H. and Yoo, K. M. (2024). Instruction tuning with human curriculum. In Findings of the Association for Computational Linguistics: NAACL 2024. find
- Li, Z., Kang, L., Xiao, F., Xing, L., Si, Q., Li, Z., Gong, W., Yang, D., Xiao, Y. and Guo, H. (2026). Outcome-grounded advantage reshaping for fine-grained credit assignment in mathematical reasoning. arXiv preprint arXiv:2601.07408 . find
- Lightman, H., Kosaraju, V., Burda, Y., Edwards, H., Baker, B., Lee, T., Leike, J., Schulman, J., Sutskever, I. and Cobbe, K. (2023). Let’s verify step by step. In The twelfth international conference on learning representations.
- Liu, Z., Chen, C., Li, W., Qi, P., Pang, T., Du, C., Lee, W. S. and Lin, M. (2025). Understanding r1-zero-like training: A critical perspective. arXiv preprint arXiv:2503.20783 .
- Loshchilov, I. and Hutter, F. (2018). Decoupled weight decay regularization. In International Conference on Learning Representations. find
- Lu, K. and Lab, T. M. (2025). On-policy distillation. Thinking Machines Lab: Connectionism . find
- Luo, L., Liu, Y., Liu, R., Phatale, S., Guo, M., Lara, H., Li, Y., Shu, L., Zhu, Y., Meng, L. et al. (2024). Improve mathematical reasoning in language models by automated process supervision. arXiv preprint arXiv:2406.06592 .
- MAA, M. A. o. A. A. M. C. (2024a). 2024 AIME-I. Accessed: 2025-05-08. find
- MAA, M. A. o. A. A. M. C. (2024b). 2024 AIME-II. Accessed: 2025-05-08. find
- MAA, M. A. o. A. A. M. C. (2025a). 2025 AIME-I. find
- MAA, M. A. o. A. A. M. C. (2025b). 2025 AIME-II. find
- Minka, T. (2005). Divergence measures and message passing . find
- Penaloza, E., Vattikonda, D., Gontier, N., Lacoste, A., Charlin, L. and Caccia, M. (2026). Privileged information distillation for language models. arXiv preprint arXiv:2602.04942 . — [arXiv:2602.04942](https://arxiv.org/abs/2602.04942) find
- Schulman, J. (2020). Approximating kl divergence. http://joschu.net/blog/kl-approx.html. Accessed on June 4, 2026. ↗
- Schulman, J., Wolski, F., Dhariwal, P., Radford, A. and Klimov, O. (2017). Proximal policy optimization algorithms. arXiv preprint arXiv:1707.06347 . find
- Shao, Z., Wang, P., Zhu, Q., Xu, R., Song, J., Bi, X., Zhang, H., Zhang, M., Li, Y., Wu, Y. et al. (2024). Deepseekmath: Pushing the limits of mathematical reasoning in open language models. arXiv preprint arXiv:2402.03300 .
- Shenfeld, I., Damani, M., Hübotter, J. and Agrawal, P. (2026). Self-distillation enables continual learning. arXiv preprint arXiv:2601.19897 .
- Sheng, G., Zhang, C., Ye, Z., Wu, X., Zhang, W., Zhang, R., Peng, Y., Lin, H. and Wu, C. (2025). Hybridflow: A flexible and efficient rlhf framework. In Proceedings of the Twentieth European Conference on Computer Systems.
- Shi, T., Wu, Y., Song, L., Zhou, T. and Zhao, J. (2025). Efficient reinforcement finetuning via adaptive curriculum learning. In NeurIPS 2025 Workshop on Efficient Reasoning. find
- Srivastava, S. S. and Aggarwal, V. (2025). A technical survey of reinforcement learning techniques for large language models. arXiv preprint arXiv:2507.04136 . — [arXiv:2507.04136](https://arxiv.org/abs/2507.04136) find
- Wang, P., Li, L., Shao, Z., Xu, R., Dai, D., Li, Y., Chen, D., Wu, Y. and Sui, Z. (2024). Math-shepherd: Verify and reinforce llms step-by-step without human annotations. In Proceedings of the 62nd Annual Meeting of the Association for Computational Linguistics (Volume 1: Long Papers).
- Wang, S., Yu, L., Gao, C., Zheng, C., Liu, S., Lu, R., Dang, K., Chen, X.-H., Yang, J., Zhang, Z. et al. (2025a). Beyond the 80/20 rule: High-entropy minority tokens drive effective reinforcement learning for llm reasoning. In The Thirty-ninth Annual Conference on Neural Information Processing Syste…
- Wang, X., Chen, Y. and Zhu, W. (2021). A survey on curriculum learning. IEEE transactions on pattern analysis and machine intelligence 44 4555–4576. find
- Wang, Y., Yang, Q., Zeng, Z., Ren, L., Liu, L., Peng, B., Cheng, H., He, X., Wang, K., Gao, J. et al. (2025b). Reinforcement learning for reasoning in large language models with one training example. In The Thirty-ninth Annual Conference on Neural Information Processing Systems.
- Wen, L., Cai, Y., Xiao, F., He, X., An, Q., Duan, Z., Du, Y., Liu, J., Tanglifu, T., Lv, X. et al. (2025). Light-r1: Curriculum sft, dpo and rl for long cot from scratch and beyond. In Proceedings of the 63rd Annual Meeting of the Association for Computational Linguistics (Volume 6: Industry Track).
- Williams, R. J. (1992). Simple statistical gradient-following algorithms for connectionist reinforcement learning. Machine learning 8 229–256. find
- Xiao, B., Xia, B., Yang, B., Gao, B., Shen, B., Zhang, C., He, C., Lou, C., Luo, F., Wang, G. et al. (2026). Mimo-v2-flash technical report. arXiv preprint arXiv:2601.02780 .
- Xie, C., Pan, R., Wu, X., Zhang, Y., Fu, J., Gao, T. and Zhou, G. (2025). Unlocking exploration in rlvr: Uncertainty-aware advantage shaping for deeper reasoning. arXiv preprint arXiv:2510.10649 .
- Xu, W., Han, R., Wang, Z., Le, L., Madeka, D., Li, L., Wang, W. Y., Agarwal, R., Lee, C.-Y. and Pfister, T. (2025). Speculative knowledge distillation: Bridging the teacher-student gap through interleaved sampling. In The Thirteenth International Conference on Learning Representations.
- Yang, A., Li, A., Yang, B., Zhang, B., Hui, B., Zheng, B., Yu, B., Gao, C., Huang, C., Lv, C. et al. (2025). Qwen3 technical report. arXiv preprint arXiv:2505.09388 .
- Yang, C., Qin, C., Si, Q., Chen, M., Gu, N., Yao, D., Lin, Z., Wang, W., Wang, J. and Duan, N. (2026a). Self-distilled rlvr. arXiv preprint arXiv:2604.03128 .
- Yang, C., Si, Q., Dai, M., Yao, D., Zheng, M., Chen, M., Lin, Z. and Wang, W. (2026b). Test-time prompt intervention. In Proceedings of the AAAI Conference on Artificial Intelligence, vol. 40. find
- Ye, T., Dong, L., Wu, X., Huang, S. and Wei, F. (2026). On-policy context distillation for language models. arXiv preprint arXiv:2602.12275 .
- Yu, Q., Zhang, Z., Zhu, R., Yuan, Y., Zuo, X., Yue, Y., Dai, W., Fan, T., Liu, G., Liu, L. et al. (2025). Dapo: An open-source llm reinforcement learning system at scale. arXiv preprint arXiv:2503.14476 .
- Zhang, L., Hosseini, A., Bansal, H., Kazemi, M., Kumar, A. and Agarwal, R. (2025). Generative verifiers: Reward modeling as next-token prediction. In The Thirteenth International Conference on Learning Representations.
- Zhang, Y., Liu, Y., Yuan, H., Yuan, Y., Gu, Q. and Yao, A. C. (2026a). On the design of kl-regularized policy gradient algorithms for llm reasoning. In The Fourteenth International Conference on Learning Representations. find
- Zhang, Z., Jiang, S., Shen, Y., Zhang, Y., Ram, D., Yang, S., Tu, Z., Xia, W. and Soatto, S. (2026b). Reinforcement-aware knowledge distillation for llm reasoning. arXiv preprint arXiv:2602.22495 .
- Zhao, S., Xie, Z., Liu, M., Huang, J., Pang, G., Chen, F. and Grover, A. (2026). Self-distilled reasoner: On-policy self-distillation for large language models. arXiv preprint arXiv:2601.18734 .
- Zheng, C., Liu, S., Li, M., Chen, X.-H., Yu, B., Gao, C., Dang, K., Liu, Y., Men, R., Yang, A. et al. (2025). Group sequence policy optimization. arXiv preprint arXiv:2507.18071 .
- Zhu, H. and Rohwer, R. (1995). Information geometric measurements of generalisation. Preprint . find
Updated
This page in Markdown → https://md.research.elamur.ai/free/en/papers/liu-2026-self-distilled-policy-gradient/