paper-with-me

홈 › Papers

Uni-OPD: Unifying On-Policy Distillation with a Dual-Perspective Recipe

2026-05-05 · Wenjin Hou, Shangpin Peng, Weinong Wang, Zheng Ruan, Yue Zhang, Zhenglin Zhou, Mingqi Gao, Yifei Chen, Kaiqi Wang, Hongming Yang, Chengquan Zhang, Zhuotao Tian, Han Hu, Yi Yang, Fei Wu, Hehe Fan arxiv

On-policy distillation (OPD) has recently emerged as an effective post-training paradigm for consolidating the capabilities of specialized expert models into a single student model. Despite its empirical success, the conditions under which OPD yields reliable improvement remain poorly understood. In this work, we identify two fundamental bottlenecks that limit effective OPD: insufficient exploration of informative states and unreliable teacher supervision for student rollouts. Building on this insight, we propose Uni-OPD, a unified OPD framework that generalizes across Large Language Models (LLMs) and Multimodal Large Language Models (MLLMs), centered on a dual-perspective optimization strategy. Specifically, from the student's perspective, we adopt two data balancing strategies to promote exploration of informative student-generated states during training. From the teacher's perspective, we show that reliable supervision hinges on whether aggregated token-level guidance remains order-consistent with the outcome reward. To this end, we develop an outcome-guided margin calibration mechanism to restore order consistency between correct and incorrect trajectories. We conduct extensive experiments on 5 domains and 16 benchmarks covering diverse settings, including single-teacher and multi-teacher distillation across LLMs and MLLMs, strong-to-weak distillation, and cross-modal distillation. Our results verify the effectiveness and versatility of Uni-OPD and provide practical insights into reliable OPD.

📄 PDF Abstract BibTeX arXiv:2605.03677

Code (0)

등록된 구현이 없습니다.

Similar Papers 제목 키워드 기반

Advantage Shaping as Surrogate Reward Maximization: Unifying Pass@K Policy Gradients

2025-10-27 · Christos Thrampoulidis, Sadegh Mahdavi, Wenlong Deng arxiv

This note reconciles two seemingly distinct approaches to policy gradient optimization for the Pass@K objective in reinforcement learning with verifiable rewards: (1) direct REINFORCE-style methods, and (2) advantage-sha…

Reinforcement Learning

Rethinking On-Policy Distillation of Large Language Models: Phenomenology, Mechanism, and Recipe

2026-04-14 · Yaxuan Li, Yuxin Zuo, Bingxiang He, Jinqian Zhang 외 arxiv

On-policy distillation (OPD) has become a core technique in the post-training of large language models, yet its training dynamics remain poorly understood. This paper provides a systematic investigation of OPD dynamics a…

Unifying Generative Models with GFlowNets and Beyond

2022-09-06 · Dinghuai Zhang, Ricky T. Q. Chen, Nikolay Malkin, Yoshua Bengio

There are many frameworks for deep generative modeling, each often presented with their own specific training algorithms and inference methods. Here, we demonstrate the connections between existing deep generative models…

Decision Making

ARLArena: A Unified Framework for Stable Agentic Reinforcement Learning

2026-02-25 · Xiaoxuan Wang, Han Zhang, Haixin Wang, Yidan Shi 외 arxiv

Agentic reinforcement learning (ARL) has rapidly gained attention as a promising paradigm for training agents to solve complex, multi-step interactive tasks. Despite encouraging early results, ARL remains highly unstable…

Reinforcement Learning

Dual Policy Distillation

2020-06-07 · Kwei-Herng Lai, Daochen Zha, Yuening Li, Xia Hu

Policy distillation, which transfers a teacher policy to a student policy has achieved great success in challenging tasks of deep reinforcement learning. This teacher-student framework requires a well-trained teacher mod…

continuous-controlContinuous ControlDeep Reinforcement Learningreinforcement-learning+2