ContraNorm: A Contrastive Learning Perspective on Oversmoothing and Beyond
Not on the map. Why?
Abstract (arXiv)
Oversmoothing is a common phenomenon in a wide range of Graph Neural Networks (GNNs) and Transformers, where performance worsens as the number of layers increases. Instead of characterizing oversmoothing from the view of complete collapse in which representations converge to a single point, we dive into a more general perspective of dimensional collapse in which representations lie in a narrow cone. Accordingly, inspired by the effectiveness of contrastive learning in preventing dimensional collapse, we propose a novel normalization layer called ContraNorm. Intuitively, ContraNorm implicitly shatters representations in the embedding space, leading to a more uniform distribution and a slighter dimensional collapse. On the theoretical analysis, we prove that ContraNorm can alleviate both complete collapse and dimensional collapse under certain conditions. Our proposed normalization layer can be easily integrated into GNNs and Transformers with negligible parameter overhead. Experiments on various real-world datasets demonstrate the effectiveness of our proposed ContraNorm. Our implementation is available at https://github.com/PKU-ML/ContraNorm.
Key facts. Cited by 39 papers in the map (data through 14 Oct 2026). Builds on 55 of its 69 references that are in the map. First posted on arXiv 12 Mar 2023.
Builds on
55 of 69 references.
References · 55
- 2017Attention Is All You Needcited by 60,361
- 2018BERT: Pre-training of Deep Bidirectional Transformers for Language Understandingcited by 40,399
- 2014Adam: A Method for Stochastic Optimizationcited by 37,240
- 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
- 2021Swin Transformer: Hierarchical Vision Transformer using Shifted Windowscited by 12,139
- 2016Semi-Supervised Classification with Graph Convolutional Networkscited by 11,361
- 2014ImageNet Large Scale Visual Recognition Challengecited by 8,498
- 2019Momentum Contrast for Unsupervised Visual Representation Learningcited by 7,513
- 2015Batch Normalization: Accelerating Deep Network Training by Reducing Internal Covariate Shiftcited by 7,005
- 2016Layer Normalizationcited by 5,669
- 2018GLUE: A Multi-Task Benchmark and Analysis Platform for Natural Language Understandingcited by 4,899
38 more
- 2020Bootstrap your own latent: A new approach to self-supervised Learningcited by 4,750
- 2020Training data-efficient image transformers & distillation through attentioncited by 4,586
- 2020Exploring Simple Siamese Representation Learningcited by 2,866
- 2019ALBERT: A Lite BERT for Self-supervised Learning of Language Representationscited by 2,790
- 2020Open Graph Benchmark: Datasets for Machine Learning on Graphscited by 2,163
- 2021Barlow Twins: Self-Supervised Learning via Redundancy Reductioncited by 1,838
- 2018Graph Neural Networks: A Review of Methods and Applicationscited by 1,576
- 2018Know What You Don't Know: Unanswerable Questions for SQuADcited by 1,554
- 2020Understanding Contrastive Representation Learning through Alignment and Uniformity on the Hyperspherecited by 1,405
- 2018Graph Convolutional Neural Networks for Web-Scale Recommender Systemscited by 1,138
- 2018Deeper Insights into Graph Convolutional Networks for Semi-Supervised Learningcited by 1,102
- 2021VICReg: Variance-Invariance-Covariance Regularization for Self-Supervised Learningcited by 1,057
- 2020Simple and Deep Graph Convolutional Networkscited by 860
- 2021Segmenter: Transformer for Semantic Segmentationcited by 855
- 2019How Contextual are Contextualized Word Representations? Comparing the Geometry of BERT, ELMo, and GPT-2 Embeddingscited by 784
- 2019Multi-scale Attributed Node Embeddingcited by 548
- 2021Attention is Not All You Need: Pure Attention Loses Rank Doubly Exponentially with Depthcited by 419
- 2021Understanding Dimensional Collapse in Contrastive Self-supervised Learningcited by 378
- 2021XCiT: Cross-Covariance Image Transformerscited by 338
- 2019PairNorm: Tackling Oversmoothing in GNNscited by 297
- 2021DeepViT: Towards Deeper Vision Transformercited by 289
- 2020Whitening for Self-Supervised Representation Learningcited by 279
- 2019Representation Degeneration Problem in Training Natural Language Generation Modelscited by 255
- 2021Training Graph Neural Networks with 1000 Layerscited by 169
- 2021On Feature Decorrelation in Self-Supervised Learningcited by 160
- 2022Contrastive and Non-Contrastive Self-Supervised Learning Recover Global and Local Spectral Embedding Methodscited by 149
- 2021Node Feature Extraction by Self-Supervised Multi-scale Neighborhood Predictioncited by 138
- 2022Anti-Oversmoothing in Deep Vision Transformers via the Fourier Domain Analysis: From Theory to Practicecited by 136
- 2021Interpreting and Unifying Graph Neural Networks with An Optimization Frameworkcited by 131
- 2022On the duality between contrastive and non-contrastive self-supervised learningcited by 108
- 2021Towards the Generalization of Contrastive Self-Supervised Learningcited by 99
- 2019Iterative Normalization: Beyond Standardization towards Efficient Whiteningcited by 92
- 2018Decorrelated Batch Normalizationcited by 83
- 2022Revisiting Over-smoothing in BERT from the Perspective of Graphcited by 72
- 2021Graph Neural Networks Inspired by Classical Iterative Algorithmscited by 60
- 2021Exploring the Equivalence of Siamese Self-Supervised Learning via A Unified Gradient Frameworkcited by 49
- 2021Vision Transformers with Patch Diversificationcited by 45
- 2022Optimization-Induced Graph Implicit Nonlinear Diffusioncited by 28
50 most cited of 55 All 55 (JSON)
Built on by
Citing papers per year: 2023: 10; 2024: 12; 2025: 16; 2026: 1.
39 papers.
Sort:
By area:
- 2024CATCH: Channel-Aware multivariate Time Series Anomaly Detection via Frequency Patchingcited by 94
- 2024Fredformer: Frequency Debiased Transformer for Time Series Forecastingcited by 45
- 2024Residual Connections and Normalization Can Provably Prevent Oversmoothing in GNNscited by 34
- 2024FairGT: A Fairness-aware Graph Transformercited by 29
- 2024Measure-to-measure interpolation using Transformerscited by 29
- 2023An Attentive Inductive Bias for Sequential Recommendation beyond the Self-Attentioncited by 28
- 2024Do Generated Data Always Help Contrastive Learning?cited by 25
- 2023Towards Deep Attention in Graph Neural Networks: Problems and Remediescited by 24
- 2024Rethinking Graph Masked Autoencoders through Alignment and Uniformitycited by 22
- 2024Setting the Record Straight on Transformer Oversmoothingcited by 21
- 2024Non-negative Contrastive Learningcited by 19
- 2023Rank Collapse Causes Over-Smoothing and Over-Correlation in Graph Neural Networkscited by 18
27 more
- 2023Identifiable Contrastive Learning with Automatic Feature Importance Discoverycited by 14
- 2023Graph Convolutions Enrich the Self-Attention in Transformers!cited by 11
- 2025Projection Head is Secretly an Information Bottleneckcited by 11
- 2024GTC: GNN-Transformer Co-contrastive Learning for Self-supervised Heterogeneous Graph Representationcited by 9
- 2023Centered Self-Attention Layerscited by 7
- 2023Architecture Matters: Uncovering Implicit Mechanisms in Graph Contrastive Learningcited by 7
- 2024Idempotent Unsupervised Representation Learning for Skeleton-Based Action Recognitioncited by 7
- 2025Only Large Weights (And Not Skip Connections) Can Prevent the Perils of Rank Collapsecited by 7
- 2025STAR: Boosting Time Series Foundation Models for Anomaly Detection through State-aware Adaptercited by 7
- 2025Deeper with Riemannian Geometry: Overcoming Oversmoothing and Oversquashing for Graph Foundation Modelscited by 7
- 2025Are We Measuring Oversmoothing in Graph Neural Networks Correctly?cited by 6
- 2025MOOSComp: Improving Lightweight Long-Context Compressor via Mitigating Over-Smoothing and Incorporating Outlier Scorescited by 4
- 2025Integrating Structural and Semantic Signals in Text-Attributed Graphs with BiGTexcited by 3
- 2025Transformers as Intrinsic Optimizers: Forward Inference through the Energy Principlecited by 3
- 2024Generalized Probabilistic Attention Mechanism in Transformerscited by 2
- 2025A Signed Graph Approach to Understanding and Mitigating Oversmoothing in GNNscited by 2
- 2025Hyper-SET: Designing Transformers via Hyperspherical Energy Minimizationcited by 2
- 2025Fairness in Augmented Graph Learning: A Surveycited by 2
- 2023Polynomial-based Self-Attention for Table Representation learningcited by 1
- 2023Degree-based stratification of nodes in Graph Neural Networkscited by 1
- 2025On the Role of Hidden States of Modern Hopfield Network in Transformercited by 1
- 2023Oversmoothing: A Nightmare for Graph Contrastive Learning?cited by 0
- 2025Channel-Attentive Graph Neural Networkscited by 0
- 2025Simplifying Graph Convolutional Networks with Redundancy-Free Neighborscited by 0
- 2025Identifying and Understanding Cross-Class Features in Adversarial Trainingcited by 0
- 2025Towards a deeper GCN: Alleviate over-smoothing with iterative training and fine-tuningcited by 0
- 2026Graph Signal Processing Meets Mamba2: Adaptive Filter Bank via Delta Modulationcited by 0
- 2026Graph Signal Processing Meets Mamba2: Adaptive Filter Bank via Delta Modulation20 Mar 2026
- 2025On the Role of Hidden States of Modern Hopfield Network in Transformer24 Nov 2025
- 2025Transformers as Intrinsic Optimizers: Forward Inference through the Energy Principle2 Nov 2025
- 2025Deeper with Riemannian Geometry: Overcoming Oversmoothing and Oversquashing for Graph Foundation Models20 Oct 2025
- 2025STAR: Boosting Time Series Foundation Models for Anomaly Detection through State-aware Adapter15 Oct 2025
- 2025Towards a deeper GCN: Alleviate over-smoothing with iterative training and fine-tuning21 Jun 2025
- 2025Identifying and Understanding Cross-Class Features in Adversarial Training5 Jun 2025
- 2025Only Large Weights (And Not Skip Connections) Can Prevent the Perils of Rank Collapse22 May 2025
- 2025Fairness in Augmented Graph Learning: A Survey30 Apr 2025
- 2025MOOSComp: Improving Lightweight Long-Context Compressor via Mitigating Over-Smoothing and Incorporating Outlier Scores23 Apr 2025
- 2025Simplifying Graph Convolutional Networks with Redundancy-Free Neighbors18 Apr 2025
- 2025Integrating Structural and Semantic Signals in Text-Attributed Graphs with BiGTex16 Apr 2025
27 more
- 2025Channel-Attentive Graph Neural Networks1 Mar 2025
- 2025Projection Head is Secretly an Information Bottleneck1 Mar 2025
- 2025Hyper-SET: Designing Transformers via Hyperspherical Energy Minimization17 Feb 2025
- 2025A Signed Graph Approach to Understanding and Mitigating Oversmoothing in GNNs17 Feb 2025
- 2025Are We Measuring Oversmoothing in Graph Neural Networks Correctly?7 Feb 2025
- 2024Measure-to-measure interpolation using Transformers7 Nov 2024
- 2024Idempotent Unsupervised Representation Learning for Skeleton-Based Action Recognition27 Oct 2024
- 2024Generalized Probabilistic Attention Mechanism in Transformers21 Oct 2024
- 2024CATCH: Channel-Aware multivariate Time Series Anomaly Detection via Frequency Patching16 Oct 2024
- 2024Fredformer: Frequency Debiased Transformer for Time Series Forecasting13 Jun 2024
- 2024Residual Connections and Normalization Can Provably Prevent Oversmoothing in GNNs5 Jun 2024
- 2024FairGT: A Fairness-aware Graph Transformer26 Apr 2024
- 2024GTC: GNN-Transformer Co-contrastive Learning for Self-supervised Heterogeneous Graph Representation22 Mar 2024
- 2024Non-negative Contrastive Learning19 Mar 2024
- 2024Do Generated Data Always Help Contrastive Learning?19 Mar 2024
- 2024Rethinking Graph Masked Autoencoders through Alignment and Uniformity11 Feb 2024
- 2024Setting the Record Straight on Transformer Oversmoothing9 Jan 2024
- 2023Degree-based stratification of nodes in Graph Neural Networks16 Dec 2023
- 2023An Attentive Inductive Bias for Sequential Recommendation beyond the Self-Attention16 Dec 2023
- 2023Polynomial-based Self-Attention for Table Representation learning12 Dec 2023
- 2023Graph Convolutions Enrich the Self-Attention in Transformers!7 Dec 2023
- 2023Architecture Matters: Uncovering Implicit Mechanisms in Graph Contrastive Learning5 Nov 2023
- 2023Identifiable Contrastive Learning with Automatic Feature Importance Discovery29 Oct 2023
- 2023Rank Collapse Causes Over-Smoothing and Over-Correlation in Graph Neural Networks31 Aug 2023
- 2023Towards Deep Attention in Graph Neural Networks: Problems and Remedies4 Jun 2023
- 2023Oversmoothing: A Nightmare for Graph Contrastive Learning?3 Jun 2023
- 2023Centered Self-Attention Layers2 Jun 2023
- 2024Do Generated Data Always Help Contrastive Learning?cited by 25
- 2024Do Generated Data Always Help Contrastive Learning?19 Mar 2024
- 2024CATCH: Channel-Aware multivariate Time Series Anomaly Detection via Frequency Patchingcited by 94
- 2024Fredformer: Frequency Debiased Transformer for Time Series Forecastingcited by 45
- 2024Residual Connections and Normalization Can Provably Prevent Oversmoothing in GNNscited by 34
- 2024FairGT: A Fairness-aware Graph Transformercited by 29
- 2024Measure-to-measure interpolation using Transformerscited by 29
- 2023An Attentive Inductive Bias for Sequential Recommendation beyond the Self-Attentioncited by 28
- 2023Towards Deep Attention in Graph Neural Networks: Problems and Remediescited by 24
- 2024Rethinking Graph Masked Autoencoders through Alignment and Uniformitycited by 22
- 2024Setting the Record Straight on Transformer Oversmoothingcited by 21
- 2024Non-negative Contrastive Learningcited by 19
- 2023Rank Collapse Causes Over-Smoothing and Over-Correlation in Graph Neural Networkscited by 18
- 2023Identifiable Contrastive Learning with Automatic Feature Importance Discoverycited by 14
12 most cited of 38 All 38 (JSON)
- 2026Graph Signal Processing Meets Mamba2: Adaptive Filter Bank via Delta Modulation20 Mar 2026
- 2025On the Role of Hidden States of Modern Hopfield Network in Transformer24 Nov 2025
- 2025Transformers as Intrinsic Optimizers: Forward Inference through the Energy Principle2 Nov 2025
- 2025Deeper with Riemannian Geometry: Overcoming Oversmoothing and Oversquashing for Graph Foundation Models20 Oct 2025
- 2025STAR: Boosting Time Series Foundation Models for Anomaly Detection through State-aware Adapter15 Oct 2025
- 2025Towards a deeper GCN: Alleviate over-smoothing with iterative training and fine-tuning21 Jun 2025
- 2025Identifying and Understanding Cross-Class Features in Adversarial Training5 Jun 2025
- 2025Only Large Weights (And Not Skip Connections) Can Prevent the Perils of Rank Collapse22 May 2025
- 2025Fairness in Augmented Graph Learning: A Survey30 Apr 2025
- 2025MOOSComp: Improving Lightweight Long-Context Compressor via Mitigating Over-Smoothing and Incorporating Outlier Scores23 Apr 2025
- 2025Simplifying Graph Convolutional Networks with Redundancy-Free Neighbors18 Apr 2025
- 2025Integrating Structural and Semantic Signals in Text-Attributed Graphs with BiGTex16 Apr 2025
12 newest of 38 All 38 (JSON)