arXiv · 2604.20244
Hybrid Policy Distillation for LLMs
Abstract
Knowledge distillation (KD) is a powerful paradigm for compressing large language models (LLMs), whose effectiveness depends on intertwined choices of divergence direction, optimization strategy, and data regime. We break down the design of existing KD methods and present a unified view that establishes connections between them, reformulating KD as a reweighted log-likelihood objective at the token level. We further propose Hybrid Policy Distillation (HPD), which integrates the complementary advantages of forward and reverse KL to balance mode coverage and mode-seeking, and combines off-policy data with lightweight, approximate on-policy sampling. We validate HPD on long-generation math reasoning as well as short-generation dialogue and code tasks, demonstrating improved optimization stability, computational efficiency, and final performance across diverse model families and scales. The code related to this work is available at https://github.com/zwhong714/Hybrid-Policy-Distillation.
Explore related subjects
Keep this discovery
Explore connections, maps & timelines
Wenhong Zhu, Ruobing Xie, Rui Wang, Pengfei Liu. 2026-04-22. Hybrid Policy Distillation for LLMs. https://arxiv.org/abs/2604.20244
Cite the original work for its findings. Save a collection to share your selection of sources.