Linear Algebra Foundations of Efficient Attention: A Phase Reversal in Rank Collapse Under SVD Compression
Abstract
Linear algebra provides the framework of concepts (matrix rank, singular value decomposition (SVD), and eigendecomposition) that modern artificial intelligence employs to encode, compress, and propagate information through neural networks. This paper unifies fourteen separate peer-reviewed works analyzing the usage of these techniques in the context of transformer-based foundation model research, focusing on three areas of the topic: derivations and properties of self-attention matrices' output rank, compression methods that purposefully utilize this phenomenon, and the low-rank key-value (KV) cache projection and its semiseparable-matrix duality to linear attention and state-space structured models. We were motivated to conduct this work after observing an open problem in this literature: the interplay of the mentioned compression methods with natural rank collapse of the network. With this paper, we report an original finding that using SVD compression of attention projections actually has the opposite effect on the rank collapse of the network: while it strongly suppresses it at initialization, it accelerates on pretrained models (for GPT-2 124M, GPT-2 Medium 355M, and Pythia-160M) with minimal risk of object aliasing artifacts appearing (verified on all compression ratios) and is consistent across four rank estimation methods. A controlled causal decomposition of the effect in both settings showed that the reason for this behavior can be explained by the choice of the subspace SVD makes when compressing the matrix better than the reduction of the operator norm it achieves, explaining roughly 76% of the effect at initialization and 83% on the pretrained weights, providing a refinement to the calibration-aware compression viewpoint and an explanation of why it outperformed naive SVD truncation.
Figures & tables
| Study | Method | Reported Effect |
|---|---|---|
| Hu et al. [ 7 ] | LoRA | 10,000 fewer trainable params vs. full fine-tuning; no added inference latency |
| Saxena et al. [ 12 ] | Eigen Attention | Up to 40% KV-cache reduction; up to 60% attention-latency reduction |
| Kusupati et al. [ 17 ] | MRL | Up to 14 smaller embeddings; up to 14 retrieval speed-up |
| Wang et al. [ 11 ] | SVD-LLM | Improved accuracy retention vs. magnitude-based SVD at high compression |
| Chang et al. [ 14 ] | PALU | Reduced KV-cache memory with smaller accuracy loss than uniform-rank baselines |
| Dong et al. [ 4 ] | Rank-collapse bound | Doubly-exponential convergence to rank-1, |
Low-Rank Compression
Low-rank compression reduces model size, memory use, and computation by representing weights, activations, or caches with compact factors.
See all 108 papers