Dense Local Dependencies Induce Attention-Logit Explosion and Training Instability During Long-Sequence Transformer Training
Not on the map. Why?
Abstract (arXiv)
Autoregressive transformer language models frequently exhibit training instability when trained on long sequences, particularly under low-precision arithmetic. Although this instability is often accompanied by attention-logit explosion, its underlying cause remains poorly understood. In this work, we present analytical insights and empirical evidence that dense local dependencies are a major contributor to attention-logit explosion. We demonstrate that dense local dependency patterns yield an effectively high-rank attention structure, which the low-rank parameterization of self-attention can only approximate with increasingly large logits as the sequence length grows. This logit inflation ultimately leads to training instability under low-precision arithmetic. We support this explanation through extensive experiments on synthetic and language modeling tasks. Our results consistently show that attention-logit growth increases with sequence length, is mitigated by increasing the attention dimension, and is substantially reduced by explicitly modeling dense local dependencies. Furthermore, we show that this growth is driven by the density of local dependencies rather than by locality alone. More broadly, our findings suggest that explicitly modeling dense local dependencies constitutes an important design principle for developing stable, efficient, and scalable long-context transformer architectures for autoregressive language modeling.
Key facts. Cited by 2 papers in the map (data through 14 Oct 2026). Builds on 17 of its 27 references that are in the map. First posted on arXiv 21 May 2025.
Builds on
17 of 27 references.
References · 17
- 2017Attention Is All You Needcited by 60,361
- 2014Adam: A Method for Stochastic Optimizationcited by 37,240
- 2021RoFormer: Enhanced Transformer with Rotary Position Embeddingcited by 4,703
- 2020wav2vec 2.0: A Framework for Self-Supervised Learning of Speech Representationscited by 3,905
- 2020Longformer: The Long-Document Transformercited by 3,413
- 2021HuBERT: Self-Supervised Speech Representation Learning by Masked Prediction of Hidden Unitscited by 2,342
- 2020Conformer: Convolution-augmented Transformer for Speech Recognitioncited by 1,863
- 2019Generating Long Sequences with Sparse Transformerscited by 1,546
- 2021Train Short, Test Long: Attention with Linear Biases Enables Input Length Extrapolationcited by 1,103
- 2019Compressive Transformers for Long-Range Sequence Modellingcited by 726
- 2019Libri-Light: A Benchmark for ASR with Limited or No Supervisioncited by 467
- 2024MInference 1.0: Accelerating Pre-filling for Long-Context LLMs via Dynamic Sparse Attentioncited by 383
5 more
- 2020Query-Key Normalization for Transformerscited by 280
- 2023Small-scale proxies for large-scale Transformer training instabilitiescited by 183
- 2020Low-Rank Bottleneck in Multi-head Attention Modelscited by 89
- 2024Transformers Get Stable: An End-to-End Signal Propagation Theory for Language Modelscited by 18
- 2025Zero-Shot Styled Text Image Generation, but Make It Autoregressivecited by 13
Built on by
Citing papers per year: 2025: 1; 2026: 1.
2 papers.