Scaling Bidirectional Spans and Span Violations in Attention Mechanism
Not on the map. Why?
Abstract (arXiv)
The canonical O(N2) Transformer remains the empirical performance frontier in sequence modeling, and its training can be further optimized by addressing geometric inefficiency. We propose an optimization framework that leverages an asymmetric projection to decompose the backward-pass gradients into parallel spans and orthogonal violations, while keeping the canonical forward-pass QKV structure intact. Through consistent experimental validation across various decomposition and projection setups, we provide strong theoretical evidence: the standard attention gradient is suboptimal. We demonstrated that selectively scaling these components, focusing primarily on 0th order bidirectional parallel spans, yields the most effective learning signal. On the limited WikiText-2 dataset, and using a crude configuration, this method achieved a 0.56% reduction in validation loss, confirming the framework's fundamental validity and suggesting significant potential gains on larger datasets and deeper training regimes
Key facts. No citing papers in the map yet (data through 7 Oct 2026). Builds on 5 of its 6 references that are in the map. First posted on arXiv 15 Dec 2025.
Builds on
5 of 6 references.
References · 5
- 2017Attention Is All You Needcited by 59,932
- 2020A Primer in BERTology: What we know about how BERT workscited by 807
- 2019Are Sixteen Heads Really Better than One?cited by 803
- 2019Why gradient clipping accelerates training: A theoretical justification for adaptivitycited by 335
- 2024Transformers Get Stable: An End-to-End Signal Propagation Theory for Language Modelscited by 17
Built on by
Not cited yet.