Attention Sinks Induce Gradient Sinks: Massive Activations as Gradient Regulators in Transformers
cs.LG, cs.AI
Submitted: 2026-03-18
Updated: 2026-10-19
Comments: 30 pages, 15 figures, 9 tables
Code: https://github.com/LARS-research/GradientSink
License: http://creativecommons.org/licenses/by/4.0/
The gist: Attention sinks and massive activations are recurring and closely related phenomena in Transformer models.
Terminology
Abstract
Attention sinks and massive activations are recurring and closely related phenomena in Transformer models. Existing explanations have largely focused on the forward pass, yet in pre-norm Transformers, large residual-stream norms play only an indirect forward role because sublayers operate on normalized inputs. We study this relationship from the perspective of backpropagation. Empirically and theoretically, we show that under causal masking, attention sinks can induce pronounced gradient concentration, which we term gradient sinks. Since the RMSNorm Jacobian attenuates gradients roughly in inverse proportion to input norm, massive activations can be understood as adaptive regulators of this localized gradient pressure during training. This interpretation predicts that attenuating sink-induced gradients should weaken massive activations. We test this prediction with V-scale, a modification that adjusts backpropagated gradients on the value paths. In V-scale models and a backward-only control, attention sinks are preserved, whereas massive activations are suppressed. These results identify gradient sinks as a backward-pass counterpart of attention sinks, and massive activations as an adaptive RMSNorm-mediated response that attenuates the resulting localized training pressure. Our code is available at https://github.com/LARS-research/GradientSink.
Sources
- Layer Normalization
- Value-State Gated Attention for Mitigating Extreme-Token Phenomena in Transformers
- Attention Sink Forges Native MoE in Attention Layers: Sink-Aware Training to Address Head Collapse
- Hidden Dynamics of Massive Activations in Transformer Training
- The Llama 3 Herd of Models
- An Empirical Model of Large-Batch Training
- A Refined Analysis of Massive Activations in LLMs
- A Unified View of Attention and Residual Sinks: Outlier-Driven Rescaling is Essential for Transformer Training
- GLU Variants Improve Transformer
- The Spike, the Sparse and the Sink: Anatomy of Massive Activations and Attention Sinks
- Llama 2: Open Foundation and Fine-Tuned Chat Models
- Qwen2.5 Technical Report
- Qwen3 Technical Report
- ChatGLM: A Family of Large Language Models from GLM-130B to GLM-4 All Tools
- Softpick: No Attention Sink, No Massive Activations with Rectified Softmax
Related papers
- Polynomial-Augmented Neural Networks (PANNs) with Weak Orthogonality Constraints for Enhanced Function and PDE Approximation
- AIRL-S: Unifying Reinforcement Learning and Search-Based Test-Time Scaling via Adversarial Inverse Reinforcement Learning
- Transformers as Bayesian In-Context Experimenters: Smoothness-Adaptive Efficient ATE Estimation
- Convergence issues in Relational Concept Analysis based on AOC-posets
- Beliefs Beyond Posteriors: The Predictive Cost of ELBO Binding
- Understanding Diffusion Models via Ratio-Based Function Approximation with SignReLU Networks