ICLR 2026poster0 citations

Lossless Vocabulary Reduction for Auto-Regressive Language Models

Daiki Chijiwa, Taku Hasegawa, Kyosuke Nishida, Shin'ya Yamaguchi, Tomoya Ohba, Tamao Sakao, Susumu Takeuchi

Abstract

Tokenization---the process of decomposing a given text into a sequence of subwords called tokens---is one of the key components in the development of language models. Particularly, auto-regressive language models generate texts token by token, i.e., by predicting the next-token distribution given the previous ones, and thus tokenization directly affects their efficiency in text generation. Since each language model has their own vocabulary as a set of possible tokens, they struggle to cooperate with each other at the level of next-token distributions such as model ensemble. In this paper, we establish a theoretical framework of lossless vocabulary reduction, which efficiently converts a given auto-regressive language model into the one with an arbitrarily small vocabulary without any loss in accuracy. As an application, we demonstrate that language models with different tokenization can cooperate with each other efficiently through their maximal common vocabulary.

Language ModelsNext-Token DistributionTokenizationVocabulary
BibTeX
@inproceedings{
chijiwa2026lossless,
title={Lossless Vocabulary Reduction for Auto-Regressive Language Models},
author={Daiki Chijiwa and Taku Hasegawa and Kyosuke Nishida and Shin'ya Yamaguchi and Tomoya Ohba and Tamao Sakao and Susumu Takeuchi},
booktitle={The Fourteenth International Conference on Learning Representations},
year={2026},
url={https://openreview.net/forum?id=xAvqHtLVgz}
}