← Search

Andy Yang

5 accepted papers

2026

Length Generalization Bounds for Transformers

ICML 2026poster

Length generalization is a key property of a learning algorithm that enables it to make correct predictions on inputs of unbounded length, given finite training data. To provide such a guarantee, one needs to be able to compute a length generalization bound, beyond which the model is guaranteed to g…

Cited by 3SourceScholar
2026

Probability Distributions Computed by Autoregressive Transformers

ICLR 2026poster

Most expressivity results for transformers treat them as language recognizers (which accept or reject strings), and not as they are used in practice, as language models (which generate strings autoregressively and probabilistically). We characterize the probability distributions that transformer lan…

Cited by 0SourceScholar
2025

A Formal Framework for Understanding Length Generalization in Transformers

ICLR 2025poster

A major challenge for transformers is generalizing to sequences longer than those observed during training. While previous works have empirically shown that transformers can either succeed or fail at length generalization depending on the task, theoretical understanding of this phenomenon remains li…

2024

Masked Hard-Attention Transformers Recognize Exactly the Star-Free Languages

NeurIPS 2024poster

The expressive power of transformers over inputs of unbounded size can be studied through their ability to recognize classes of formal languages. In this paper, we establish exact characterizations of transformers with hard attention (in which all attention is focused on exactly one position) and at…

Cited by 15SourcePDFScholar