mlsanityData to 14 Oct 2026

On Generalisation Error Bounds for Transformers

arXiv 2410.11500 ↗First posted 15 Oct 2024stat.ML cs.LG math.FAcited by 4

Not on the map. Why?

Abstract (arXiv)

In this paper, we establish a collection of covering number bounds for linear function classes under various norm constraints on the inputs and matrices. We then combine these results with existing covering number bounds to derive improved estimates and, based on these estimates, develop generalization error bounds for single-layer Transformers. The resulting generalization bounds improve upon several existing results in the literature and, in particular, are independent of the input sequence length. Moreover, our generalization error bound decays at the rate O(1/√n), where n denotes the sample size, thereby improving upon existing bounds that scale as O((log n)/√n). Furthermore, our covering number analysis explicitly incorporates rank constraints on the underlying matrix classes, allowing us to characterize how low-rank structures affect the metric entropy and, consequently, the resulting generalization bounds for Transformer architectures.

Key facts. Cited by 4 papers in the map (data through 14 Oct 2026). Builds on 12 of its 24 references that are in the map. First posted on arXiv 15 Oct 2024.

Builds on

12 of 24 references.

References · 12

  1. 2017Attention Is All You Needcited by 60,361
  2. 2020An Image is Worth 16x16 Words: Transformers for Image Recognition at Scalecited by 30,186
  3. 2017Spectrally-normalized margin bounds for neural networkscited by 686
  4. 2017A PAC-Bayesian Approach to Spectrally-Normalized Margin Bounds for Neural Networkscited by 314
  5. 2015Norm-Based Capacity Control in Neural Networkscited by 299
  6. 2017Size-Independent Sample Complexity of Neural Networkscited by 290
  7. 2021Inductive Biases and Variable Creation in Self-Attention Mechanismscited by 144
  8. 2024LoRA Training in the NTK Regime has No Spurious Local Minimacited by 48
  9. 2023Are Transformers with One Layer Self-Attention Using Low-Rank Weight Matrices Universal Approximators?cited by 47
  10. 2023Sequence Length Independent Norm-Based Generalization Bounds for Transformerscited by 29
  11. 2022On Rademacher Complexity-based Generalization Bounds for Deep Learningcited by 16
  12. 2024Learning on Transformers is Provable Low-Rank and Sparse: A One-layer Analysiscited by 8

Built on by

Cited by year (2026 to 14 Oct)

Citing papers per year: 2025: 2; 2026: 2.

4 papers.