# TREX: Tokenizer Regression for Optimal Data Mixture

Source: [TREX: Tokenizer Regression for Optimal Data Mixture](<https://www.sota2.com/research/paper/69705dbde25ad2fa9d1ed43a>)

Collection index: [llms.txt](<https://www.sota2.com/research/llms.txt>)

Data updated: 2026-01-22T05:00:31.014Z

Authors: Inho Won, Hangyeol Yoo, Minkyung Cho, Jungyeul Park, Hoyun Song, KyungTae Lim

Published: 2026-01-20T04:41:09.000Z

## Abstract

Building effective tokenizers for multilingual Large Language Models (LLMs) requires careful control over language-specific data mixtures. While a tokenizer's compression performance critically affects the efficiency of LLM training and inference, existing approaches rely on heuristics or costly large-scale searches to determine optimal language ratios. We introduce Tokenizer Regression for Optimal Data MiXture (TREX), a regression-based framework that efficiently predicts the optimal data mixture for tokenizer training. TREX trains small-scale proxy tokenizers on random mixtures, gathers their compression statistics, and learns to predict compression performance from data mixtures. This learned model enables scalable mixture search before large-scale tokenizer training, mitigating the accuracy-cost trade-off in multilingual tokenizer design. Tokenizers trained with TReX's predicted mixtures outperform mixtures based on LLaMA3 and uniform distributions by up to 12% in both inand out-of-distribution compression efficiency, demonstrating strong scalability, robustness, and practical effectiveness.

## Recorded benchmark results

A research result applies to its stated dataset, split, metric and evaluation configuration. Compare like-for-like settings, preserve the metric's unit and direction, and distinguish paper-reported results from independent reproduction. The best result recorded here is not a claim of exhaustive, current global coverage.

- [Cross-lingual Reasoning and Factual Knowledge on Global MMLU (test)](<https://www.sota2.com/research/sota/cross-lingual-reasoning-and-factual-knowledge-on-global-mmlu-test.md>): Cross-lingual Reasoning and Factual Knowledge; Global MMLU (test); Accuracy (RUS): 23.46.

## Original sources and implementation

[Original paper](<https://arxiv.org/abs/2601.13588>)

[PDF](<https://arxiv.org/pdf/2601.13588.pdf>)
