Joint Edge-Model Sparse Learning is Provably Efficient for Graph Neural Networks
Not on the map. Why?
Abstract (arXiv)
Due to the significant computational challenge of training large-scale graph neural networks (GNNs), various sparse learning techniques have been exploited to reduce memory and storage costs. Examples include textitgraph sparsification that samples a subgraph to reduce the amount of data aggregation and textitmodel sparsification that prunes the neural network to reduce the number of trainable weights. Despite the empirical successes in reducing the training cost while maintaining the test accuracy, the theoretical generalization analysis of sparse learning for GNNs remains elusive. To the best of our knowledge, this paper provides the first theoretical characterization of joint edge-model sparse learning from the perspective of sample complexity and convergence rate in achieving zero generalization error. It proves analytically that both sampling important nodes and pruning neurons with the lowest-magnitude can reduce the sample complexity and improve convergence without compromising the test accuracy. Although the analysis is centered on two-layer GNNs with structural constraints on data, the insights are applicable to more general setups and justified by both synthetic and practical citation datasets.
Key facts. Cited by 16 papers in the map (data through 14 Oct 2026). Builds on 58 of its 87 references that are in the map. First posted on arXiv 6 Feb 2023.
Builds on
58 of 87 references.
References · 58
- 2016Semi-Supervised Classification with Graph Convolutional Networkscited by 11,361
- 2015Distilling the Knowledge in a Neural Networkcited by 10,324
- 2017Graph Attention Networkscited by 8,923
- 2017Inductive Representation Learning on Large Graphscited by 6,320
- 2019A Comprehensive Survey on Graph Neural Networkscited by 2,919
- 2018Neural Tangent Kernel: Convergence and Generalization in Neural Networkscited by 2,621
- 2018The Lottery Ticket Hypothesis: Finding Sparse, Trainable Neural Networkscited by 2,262
- 2020Open Graph Benchmark: Datasets for Machine Learning on Graphscited by 2,163
- 2017Modeling Relational Data with Graph Convolutional Networkscited by 1,694
- 2014DeepWalk: Online Learning of Social Representationscited by 1,547
- 2019Simplifying Graph Convolutional Networkscited by 1,477
- 2016Google's Neural Machine Translation System: Bridging the Gap between Human and Machine Translationcited by 1,475
38 more
- 2018Spatial Temporal Graph Convolutional Networks for Skeleton-Based Action Recognitioncited by 1,209
- 2020LightGCN: Simplifying and Powering Graph Convolution Network for Recommendationcited by 1,139
- 2018Graph Convolutional Neural Networks for Web-Scale Recommender Systemscited by 1,138
- 2018Representation Learning on Graphs with Jumping Knowledge Networkscited by 877
- 2018A Convergence Theory for Deep Learning via Over-Parameterizationcited by 787
- 2019DropEdge: Towards Deep Graph Convolutional Networks on Node Classificationcited by 680
- 2018Gradient Descent Provably Optimizes Over-parameterized Neural Networkscited by 659
- 2017Deep Neural Networks as Gaussian Processescited by 622
- 2018FastGCN: Fast Learning with Graph Convolutional Networks via Importance Samplingcited by 584
- 2019Fine-Grained Analysis of Optimization and Generalization for Overparameterized Two-Layer Neural Networkscited by 566
- 2020Principal Neighbourhood Aggregation for Graph Netscited by 478
- 2018Learning and Generalization in Overparameterized Neural Networks, Going Beyond Two Layerscited by 340
- 2020Point-GNN: Graph Neural Network for 3D Object Detection in a Point Cloudcited by 335
- 2018Learning Overparameterized Neural Networks via Stochastic Gradient Descent on Structured Datacited by 261
- 2020Generalization and Representational Limits of Graph Neural Networkscited by 218
- 2020Proving the Lottery Ticket Hypothesis: Pruning is All You Needcited by 201
- 2019Layer-Dependent Importance Sampling for Training Deep and Large Graph Convolutional Networkscited by 177
- 2019Towards moderate overparameterization: global convergence guarantees for training shallow neural networkscited by 176
- 2019Graph Neural Tangent Kernel: Fusing Graph Neural Networks with Graph Kernelscited by 161
- 2022Neural Networks can Learn Representations with Gradient Descentcited by 161
- 2020Distilling Knowledge from Graph Convolutional Networkscited by 129
- 2020Logical Neural Networkscited by 119
- 2021Toward Understanding the Feature Learning Process of Self-supervised Contrastive Learningcited by 119
- 2021A Unified Lottery Ticket Hypothesis for Graph Neural Networkscited by 118
- 2016An Analysis of Deep Neural Network Models for Practical Applicationscited by 116
- 2020Feature Purification: How Adversarial Training Performs Robust Deep Learningcited by 107
- 2017Spurious Local Minima are Common in Two-Layer ReLU Neural Networkscited by 104
- 2020Degree-Quant: Quantization-Aware Training for Graph Neural Networkscited by 96
- 2017Recovery Guarantees for One-hidden-layer Neural Networkscited by 94
- 2019Stability and Generalization of Graph Convolutional Neural Networkscited by 81
- 2020Learning Parities with Neural Networkscited by 78
- 2019Graph Few-shot Learning via Knowledge Transfercited by 63
- 2021On Provable Benefits of Depth in Training Graph Convolutional Networkscited by 58
- 2021E-GraphSAGE: A Graph Neural Network based Intrusion Detection System for IoTcited by 56
- 2021Topology-Imbalance Learning for Semi-Supervised Node Classificationcited by 56
- 2021Optimization of Graph Neural Networks: Implicit Acceleration by Skip Connections and More Depthcited by 55
- 2022A Theoretical Analysis on Feature Learning in Neural Networks: Emergence from Inputs and Advantage over Fixed Featurescited by 49
- 2020Binary Graph Neural Networkscited by 34
50 most cited of 58 All 58 (JSON)
Built on by
Citing papers per year: 2023: 4; 2024: 9; 2025: 2; 2026: 1.
16 papers.
Sort:
By area:
- 2023A Theoretical Understanding of Shallow Vision Transformers: Learning, Generalization, and Sample Complexitycited by 77
- 2025When is Task Vector Provably Effective for Model Editing? A Generalization Analysis of Nonlinear Transformerscited by 38
- 2024How Do Nonlinear Transformers Learn and Generalize in In-Context Learning?cited by 35
- 2024A Provably Effective Method for Pruning Experts in Fine-tuned Sparse Mixture-of-Expertscited by 21
- 2023On the Convergence and Sample Complexity Analysis of Deep Q-Networks with ε-Greedy Explorationcited by 17
- 2024What Improves the Generalization of Graph Transformers? A Theoretical Dive into the Self-attention and Positional Encodingcited by 14
- 2024Graph Neural Networks for Tabular Data Learning: A Survey with Taxonomy and Directionscited by 13
- 2023Quantifying the Optimization and Generalization Advantages of Graph Neural Networks Over Multilayer Perceptronscited by 12
- 2024Node Identifiers: Compact, Discrete Representations for Efficient Graph Learningcited by 10
- 2024Learning on Transformers is Provable Low-Rank and Sparse: A One-layer Analysiscited by 8
- 2024SF-DQN: Provable Knowledge Transfer using Successor Feature for Deep Reinforcement Learningcited by 5
- 2024IGNN-Solver: A Graph Neural Solver for Implicit Graph Neural Networkscited by 3
4 more
- 2024Unifews: You Need Fewer Operations for Efficient Graph Neural Networkscited by 2
- 2026Efficient Quantization of Mixture-of-Experts with Theoretical Generalization Guaranteescited by 2
- 2023On the Equivalence of Graph Convolution and Mixupcited by 0
- 2025Theoretical Learning Performance of Graph Neural Networks: The Impact of Jumping Connections and Layer-wise Sparsificationcited by 0
- 2026Efficient Quantization of Mixture-of-Experts with Theoretical Generalization Guarantees7 Apr 2026
- 2025Theoretical Learning Performance of Graph Neural Networks: The Impact of Jumping Connections and Layer-wise Sparsification7 Jul 2025
- 2025When is Task Vector Provably Effective for Model Editing? A Generalization Analysis of Nonlinear Transformers15 Apr 2025
- 2024IGNN-Solver: A Graph Neural Solver for Implicit Graph Neural Networks11 Oct 2024
- 2024Learning on Transformers is Provable Low-Rank and Sparse: A One-layer Analysis24 Jun 2024
- 2024What Improves the Generalization of Graph Transformers? A Theoretical Dive into the Self-attention and Positional Encoding4 Jun 2024
- 2024A Provably Effective Method for Pruning Experts in Fine-tuned Sparse Mixture-of-Experts26 May 2024
- 2024Node Identifiers: Compact, Discrete Representations for Efficient Graph Learning26 May 2024
- 2024SF-DQN: Provable Knowledge Transfer using Successor Feature for Deep Reinforcement Learning24 May 2024
- 2024Unifews: You Need Fewer Operations for Efficient Graph Neural Networks20 Mar 2024
- 2024How Do Nonlinear Transformers Learn and Generalize in In-Context Learning?23 Feb 2024
- 2024Graph Neural Networks for Tabular Data Learning: A Survey with Taxonomy and Directions4 Jan 2024
4 more
- 2023On the Convergence and Sample Complexity Analysis of Deep Q-Networks with ε-Greedy Exploration24 Oct 2023
- 2023On the Equivalence of Graph Convolution and Mixup29 Sep 2023
- 2023Quantifying the Optimization and Generalization Advantages of Graph Neural Networks Over Multilayer Perceptrons24 Jun 2023
- 2023A Theoretical Understanding of Shallow Vision Transformers: Learning, Generalization, and Sample Complexity12 Feb 2023
- 2023A Theoretical Understanding of Shallow Vision Transformers: Learning, Generalization, and Sample Complexitycited by 77
- 2024How Do Nonlinear Transformers Learn and Generalize in In-Context Learning?cited by 35
- 2023On the Convergence and Sample Complexity Analysis of Deep Q-Networks with ε-Greedy Explorationcited by 17
- 2024What Improves the Generalization of Graph Transformers? A Theoretical Dive into the Self-attention and Positional Encodingcited by 14
- 2024Graph Neural Networks for Tabular Data Learning: A Survey with Taxonomy and Directionscited by 13
- 2023Quantifying the Optimization and Generalization Advantages of Graph Neural Networks Over Multilayer Perceptronscited by 12
- 2024Node Identifiers: Compact, Discrete Representations for Efficient Graph Learningcited by 10
- 2024Learning on Transformers is Provable Low-Rank and Sparse: A One-layer Analysiscited by 8
- 2024IGNN-Solver: A Graph Neural Solver for Implicit Graph Neural Networkscited by 3
- 2024Unifews: You Need Fewer Operations for Efficient Graph Neural Networkscited by 2
- 2023On the Equivalence of Graph Convolution and Mixupcited by 0
- 2025Theoretical Learning Performance of Graph Neural Networks: The Impact of Jumping Connections and Layer-wise Sparsificationcited by 0
- 2025Theoretical Learning Performance of Graph Neural Networks: The Impact of Jumping Connections and Layer-wise Sparsification7 Jul 2025
- 2024IGNN-Solver: A Graph Neural Solver for Implicit Graph Neural Networks11 Oct 2024
- 2024Learning on Transformers is Provable Low-Rank and Sparse: A One-layer Analysis24 Jun 2024
- 2024What Improves the Generalization of Graph Transformers? A Theoretical Dive into the Self-attention and Positional Encoding4 Jun 2024
- 2024Node Identifiers: Compact, Discrete Representations for Efficient Graph Learning26 May 2024
- 2024Unifews: You Need Fewer Operations for Efficient Graph Neural Networks20 Mar 2024
- 2024How Do Nonlinear Transformers Learn and Generalize in In-Context Learning?23 Feb 2024
- 2024Graph Neural Networks for Tabular Data Learning: A Survey with Taxonomy and Directions4 Jan 2024
- 2023On the Convergence and Sample Complexity Analysis of Deep Q-Networks with ε-Greedy Exploration24 Oct 2023
- 2023On the Equivalence of Graph Convolution and Mixup29 Sep 2023
- 2023Quantifying the Optimization and Generalization Advantages of Graph Neural Networks Over Multilayer Perceptrons24 Jun 2023
- 2023A Theoretical Understanding of Shallow Vision Transformers: Learning, Generalization, and Sample Complexity12 Feb 2023