Subword Segmental BabyLMs: Learning to Tokenise for Sample-Efficient Pretraining

Francois Meyer


Abstract
In the standard LM training pipeline, subword tokenisation is applied as a preprocessing step. Subword segmental language modelling is an alternative paradigm in which tokenisation is learned during training, allowing the model to discover subword units that optimise its training objective. In this paper, we present our submission to the 2026 BabyLM Challenge, for which we develop two new subword segmental LMs: SubSegGPT and SubSegDeBERTa. SubSegGPT is a decoder-only model that learns tokenisation during autoregressive pretraining. SubSegDeBERTa is an encoder-based model that jointly learns to generate and tokenise masked words. We train both for the Strict and Strict-small tracks. Our top submission to Strict is SubSegDeBERTa, which achieves notable gains in zero-shot evaluation. Our top submission to Strict-small is SubSegGPT, which outperforms tokenisation-based baselines. Our results show that learnable subword tokenisation can improve sample-efficiency for BabyLM pretraining. We analyse the subword learning dynamics of our models and find that tokenisation gradually converges on subword units that balance morphological alignment and fine-grained segmentation.
Anthology ID:
2026.babylm-main.46
Volume:
Proceedings of the Second BabyLM Workshop
Month:
October
Year:
2026
Address:
Budapest, Hungary
Editors:
Leshem Choshen, Ryan Cotterell, Mustafa Omer Gul, Jaap Jumelet, Tal Linzen, Aaron Mueller, Suchir Salhan, Raj Sanjay Shah, Alex Warstadt, Ethan Gotlieb Wilcox
Venues:
BabyLM | WS
SIG:
Publisher:
Association for Computational Linguistics
Note:
Pages:
585–597
Language:
URL:
https://aclanthology.org/2026.babylm-main.46/
DOI:
Bibkey:
Cite (ACL):
Francois Meyer. 2026. Subword Segmental BabyLMs: Learning to Tokenise for Sample-Efficient Pretraining. In Proceedings of the Second BabyLM Workshop, pages 585–597, Budapest, Hungary. Association for Computational Linguistics.
Cite (Informal):
Subword Segmental BabyLMs: Learning to Tokenise for Sample-Efficient Pretraining (Meyer, BabyLM 2026)
Copy Citation:
PDF:
https://aclanthology.org/2026.babylm-main.46.pdf