Cross-Tokenizer Likelihood Scoring Algorithms for Language Model Distillation

TL;DR

This paper introduces a probabilistic framework for likelihood scoring across different tokenizers, enabling efficient and accurate language model distillation even with vocabulary mismatches.

Contribution

It uncovers an implicit recursive structure in BPE and develops a method for cross-tokenizer likelihood evaluation, improving distillation efficiency and performance.

Findings

Up to 12% memory reduction in model size during distillation.

Improved task performance by up to 4% on evaluated benchmarks.

Enhanced accuracy by over 2% in GSM8K mathematical reasoning distillation.

Abstract

Computing next-token likelihood ratios between two language models (LMs) is a standard task in training paradigms such as knowledge distillation. Since this requires both models to share the same probability space, it becomes challenging when the teacher and student LMs use different tokenizers, for instance, when edge-device deployment necessitates a smaller vocabulary size to lower memory overhead. This work addresses this vocabulary misalignment problem by uncovering an implicit recursive structure in the commonly deployed Byte-Pair Encoding (BPE) algorithm and utilizing it to create a probabilistic framework for cross-tokenizer likelihood scoring. Our method enables sequence likelihood evaluation for vocabularies different from the teacher model native tokenizer, addressing two specific scenarios: when the student vocabulary is a subset of the teacher vocabulary, and the general case where it is arbitrary. In the subset regime, our framework computes exact likelihoods and provides next-token probabilities for sequential sampling with only ${O}(1)$ model evaluations per token. When used for distillation, this yields up to a $12\%$ reduction in memory footprint for the Qwen2.5-1.5B model while also improving baseline performance up to $4\%$ on the evaluated tasks. For the general case, we introduce a rigorous lossless procedure that leverages BPE recursive structure, complemented by a fast approximation that keeps large-vocabulary settings practical. Applied to GSM8K mathematical reasoning distillation, our method improves accuracy by over $2\%$ the current state of the art. Code: github.com/truongbuu/cross-tokenizer-scoring