Token-Level On-Policy Contrastive Learning for Black-box Distillation Without Full Rollouts

Jian Yang, Chern Hong Lim, Thuy-Trang Vu


Abstract
Black-box distillation trains student models using only teacher-generated outputs, making it practical when teacher models are accessible only through APIs. However, existing methods either rely mainly on positive imitation or require costly full-response rollouts to obtain corrective signals. We show that such corrective signals can be obtained locally from teacher trajectories alone. At each position, the teacher token provides a positive anchor, while a high-probability alternative sampled from the student provides an informative negative. Based on this insight, we propose Token-level On-policy contrastive Learning for blackbox Distillation (TOLD), which samples a single negative token from the student’s distribution and optimizes a Bradley-Terry objective anchored by negative log-likelihood. This enables the student to reinforce teacher tokens while suppressing its own likely mistakes, without teacher logits, reward models, or full-response rollouts. Experiments on reasoning and instruction-following tasks show that TOLD outperforms or matches strong black-box distillation baselines and achieves up to 1.88 times training speedup over rollout-based on-policy baselines.
Anthology ID:
2026.aacl-long.34
Volume:
Proceedings of the 15th International Joint Conference on Natural Language Processing and the 5th Conference of the Asia-Pacific Chapter of the Association for Computational Linguistics (Volume 1: Long Papers)
Month:
November
Year:
2026
Address:
Hengqin, China
Editors:
Yue Zhang, Ruifeng Xu, Ayu Purwarianti, Tong Rong, Jun Suzuki
Venues:
AACL | IJCNLP
SIG:
Publisher:
Association for Computational Linguistics
Note:
Pages:
705–718
Language:
URL:
https://aclanthology.org/2026.aacl-long.34/
DOI:
Bibkey:
Cite (ACL):
Jian Yang, Chern Hong Lim, and Thuy-Trang Vu. 2026. Token-Level On-Policy Contrastive Learning for Black-box Distillation Without Full Rollouts. In Proceedings of the 15th International Joint Conference on Natural Language Processing and the 5th Conference of the Asia-Pacific Chapter of the Association for Computational Linguistics (Volume 1: Long Papers), pages 705–718, Hengqin, China. Association for Computational Linguistics.
Cite (Informal):
Token-Level On-Policy Contrastive Learning for Black-box Distillation Without Full Rollouts (Yang et al., AACL-IJCNLP 2026)
Copy Citation:
PDF:
https://aclanthology.org/2026.aacl-long.34.pdf