English

Spelling Bee Embeddings for Language Modeling

Machine Learning 2026-01-27 v1

Abstract

We introduce a simple modification to the embedding layer. The key change is to infuse token embeddings with information about their spelling. Models trained with these embeddings improve not only on spelling, but also across standard benchmarks. We conduct scaling studies for models with 40M to 800M parameters, which suggest that the improvements are equivalent to needing about 8% less compute and data to achieve the same test loss.

Keywords

Cite

@article{arxiv.2601.18030,
  title  = {Spelling Bee Embeddings for Language Modeling},
  author = {Markus N. Rabe and Judith Clymo and Zheren Dong},
  journal= {arXiv preprint arXiv:2601.18030},
  year   = {2026}
}
R2 v1 2026-07-01T09:19:29.553Z