ICML 2026poster0 citations

Entropy-aware Span-Constrained Optimal Transport for Robust Cross-Tokenizer Knowledge Distillation

Zhi-Ping Liu, Simiao Li, Wei Li, Hanting Chen, Jie Hu, Hua-Lei Yin, Xinghao Chen

Abstract

Existing Cross-Tokenizer Knowledge Distillation (CTKD) methods fail to outperform simple supervised fine-tuning when vocabulary overlap is low due to severe alignment noise. We identify this phenomenon as the **``Low-Overlap negative transfer regime,''** To overcome this, we propose **Entropy-aware Span-Constrained Optimal Transport (E-SCOT)**, a robust framework that treats distillation as a sparse transport problem with a vocabulary-agnostic ground metric. Unlike prior OT approaches that incur quadratic costs via dense optimization, E-SCOT employs span-anchored lexical alignment to construct a deterministic, locality-preserving coupling in linear time. Furthermore, we introduce R\'enyi-entropy adaptive reweighting to dynamically concentrate the distillation budget on informative positions exhibiting significant uncertainty-profile gaps. Extensive experiments demonstrate that E-SCOT achieves state-of-the-art performance across diverse model families, effectively eliminating negative transfer even in challenging low-overlap scenarios.

OptimizationRobustness
BibTeX
@inproceedings{
liu2026entropyaware,
title={Entropy-aware Span-Constrained Optimal Transport for Robust Cross-Tokenizer Knowledge Distillation},
author={Zhi-Ping Liu and Simiao Li and Wei Li and Hanting Chen and Jie Hu and Hua-Lei Yin and Xinghao Chen},
booktitle={Forty-third International Conference on Machine Learning},
year={2026},
url={https://openreview.net/forum?id=ci0YXenfYe}
}