13. SWA: Sliding Window Attention / Longformer (2020, Beltagy et al.)
arXiv: 2004.05150
Core Problem
Sparse Transformer's sparse patterns are somewhat complex (alternating strided + fixed). Can we make it simpler? For example, let each word attend only to words within a fixed window around itself? The attention matrix then becomes a "banded" matrix, with each position having only w nonzero elements, giving total complexity O(n·w).
Method Innovation
Longformer's SWA is extremely simple: each token only attends to w neighbors on its left and right.
But there is a crucial addition — global attention: certain special tokens (e.g., [CLS], section markers) can attend to all positions and are attended to by all positions. This guarantees aggregation of long-range information.
Implementation: 1. Local windowed attention: each position i attends only to positions within [i-w, i+w]. O(n·w). 2. Global attention: designated "global tokens" can attend to all positions, and all positions can attend to them. 3. Combined: the local component handles fine-grained relations, the global component handles coarse-grained aggregation.
Key Numbers
- Achieves SOTA on character-level LM (text8, enwik8)
- After pretraining, Longformer "consistently outperforms RoBERTa" on long-document tasks
- New SOTA on WikiHop and TriviaQA
- LED (Longformer-Encoder-Decoder) is effective on arXiv summarization
Impact Assessment
SWA is the most practical sparse attention scheme. Longformer can directly replace the standard Transformer attention layer as a "drop-in replacement." Later models (such as Gemma 2 and various long-context LLMs) adopted SWA or its variants. Its core idea — "locally dense + globally sparse" — has become the standard paradigm for long-sequence modeling.
Feynman-style Commentary
> SWA's way of thinking is the "proximity assumption" — in language, related things tend to be nearby. This is not an absolute truth (long-range dependencies do exist), but it is a strong statistical assumption. SWA turns this assumption into an architecture: each word only "cares about" its neighbors, but indirectly obtains global information through global tokens (like [CLS]). It's like a small-town gossip network — everyone chats only with neighbors, but the mail carrier (the global token) spreads messages across the whole town. Simple, efficient, effective. Feynman would say: a good physical model is not the most precise one, but one that is "precise enough yet computable." SWA is attention's "computable model."
---
arXiv: 2004.05150