Geometric Knowledge Distillation: Topology Compression for Graph Neural Networks
Not on the map. Why?
Abstract (arXiv)
We study a new paradigm of knowledge transfer that aims at encoding graph topological information into graph neural networks (GNNs) by distilling knowledge from a teacher GNN model trained on a complete graph to a student GNN model operating on a smaller or sparser graph. To this end, we revisit the connection between thermodynamics and the behavior of GNN, based on which we propose Neural Heat Kernel (NHK) to encapsulate the geometric property of the underlying manifold concerning the architecture of GNNs. A fundamental and principled solution is derived by aligning NHKs on teacher and student models, dubbed as Geometric Knowledge Distillation. We develop non- and parametric instantiations and demonstrate their efficacy in various experimental settings for knowledge distillation regarding different types of privileged topological information and teacher-student schemes.
Key facts. Cited by 22 papers in the map (data through 7 Oct 2026). Builds on 45 of its 74 references that are in the map. First posted on arXiv 24 Oct 2022.
Builds on
45 of 74 references.
References · 45
- 2016Semi-Supervised Classification with Graph Convolutional Networkscited by 11,282
- 2015Distilling the Knowledge in a Neural Networkcited by 10,076
- 2017Graph Attention Networkscited by 8,854
- 2017Inductive Representation Learning on Large Graphscited by 6,284
- 2020Open Graph Benchmark: Datasets for Machine Learning on Graphscited by 2,148
- 2014FitNets: Hints for Thin Deep Netscited by 1,619
- 2020Knowledge Distillation: A Surveycited by 1,578
- 2019Simplifying Graph Convolutional Networkscited by 1,470
- 2021Geometric Deep Learning: Grids, Groups, Graphs, Geodesics, and Gaugescited by 1,131
- 2016Paying More Attention to Attention: Improving the Performance of Convolutional Neural Networks via Attention Transfercited by 1,059
- 2016Geometric deep learning: going beyond Euclidean datacited by 1,042
- 2018Predict then Propagate: Graph Neural Networks meet Personalized PageRankcited by 987
33 more
- 2019Relational Knowledge Distillationcited by 873
- 2018Born Again Neural Networkscited by 539
- 2021Understanding over-squashing and bottlenecks on graphs via curvaturecited by 471
- 2016Geometric deep learning on graphs and manifolds using mixture model CNNscited by 468
- 2013Do Deep Nets Really Need to be Deep?cited by 466
- 2015Deep Kernel Learningcited by 440
- 2019Diffusion Improves Graph Learningcited by 414
- 2019A Comprehensive Overhaul of Feature Distillationcited by 341
- 2021GRAND: Graph Neural Diffusioncited by 258
- 2018Paraphrasing Complex Network: Network Compression via Factor Transfercited by 255
- 2015Unifying distillation and privileged informationcited by 218
- 2020How Neural Networks Extrapolate: From Feedforward to Graph Neural Networkscited by 218
- 2020Generalization and Representational Limits of Graph Neural Networkscited by 217
- 2022Neural Sheaf Diffusion: A Topological Perspective on Heterophily and Oversmoothing in GNNscited by 208
- 2020Graph-Bert: Only Attention is Needed for Learning Graph Representationscited by 176
- 2020Self-Distillation Amplifies Regularization in Hilbert Spacecited by 171
- 2022Handling Distribution Shifts on Graphs: An Invariance Perspectivecited by 163
- 2021Graph-less Neural Networks: Teaching Old MLPs New Tricks via Distillationcited by 151
- 2021Subgraph Federated Learning with Missing Neighbor Generationcited by 132
- 2020Distilling Knowledge from Graph Convolutional Networkscited by 128
- 2021PDE-GCN: Novel Architectures for Graph Neural Networks Motivated by Partial Differential Equationscited by 106
- 2021Beltrami Flow and Neural Diffusion on Graphscited by 84
- 2021Extract the Knowledge of Graph Neural Networks and Go Beyond it: An Effective Knowledge Distillation Frameworkcited by 69
- 2021Dissecting the Diffusion Process in Linear Graph Convolutional Networkscited by 58
- 2018Graph Coarsening with Preserved Spectral Propertiescited by 41
- 2020On Self-Distilling Graph Neural Networkcited by 35
- 2014Heat kernel based community detectioncited by 30
- 2022Cross-Task Knowledge Distillation in Multi-Task Recommendationcited by 22
- 2021Towards Open-World Feature Extrapolation: An Inductive Graph Learning Approachcited by 20
- 2020Faster Graph Embeddings via Coarseningcited by 15
- 2019Personalized Graph Neural Networks with Attention Mechanism for Session-Aware Recommendationcited by 10
- 2021Privileged Graph Distillation for Cold Start Recommendationcited by 10
- 2013The nonlinear heat equation on dense graphs and graph limitscited by 9
Built on by
Citing papers per year: 2022: 1; 2023: 7; 2024: 9; 2025: 2; 2026: 3.
22 papers.
Sort:
By area:
- 2023NodeFormer: A Scalable Graph Structure Learning Transformer for Node Classificationcited by 207
- 2023SGFormer: Simplifying and Empowering Transformers for Large-Graph Representationscited by 131
- 2023DIFFormer: Scalable (Graph) Transformers Induced by Energy Constrained Diffusioncited by 85
- 2023Energy-based Out-of-Distribution Detection for Graph Neural Networkscited by 67
- 2024LLMs as Zero-shot Graph Learners: Alignment of GNN Representations with LLM Token Embeddingscited by 60
- 2022Graph Neural Networks are Inherently Good Generalizers: Insights by Bridging GNNs and MLPscited by 42
- 2023Knowledge Distillation on Graphs: A Surveycited by 38
- 2025Position: Beyond Euclidean -- Foundation Models Should Embrace Non-Euclidean Geometriescited by 15
- 2023GraphGLOW: Universal and Generalizable Structure Learning for Graph Neural Networkscited by 12
- 2024Node Identifiers: Compact, Discrete Representations for Efficient Graph Learningcited by 10
- 2023Accelerating Scalable Graph Neural Network Inference with Node-Adaptive Propagationcited by 8
- 2024Mitigating Label Noise on Graph via Topological Sample Selectioncited by 6
10 more
- 2024EasyST: A Simple Framework for Spatio-Temporal Predictioncited by 6
- 2025GraphSOS: Graph Sampling and Order Selection to Help LLMs Understand Graphs Bettercited by 4
- 2024Learning Divergence Fields for Shift-Robust Graph Representationscited by 2
- 2026An LLM-Guided Query-Aware Inference System for GNN Models on Large Knowledge Graphscited by 2
- 2024Introducing Diminutive Causal Structure into Graph Representation Learningcited by 1
- 2024Transformers from Diffusion: A Unified Framework for Neural Message Passingcited by 1
- 2024Teaching MLP More Graph Information: A Three-stage Multitask Knowledge Distillation Frameworkcited by 0
- 2024Acceleration Algorithms in GNNs: A Surveycited by 0
- 2026LoReC: Rethinking Large Language Models for Graph Data Analysiscited by 0
- 2026Model-to-Data Distillation for Graph Neural Networkscited by 0
- 2026Model-to-Data Distillation for Graph Neural Networks7 May 2026
- 2026LoReC: Rethinking Large Language Models for Graph Data Analysis20 Apr 2026
- 2026An LLM-Guided Query-Aware Inference System for GNN Models on Large Knowledge Graphs4 Mar 2026
- 2025Position: Beyond Euclidean -- Foundation Models Should Embrace Non-Euclidean Geometries11 Apr 2025
- 2025GraphSOS: Graph Sampling and Order Selection to Help LLMs Understand Graphs Better24 Jan 2025
- 2024Transformers from Diffusion: A Unified Framework for Neural Message Passing13 Sep 2024
- 2024EasyST: A Simple Framework for Spatio-Temporal Prediction10 Sep 2024
- 2024LLMs as Zero-shot Graph Learners: Alignment of GNN Representations with LLM Token Embeddings25 Aug 2024
- 2024Introducing Diminutive Causal Structure into Graph Representation Learning13 Jun 2024
- 2024Learning Divergence Fields for Shift-Robust Graph Representations7 Jun 2024
- 2024Node Identifiers: Compact, Discrete Representations for Efficient Graph Learning26 May 2024
- 2024Acceleration Algorithms in GNNs: A Survey7 May 2024
10 more
- 2024Mitigating Label Noise on Graph via Topological Sample Selection4 Mar 2024
- 2024Teaching MLP More Graph Information: A Three-stage Multitask Knowledge Distillation Framework2 Mar 2024
- 2023Accelerating Scalable Graph Neural Network Inference with Node-Adaptive Propagation17 Oct 2023
- 2023GraphGLOW: Universal and Generalizable Structure Learning for Graph Neural Networks20 Jun 2023
- 2023SGFormer: Simplifying and Empowering Transformers for Large-Graph Representations19 Jun 2023
- 2023NodeFormer: A Scalable Graph Structure Learning Transformer for Node Classification14 Jun 2023
- 2023Energy-based Out-of-Distribution Detection for Graph Neural Networks6 Feb 2023
- 2023Knowledge Distillation on Graphs: A Survey1 Feb 2023
- 2023DIFFormer: Scalable (Graph) Transformers Induced by Energy Constrained Diffusion23 Jan 2023
- 2022Graph Neural Networks are Inherently Good Generalizers: Insights by Bridging GNNs and MLPs18 Dec 2022
- 2023NodeFormer: A Scalable Graph Structure Learning Transformer for Node Classificationcited by 207
- 2023SGFormer: Simplifying and Empowering Transformers for Large-Graph Representationscited by 131
- 2023DIFFormer: Scalable (Graph) Transformers Induced by Energy Constrained Diffusioncited by 85
- 2023Energy-based Out-of-Distribution Detection for Graph Neural Networkscited by 67
- 2024LLMs as Zero-shot Graph Learners: Alignment of GNN Representations with LLM Token Embeddingscited by 60
- 2022Graph Neural Networks are Inherently Good Generalizers: Insights by Bridging GNNs and MLPscited by 42
- 2023Knowledge Distillation on Graphs: A Surveycited by 38
- 2025Position: Beyond Euclidean -- Foundation Models Should Embrace Non-Euclidean Geometriescited by 15
- 2023GraphGLOW: Universal and Generalizable Structure Learning for Graph Neural Networkscited by 12
- 2024Node Identifiers: Compact, Discrete Representations for Efficient Graph Learningcited by 10
- 2024Mitigating Label Noise on Graph via Topological Sample Selectioncited by 6
- 2024EasyST: A Simple Framework for Spatio-Temporal Predictioncited by 6
12 most cited of 20 All 20 (JSON)
- 2026Model-to-Data Distillation for Graph Neural Networks7 May 2026
- 2026An LLM-Guided Query-Aware Inference System for GNN Models on Large Knowledge Graphs4 Mar 2026
- 2025Position: Beyond Euclidean -- Foundation Models Should Embrace Non-Euclidean Geometries11 Apr 2025
- 2025GraphSOS: Graph Sampling and Order Selection to Help LLMs Understand Graphs Better24 Jan 2025
- 2024Transformers from Diffusion: A Unified Framework for Neural Message Passing13 Sep 2024
- 2024EasyST: A Simple Framework for Spatio-Temporal Prediction10 Sep 2024
- 2024LLMs as Zero-shot Graph Learners: Alignment of GNN Representations with LLM Token Embeddings25 Aug 2024
- 2024Introducing Diminutive Causal Structure into Graph Representation Learning13 Jun 2024
- 2024Learning Divergence Fields for Shift-Robust Graph Representations7 Jun 2024
- 2024Node Identifiers: Compact, Discrete Representations for Efficient Graph Learning26 May 2024
- 2024Acceleration Algorithms in GNNs: A Survey7 May 2024
- 2024Mitigating Label Noise on Graph via Topological Sample Selection4 Mar 2024
12 newest of 20 All 20 (JSON)