English

Perturbation is All You Need for Extrapolating Language Models

Machine Learning 2026-05-07 v1 Machine Learning Statistics Theory Statistics Theory

Abstract

We introduce a simple yet powerful framework for training large language models. In contrast to the standard autoregressive next-token prediction based on an exact prefix, we propose a perturbation-based procedure that first transforms the prefix into a semantic neighbor and then conditions on this perturbed variant for next-token prediction. This yields a hierarchical model with a pre-post-additive noise structure. Within this framework, we develop a rigorous theory of extrapolability, namely, the capacity of a model class to make reliable predictions for token sequences that lie outside the empirical support of the training corpus. We evaluate the finite-sample performance of the proposed procedure using both synthetic and real-world language data. Results show that the proposed method consistently improves out-of-support prediction while maintaining competitive in-support performance, demonstrating that perturbation offers a practical route to language modeling.

Keywords

Cite

@article{arxiv.2605.04344,
  title  = {Perturbation is All You Need for Extrapolating Language Models},
  author = {Zetai Cen and Jin Zhu and Xinwei Shen and Chengchun Shi},
  journal= {arXiv preprint arXiv:2605.04344},
  year   = {2026}
}

Comments

44 pages

R2 v1 2026-07-01T12:51:55.838Z