XpertGPT: Multi-Scale Sparse Expert Routing for Data-Constrained Language Modeling

Soham Jain, Harsh Singh, Divija Dewan, Atul Dev


Abstract
We introduce XpertGPT, a sparse decoder-only Transformer for data-constrained language modelling that routes tokens to parallel expert pathways operating at multiple contextual scales. Under the BabyLM strict-small budget of roughly ten million words, conventional dense Transformers spend identical computation on every token and impose a single fixed receptive field. Each XpertGPT layer pairs a dense global-attention stream with four sparse sliding-window expert branches, whose attention spans range from 64 down to 4 tokens, chosen by Expert-Choice routing that balances expert load by construction without any auxiliary loss, and applies rotary position embeddings and gated feed-forward networks throughout. Trained on the BabyLM 2026 Strict-Small corpus, our 52.6M-parameter model reaches an overall average of 38.42, surpassing the official GPT-2(98.4M) baseline (37.38) across grammatical acceptability, world-knowledge, and language-understanding benchmarks.
Anthology ID:
2026.babylm-main.60
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:
767–778
Language:
URL:
https://aclanthology.org/2026.babylm-main.60/
DOI:
Bibkey:
Cite (ACL):
Soham Jain, Harsh Singh, Divija Dewan, and Atul Dev. 2026. XpertGPT: Multi-Scale Sparse Expert Routing for Data-Constrained Language Modeling. In Proceedings of the Second BabyLM Workshop, pages 767–778, Budapest, Hungary. Association for Computational Linguistics.
Cite (Informal):
XpertGPT: Multi-Scale Sparse Expert Routing for Data-Constrained Language Modeling (Jain et al., BabyLM 2026)
Copy Citation:
PDF:
https://aclanthology.org/2026.babylm-main.60.pdf