English

Provable Length Generalization in Sequence Prediction via Spectral Filtering

Machine Learning 2024-11-05 v1 Artificial Intelligence Computation and Language

Abstract

We consider the problem of length generalization in sequence prediction. We define a new metric of performance in this setting -- the Asymmetric-Regret -- which measures regret against a benchmark predictor with longer context length than available to the learner. We continue by studying this concept through the lens of the spectral filtering algorithm. We present a gradient-based learning algorithm that provably achieves length generalization for linear dynamical systems. We conclude with proof-of-concept experiments which are consistent with our theory.

Keywords

Cite

@article{arxiv.2411.01035,
  title  = {Provable Length Generalization in Sequence Prediction via Spectral Filtering},
  author = {Annie Marsden and Evan Dogariu and Naman Agarwal and Xinyi Chen and Daniel Suo and Elad Hazan},
  journal= {arXiv preprint arXiv:2411.01035},
  year   = {2024}
}

Comments

34 pages, 9 figures

R2 v1 2026-06-28T19:45:05.755Z