arXiv Computation and Language By Hojun Cho, Kang Eun Jeon, Jaegul Choo

Adaptive-Policy Distillation: Knowledge Injection under Information Asymmetry

Read the original on arXiv Computation and Language →

The paper introduces Adaptive-Policy Distillation (APD), a method that improves knowledge injection from a context-conditioned teacher model into a student model by dynamically adjusting the distillation policy and loss function at each token. APD addresses two structural failures of On‑Policy Distillation (OPD): the reverse‑KL supervision’s weak influence on unlikely document tokens and the degradation of coverage for longer target sequences. Experiments on QASPER and BioASQ show that APD outperforms OPD by up to 54.6%, achieving state‑of‑the‑art F1 and MRR scores.

Machine-generated by The Flow from the publisher's headline and feed description — not written or checked by a human. The full article lives at arXiv Computation and Language.

arXiv Statistics ML
Oct 2

Unbiased Top-$k$ Estimation for On-Policy Distillation

The paper introduces Tail‑Corrected Top‑k On‑Policy Distillation (TT‑OPD), a method that improves on existing Top‑k OPD by combining the selected top‑k tokens with a sampled token from the student’s rollout. This hybrid approach recovers the probability mass discarded by limiting to top‑k, yielding an unbiased estimator of the reverse KL divergence gradient while maintaining low computational cost. Experiments show TT‑OPD outperforms other OPD variants in accuracy.

By Linjian Meng, Siyuan Gan, YuHan Li, Xiran Wang, Ziyang Ding, Ditang Gou, Yiming Wu, Zhen Zhao
arXiv Machine Learning
Jul 7

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

arXiv:2605. 03677v2 Announce Type: replace Abstract: 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.

By 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