Cross-Tokenizer LLM Distillation through a Byte-Level Interface
Organizations: MediaTek Research, Cambridge (United Kingdom)
Abstract
Cross-tokenizer distillation (CTD), the transfer of knowledge from a teacher to a student language model when the two use different tokenizers, remains a largely unsolved problem. Existing approaches rely on heuristic strategies to align mismatched vocabularies, introducing considerable complexity. In this paper, we propose a simple but effective baseline called Byte-Level Distillation (BLD) which enables CTD by operating at a common interface across tokenizers: the byte level. In more detail, we convert the teacher's output distribution to byte-level probabilities, attach a lightweight byte-level decoder head to the student, and distill through this shared byte-level interface. Despite its simplicity, BLD performs competitively with--and on several benchmarks surpasses--significantly more sophisticated CTD methods, across a range of distillation tasks with models from 1B to 8B parameters. Our results suggest that the byte level is a natural common ground for cross-tokenizer knowledge transfer, while also highlighting that consistent improvements across all tasks and benchmarks remain elusive, underscoring that CTD is still an open problem.
Figures & tables
| Model | Method | Benchmark | ||||||
| PiQA | ARC-C | BoolQ | MMLU | AGI-EN | AGI-ZH | IFEval | ||
| original (Llama3.2 3B IT) | 75.46 | 45.73 | 78.41 | 60.50 | 35.27 | 42.93 | 66.31 | |
| Qwen2 | SFT | 74.54 | 41.89 | 76.48 | 57.11 | 30.47 | 34.30 | 26.74 |
| DSKD | 62.95 | 28.84 | 71.80 | 50.48 | 26.12 | 34.18 | 28.13 | |
| MinED | 75.35 | 42.58 | 78.65 | 58.20 | 34.68 | 34.76 | 62.83 | |
| ALM + SFT | 75.46 | 45.82 | 79.36 | 58.86 | 36.64 | 35.27 | 58.51 | |
| Model | Method | Benchmark | ||||||
| PiQA | ARC-C | BoolQ | MMLU | AGI-EN | AGI-ZH | IFEval | ||
| original (Llama3.2 3B IT) | 75.46 | 45.73 | 78.41 | 60.50 | 35.27 | 42.93 | 66.31 | |
| Byte | SFT | 67.30 | 31.57 | 73.00 | 38.95 | 26.05 | 35.18 | 24.70 |
| DSKD | 64.47 | 31.31 | 60.34 | 37.62 | 23.74 | 33.36 | 23.98 | |
| MinED | 67.41 | 32.94 | 65.32 | 39.84 | 27.52 | 33.90 | 31.89 | |
| ALM + SFT | 66.32 | 31.57 | 71.41 | 39.15 | 27.66 | 35.39 | 29.74 | |
| Model | Method | GSM8K | MATH |
| OpenMath2-Llama3.1-8B | 87.26 | 37.60 | |
| Gemma2 2B IT | 51.48 | 10.60 | |
| Gemma2 2B | SFT | 59.29 | 22.40 |
| ALM + SFT | 61.56 | 19.00 | |
| Ours | 62.55 | 20.08 |
Appendix figures & tables3 assets
Supplementary material from the paper’s appendix.
Appendix
| Hyperparameter | Value | Search Space |
| LoRA | ||
| Rank ( ) | 64 | — |
| Alpha ( ) | 64 | — |
| Dropout | 0.05 | — |
| Target modules | q_proj , v_proj , k_proj , o_proj , gate_proj , up_proj , down_proj | |
| Optimiser | ||
Cross-Tokenizer Knowledge Distillation
Cross-tokenizer knowledge distillation transfers a language model’s behavior to a student with a different vocabulary and tokenization scheme.
See all 20 papers