Bayesian Graph Neural Networks with Adaptive Connection Sampling
Not on the map. Why?
Abstract (arXiv)
We propose a unified framework for adaptive connection sampling in graph neural networks (GNNs) that generalizes existing stochastic regularization methods for training GNNs. The proposed framework not only alleviates over-smoothing and over-fitting tendencies of deep GNNs, but also enables learning with uncertainty in graph analytic tasks with GNNs. Instead of using fixed sampling rates or hand-tuning them as model hyperparameters in existing stochastic regularization methods, our adaptive connection sampling can be trained jointly with GNN model parameters in both global and local fashions. GNN training with adaptive connection sampling is shown to be mathematically equivalent to an efficient approximation of training Bayesian GNNs. Experimental results with ablation studies on benchmark datasets validate that adaptively learning the sampling rate given graph training data is the key to boost the performance of GNNs in semi-supervised node classification, less prone to over-smoothing and over-fitting with more robust prediction.
Key facts. Cited by 81 papers in the map (data through 7 Oct 2026). First posted on arXiv 7 Jun 2020.
Builds on
No reference list yet.
Built on by
Citing papers per year: 2021: 13; 2022: 19; 2023: 15; 2024: 15; 2025: 13; 2026: 6.
81 papers.
Sort:
By area:
- 2023A Survey on Oversmoothing in Graph Neural Networkscited by 246
- 2021Training Graph Neural Networks with 1000 Layerscited by 169
- 2022Handling Distribution Shifts on Graphs: An Invariance Perspectivecited by 163
- 2021Dirichlet Energy Constrained Learning for Deep Graph Neural Networkscited by 97
- 2022Decoupling the Depth and Scope of Graph Neural Networkscited by 94
- 2021Graph Posterior Network: Bayesian Predictive Uncertainty for Node Classificationcited by 79
- 2022A Survey on Graph Representation Learning Methodscited by 66
- 2023Demystifying Oversmoothing in Attention-Based Graph Neural Networkscited by 47
- 2021Sampling methods for efficient training of graph convolutional networks: A surveycited by 40
- 2022What Makes Graph Neural Networks Miscalibrated?cited by 39
- 2022Confidence May Cheat: Self-Training on Graph Neural Networks under Distribution Shiftcited by 38
- 2021Bag of Tricks for Training Deeper Graph Neural Networks: A Comprehensive Benchmark Studycited by 36
38 more
- 2023Class-Imbalanced Graph Learning without Class Rebalancingcited by 29
- 2022Distribution Free Prediction Sets for Node Classificationcited by 28
- 2023Adaptive Message Passing: A General Framework to Mitigate Oversmoothing, Oversquashing, and Underreachingcited by 23
- 2022A Survey of Trustworthy Graph Learning: Reliability, Explainability, and Privacy Protectioncited by 20
- 2024Uncertainty in Graph Neural Networks: A Surveycited by 20
- 2025Oversmoothing, Oversquashing, Heterophily, Long-Range, and more: Demystifying Common Beliefs in Graph Machine Learningcited by 18
- 2021Graph Classification by Mixture of Diverse Expertscited by 16
- 2022JuryGCN: Quantifying Jackknife Uncertainty on Graph Convolutional Networkscited by 14
- 2022A General Framework for quantifying Aleatoric and Epistemic uncertainty in Graph Neural Networkscited by 12
- 2022Robust Node Classification on Graphs: Jointly from Bayesian Label Transition and Topology-based Label Propagationcited by 12
- 2024Safety in Graph Machine Learning: Threats and Safeguardscited by 11
- 2022Recent Advances in Reliable Deep Graph Learning: Inherent Noise, Distribution Shift, and Adversarial Attackcited by 10
- 2024Accurate and Scalable Estimation of Epistemic Uncertainty for Graph Neural Networkscited by 8
- 2024Uncertainty Modeling in Graph Neural Networks via Stochastic Differential Equationscited by 8
- 2021TyXe: Pyro-based Bayesian neural nets for Pytorchcited by 6
- 2022A Graph Is More Than Its Nodes: Towards Structured Uncertainty-Aware Learning on Graphscited by 6
- 2022Sparsification and Filtering for Spatial-temporal GNN in Multivariate Time-seriescited by 5
- 2022Retention Time Prediction for Chromatographic Enantioseparation by Quantile Geometry-enhanced Graph Neural Networkcited by 5
- 2023Homological Neural Networks: A Sparse Architecture for Multivariate Complexitycited by 5
- 2024MultiMax: Sparse and Multi-Modal Attention Learningcited by 5
- 2021Training Graph Neural Networks by Graphon Estimationcited by 4
- 2022Asynchronous Neural Networks for Learning in Graphscited by 4
- 2022Node Copying: A Random Graph Model for Effective Graph Samplingcited by 4
- 2023GNN-Ensemble: Towards Random Decision Graph Neural Networkscited by 4
- 2023GNNPipe: Scaling Deep GNN Training with Pipelined Model Parallelismcited by 4
- 2023On the Temperature of Bayesian Graph Neural Networks for Conformal Predictioncited by 4
- 2024Improving the interpretability of GNN predictions through conformal-based graph sparsificationcited by 4
- 2024Oversmoothing Alleviation in Graph Neural Networks: A Survey and Unified Viewcited by 4
- 2021Probabilistic Deep Learning to Quantify Uncertainty in Air Quality Forecastingcited by 3
- 2021Bayesian Graph Contrastive Learningcited by 3
- 2022MoReL: Multi-omics Relational Learningcited by 3
- 2023A Comprehensive Survey on Graph Summarization with Graph Neural Networkscited by 3
- 2024Uncertainty Quantification on Graph Learning: A Surveycited by 3
- 2024Enhancing the Resilience of Graph Neural Networks to Topological Perturbations in Sparse Graphscited by 3
- 2024Foundations and Frontiers of Graph Learning Theorycited by 3
- 2024xAI-Drop: Don't Use What You Cannot Explaincited by 3
- 2024Improving Graph Neural Networks by Learning Continuous Edge Directionscited by 3
- 2024Conditional Prediction ROC Bands for Graph Classificationcited by 3
50 most cited of 81 All 81 (JSON)
- 2026Online Bayesian Node Classification on Inductive Graphs under Distribution Shift12 Sep 2026
- 2026BLADE: Distilled LLM Regularization for Calibrated Knowledge Graph Completion26 Aug 2026
- 2026Random-Set Graph Neural Networks12 May 2026
- 2026Bayesian Neighborhood Adaptation for Graph Neural Networks5 Feb 2026
- 2026Graph Neural Network with One-side Edge Sampling for Fraud Detection11 Jan 2026
- 2026Persistent Tri-State Message Passing3 Jan 2026
- 2025Toward Efficient Testing of Graph Neural Networks via Test Input Prioritization20 Dec 2025
- 2025Supervised Learning of Random Neural Architectures Structured by Latent Random Fields on Compact Boundaryless Multiply-Connected Manifolds11 Dec 2025
- 2025Credal Graph Neural Networks2 Dec 2025
- 2025Resolving Node Identifiability in Graph Neural Processes via Laplacian Spectral Encodings24 Nov 2025
- 2025Sketch-Augmented Features Improve Learning Long-Range Dependencies in Graph Neural Networks5 Nov 2025
- 2025Two Birds with One Stone: Enhancing Uncertainty Quantification and Interpretability with Graph Functional Neural Process23 Aug 2025
38 more
- 2025GraphPPD: Posterior Predictive Modelling for Graph-Level Inference23 Aug 2025
- 2025Variational Graph Convolutional Neural Networks2 Jul 2025
- 2025Uncertainty Estimation for Heterophilic Graphs Through the Lens of Information Theory28 May 2025
- 2025Oversmoothing, Oversquashing, Heterophily, Long-Range, and more: Demystifying Common Beliefs in Graph Machine Learning21 May 2025
- 2025PRIMEDrive-CoT: A Precognitive Chain-of-Thought Framework for Uncertainty-Aware Object Interaction in Driving Scene Scenario8 Apr 2025
- 2025Effects of Dropout on Performance in Long-range Graph Learning Tasks11 Feb 2025
- 2025FedGrAINS: Personalized SubGraph Federated Learning with Adaptive Neighbor Sampling22 Jan 2025
- 2024Towards Precise Prediction Uncertainty in GNNs: Refining GNNs with Topology-grouping Strategy18 Dec 2024
- 2024Conditional Prediction ROC Bands for Graph Classification20 Oct 2024
- 2024Improving Graph Neural Networks by Learning Continuous Edge Directions18 Oct 2024
- 2024Uncertainty Modeling in Graph Neural Networks via Stochastic Differential Equations28 Aug 2024
- 2024xAI-Drop: Don't Use What You Cannot Explain29 Jul 2024
- 2024Foundations and Frontiers of Graph Learning Theory3 Jul 2024
- 2024Enhancing the Resilience of Graph Neural Networks to Topological Perturbations in Sparse Graphs5 Jun 2024
- 2024MultiMax: Sparse and Multi-Modal Attention Learning3 Jun 2024
- 2024Federated Learning for Non-factorizable Models using Deep Generative Prior Approximations25 May 2024
- 2024Safety in Graph Machine Learning: Threats and Safeguards17 May 2024
- 2024Oversmoothing Alleviation in Graph Neural Networks: A Survey and Unified View2 May 2024
- 2024Uncertainty Quantification on Graph Learning: A Survey23 Apr 2024
- 2024Improving the interpretability of GNN predictions through conformal-based graph sparsification18 Apr 2024
- 2024Uncertainty in Graph Neural Networks: A Survey11 Mar 2024
- 2024Accurate and Scalable Estimation of Epistemic Uncertainty for Graph Neural Networks7 Jan 2024
- 2023Adaptive Message Passing: A General Framework to Mitigate Oversmoothing, Oversquashing, and Underreaching27 Dec 2023
- 2023Graph Convolutional Network with Connectivity Uncertainty for EEG-based Emotion Recognition22 Oct 2023
- 2023On the Temperature of Bayesian Graph Neural Networks for Conformal Prediction17 Oct 2023
- 2023Accurate and Scalable Estimation of Epistemic Uncertainty for Graph Neural Networks20 Sep 2023
- 2023Class-Imbalanced Graph Learning without Class Rebalancing27 Aug 2023
- 2023Graph Neural Stochastic Differential Equations23 Aug 2023
- 2023GNNPipe: Scaling Deep GNN Training with Pipelined Model Parallelism19 Aug 2023
- 2023Homological Neural Networks: A Sparse Architecture for Multivariate Complexity27 Jun 2023
- 2023Oversmoothing: A Nightmare for Graph Contrastive Learning?3 Jun 2023
- 2023Demystifying Oversmoothing in Attention-Based Graph Neural Networks25 May 2023
- 2023Uncertainty Propagation in Node Classification3 Apr 2023
- 2023GNN-Ensemble: Towards Random Decision Graph Neural Networks20 Mar 2023
- 2023A Survey on Oversmoothing in Graph Neural Networks20 Mar 2023
- 2023A Comprehensive Survey on Graph Summarization with Graph Neural Networks13 Feb 2023
- 2023Understanding and Improving Deep Graph Neural Networks: A Probabilistic Graphical Model Perspective25 Jan 2023
- 2022Distribution Free Prediction Sets for Node Classification26 Nov 2022
50 newest of 81 All 81 (JSON)
- 2023A Survey on Oversmoothing in Graph Neural Networkscited by 246
- 2021Training Graph Neural Networks with 1000 Layerscited by 169
- 2022Handling Distribution Shifts on Graphs: An Invariance Perspectivecited by 163
- 2021Dirichlet Energy Constrained Learning for Deep Graph Neural Networkscited by 97
- 2022Decoupling the Depth and Scope of Graph Neural Networkscited by 94
- 2021Graph Posterior Network: Bayesian Predictive Uncertainty for Node Classificationcited by 79
- 2022A Survey on Graph Representation Learning Methodscited by 66
- 2023Demystifying Oversmoothing in Attention-Based Graph Neural Networkscited by 47
- 2021Sampling methods for efficient training of graph convolutional networks: A surveycited by 40
- 2022What Makes Graph Neural Networks Miscalibrated?cited by 39
- 2022Confidence May Cheat: Self-Training on Graph Neural Networks under Distribution Shiftcited by 38
- 2021Bag of Tricks for Training Deeper Graph Neural Networks: A Comprehensive Benchmark Studycited by 36
12 most cited of 80 All 80 (JSON)
- 2026Online Bayesian Node Classification on Inductive Graphs under Distribution Shift12 Sep 2026
- 2026BLADE: Distilled LLM Regularization for Calibrated Knowledge Graph Completion26 Aug 2026
- 2026Random-Set Graph Neural Networks12 May 2026
- 2026Bayesian Neighborhood Adaptation for Graph Neural Networks5 Feb 2026
- 2026Graph Neural Network with One-side Edge Sampling for Fraud Detection11 Jan 2026
- 2026Persistent Tri-State Message Passing3 Jan 2026
- 2025Toward Efficient Testing of Graph Neural Networks via Test Input Prioritization20 Dec 2025
- 2025Supervised Learning of Random Neural Architectures Structured by Latent Random Fields on Compact Boundaryless Multiply-Connected Manifolds11 Dec 2025
- 2025Credal Graph Neural Networks2 Dec 2025
- 2025Resolving Node Identifiability in Graph Neural Processes via Laplacian Spectral Encodings24 Nov 2025
- 2025Sketch-Augmented Features Improve Learning Long-Range Dependencies in Graph Neural Networks5 Nov 2025
- 2025Two Birds with One Stone: Enhancing Uncertainty Quantification and Interpretability with Graph Functional Neural Process23 Aug 2025
12 newest of 80 All 80 (JSON)