Track papers across labs, topics, and trends.
Search papers, labs, and topics across Lattice.
This paper introduces Trust Region Policy Distillation (TOP-D), a novel approach that stabilizes On-Policy Distillation (OPD) by dynamically constructing a proximal teacher to control gradient variance. The authors provide a theoretical framework that includes a global convergence analysis and a monotonic improvement bound, ensuring the reliability and stability of the training process. Empirical results show that TOP-D significantly improves training stability, sample efficiency, and performance on mathematical reasoning tasks without incurring additional computational costs.
TOP-D transforms the unstable OPD paradigm into a stable training method that enhances performance without any extra computational burden.
Big goals are hard to achieve all at once; breaking them into small steps is wiser. We present Trust Region Policy Distillation (TOP-D), which transforms the notoriously unstable, high-variance On-Policy Distillation (OPD) into a stable training paradigm by dynamically constructing a proximal teacher. Theoretically, we establish a rigorous framework demonstrating that TOP-D inherently controls gradient variance. By providing a formal global convergence analysis alongside a monotonic improvement bound, we mathematically formalize the reliability and stability of the overall training dynamics. Empirically, TOP-D dramatically enhances training stability, sample efficiency, and final performance on mathematical reasoning tasks. More importantly, TOP-D introduces zero additional computational overhead, positioning itself as a promising alternative to the well-established OPD paradigm.