English

Tokenisation via Convex Relaxations

Computation and Language 2026-05-22 v1 Machine Learning

Abstract

Tokenisation is an integral part of the current NLP pipeline. Current tokenisation algorithms such as BPE and Unigram are greedy algorithms -- they make locally optimal decisions without considering the resulting vocabulary as a whole. We instead formulate tokeniser construction as a linear program and solve it using convex optimisation tools, yielding a new algorithm we call ConvexTok. We find ConvexTok consistently improves intrinsic tokenisation metrics and the bits-per-byte (BpB) achieved by language models; it also improves downstream task performance, but less consistently. Furthermore, ConvexTok allows the user to certify how far their tokeniser is from optimal, with respect to a certain objective, via a lower bound, and we empirically find it to be within 1\% of optimal at common vocabulary sizes.

Keywords

Cite

@article{arxiv.2605.22821,
  title  = {Tokenisation via Convex Relaxations},
  author = {Jan Tempus and Philip Whittington and Craig W. Schmidt and Dennis Komm and Tiago Pimentel},
  journal= {arXiv preprint arXiv:2605.22821},
  year   = {2026}
}