Centered Self-Attention Layers
Not on the map. Why?
Abstract (arXiv)
The self-attention mechanism in transformers and the message-passing mechanism in graph neural networks are repeatedly applied within deep learning architectures. We show that this application inevitably leads to oversmoothing, i.e., to similar representations at the deeper layers for different tokens in transformers and different nodes in graph neural networks. Based on our analysis, we present a correction term to the aggregating operator of these mechanisms. Empirically, this simple term eliminates much of the oversmoothing problem in visual transformers, obtaining performance in weakly supervised segmentation that surpasses elaborate baseline methods that introduce multiple auxiliary networks and training phrases. In graph neural networks, the correction term enables the training of very deep architectures more effectively than many recent solutions to the same problem.
Key facts. Cited by 7 papers in the map (data through 14 Oct 2026). Builds on 56 of its 62 references that are in the map. First posted on arXiv 2 Jun 2023.
Builds on
56 of 62 references.
References · 56
- 2017Attention Is All You Needcited by 60,361
- 2018BERT: Pre-training of Deep Bidirectional Transformers for Language Understandingcited by 40,399
- 2020Language Models are Few-Shot Learnerscited by 32,419
- 2020An Image is Worth 16x16 Words: Transformers for Image Recognition at Scalecited by 30,186
- 2019RoBERTa: A Robustly Optimized BERT Pretraining Approachcited by 13,372
- 2016Semi-Supervised Classification with Graph Convolutional Networkscited by 11,361
- 2017Graph Attention Networkscited by 8,923
- 2016Layer Normalizationcited by 5,669
- 2022PaLM: Scaling Language Modeling with Pathwayscited by 5,326
- 2020Training data-efficient image transformers & distillation through attentioncited by 4,586
- 2019ALBERT: A Lite BERT for Self-supervised Learning of Language Representationscited by 2,790
- 2018Neural Tangent Kernel: Convergence and Generalization in Neural Networkscited by 2,621
38 more
- 2020On Layer Normalization in the Transformer Architecturecited by 922
- 2021Segmenter: Transformer for Semantic Segmentationcited by 855
- 2017Deep Neural Networks as Gaussian Processescited by 622
- 2021ImageNet-21K Pretraining for the Massescited by 613
- 2019Multi-scale Attributed Node Embeddingcited by 548
- 2019PairNorm: Tackling Oversmoothing in GNNscited by 297
- 2021GRAND: Graph Neural Diffusioncited by 258
- 2020A Note on Over-Smoothing for Graph Neural Networkscited by 249
- 2023A Survey on Oversmoothing in Graph Neural Networkscited by 247
- 2020How Neural Networks Extrapolate: From Feedforward to Graph Neural Networkscited by 222
- 2020Understanding the Difficulty of Training Transformerscited by 210
- 2021Training Graph Neural Networks with 1000 Layerscited by 169
- 2020Single-Stage Semantic Segmentation from Image Labelscited by 167
- 2019Transformers without Tears: Improving the Normalization of Self-Attentioncited by 153
- 2022Multi-class Token Transformer for Weakly Supervised Semantic Segmentationcited by 138
- 2022Anti-Oversmoothing in Deep Vision Transformers via the Fourier Domain Analysis: From Theory to Practicecited by 136
- 2021Railroad is not a Train: Saliency as Pseudo-pixel Supervision for Weakly Supervised Semantic Segmentationcited by 127
- 2019Pixel-Adaptive Convolutional Neural Networkscited by 126
- 2018On Regularized Losses for Weakly-supervised CNN Segmentationcited by 121
- 2022Learning Affinity from Attention: End-to-End Weakly-Supervised Semantic Segmentation with Transformerscited by 118
- 2022Not too little, not too much: a theoretical analysis of graph (over)smoothingcited by 106
- 2019Reliability Does Matter: An End-to-End Weakly Supervised Semantic Segmentation Approachcited by 88
- 2021Weakly Supervised Semantic Segmentation by Pixel-to-Prototype Contrastcited by 85
- 2019Attention Guided Graph Convolutional Networks for Relation Extractioncited by 82
- 2022Revisiting Over-smoothing in BERT from the Perspective of Graphcited by 72
- 2021Reducing Information Bottleneck for Weakly Supervised Semantic Segmentationcited by 65
- 2023Token Contrast for Weakly-Supervised Semantic Segmentationcited by 65
- 2022Regional Semantic Contrast and Aggregation for Weakly Supervised Semantic Segmentationcited by 64
- 2019Self-Supervised Difference Detection for Weakly-Supervised Semantic Segmentationcited by 58
- 2022Weakly Supervised Semantic Segmentation using Out-of-Distribution Datacited by 58
- 2022L2G: A Simple Local-to-Global Knowledge Transfer Framework for Weakly Supervised Semantic Segmentationcited by 57
- 2021Optimization of Graph Neural Networks: Implicit Acceleration by Skip Connections and More Depthcited by 55
- 2019Improving Transformer Models by Reordering their Sublayerscited by 52
- 2022A Non-Asymptotic Analysis of Oversmoothing in Graph Neural Networkscited by 49
- 2021Lipschitz Normalization for Self-Attention Layers with Application to Graph Neural Networkscited by 39
- 2023ContraNorm: A Contrastive Learning Perspective on Oversmoothing and Beyondcited by 39
- 2022Expansion and Shrinkage of Localization for Weakly-Supervised Semantic Segmentationcited by 37
- 2023ResiDual: Transformer with Dual Residual Connectionscited by 31
50 most cited of 56 All 56 (JSON)
Built on by
Citing papers per year: 2023: 2; 2024: 2; 2025: 2; 2026: 1.
7 papers.
- 2024The Hidden Attention of Mamba Modelscited by 91
- 2023Simplifying Transformer Blockscited by 37
- 2024Setting the Record Straight on Transformer Oversmoothingcited by 21
- 2023Graph Convolutions Enrich the Self-Attention in Transformers!cited by 11
- 2025Latent Mamba Operator for Partial Differential Equationscited by 10
- 2026Interpreting and Steering State-Space Models via Activation Subspace Bottleneckscited by 3
- 2025Understanding Differential Transformer Unchains Pretrained Self-Attentionscited by 0
- 2026Interpreting and Steering State-Space Models via Activation Subspace Bottlenecks26 Feb 2026
- 2025Latent Mamba Operator for Partial Differential Equations25 May 2025
- 2025Understanding Differential Transformer Unchains Pretrained Self-Attentions22 May 2025
- 2024The Hidden Attention of Mamba Models3 Mar 2024
- 2024Setting the Record Straight on Transformer Oversmoothing9 Jan 2024
- 2023Graph Convolutions Enrich the Self-Attention in Transformers!7 Dec 2023
- 2023Simplifying Transformer Blocks3 Nov 2023