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.
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