Two Heads Are Better Than One: Boosting Graph Sparse Training via Semantic and Topological Awareness
Guibin Zhang, Yanwei Yue, Kun Wang, Junfeng Fang, Yongduo Sui, Kai Wang, Yuxuan Liang, Dawei Cheng, Shirui Pan, Tianlong Chen
Abstract
Graph Neural Networks (GNNs) excel in various graph learning tasks but face computational challenges when applied to large-scale graphs. A promising solution is to remove non-essential edges to reduce the computational overheads in GNN. Previous literature generally falls into two categories: topology-guided and semantic-guided. The former maintains certain graph topological properties yet often underperforms on GNNs due to low integration with neural network training. The latter performs well at lower sparsity on GNNs but faces performance collapse at higher sparsity levels. With this in mind, we take the first step to propose a new research line and concept termed Graph Sparse Training (GST), which dynamically manipulates sparsity at the data level. Specifically, GST initially constructs a topology&semantic anchor at a low training cost, followed by performing dynamic sparse training to align the sparse graph with the anchor. We introduce the Equilibria Sparsification Principle to guide this process, effectively balancing the preservation of both topological and semantic information. Ultimately, GST produces a sparse graph with maximum topological integrity and no performance degradation. Extensive experiments on 6 datasets and 5 backbones showcase that GST (I) identifies subgraphs at higher graph sparsity levels (1.67% 15.85% ) than state-of-the-art sparsification methods, (II) preserves more key spectral properties, (III) achieves 1.27-3.42 speedup in GNN inference and (IV) successfully helps graph adversarial defense and graph lottery tickets.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Your agent calls
Luneget_paper_fulltext
Free to start. No credit card required.
Terminal
Install the CLIlune papers fulltext 4d472016-ded5-4c39-8323-ad4823b21fd9Cited by top-tier papers11
- NuwaDynamics: Discovering and Updating in Causal Spatio-Temporal ModelingKun Wang, Hao Wu, Yifan Duan, Guibin Zhang et al.ICLR 2024 · 38 citations
- Prometheus: Out-of-distribution Fluid Dynamics Modeling with Disentangled Graph ODEHao Wu, Huiyuan Wang, Kun Wang, Weiyan Wang et al.ICML 2024 · 25 citations
- Graph Lottery Ticket AutomatedGuibin Zhang, Kun Wang, Wei Huang, Yanwei Yue et al.ICLR 2024 · 17 citations
- Causal Deciphering and Inpainting in Spatio-Temporal Dynamics via Diffusion ModelYifan Duan, Jian Zhao, pengcheng, Junyuan Mao et al.NeurIPS 2024 · 14 citations
- GDeR: Safeguarding Efficiency, Balancing, and Robustness via Prototypical Graph PruningGuibin Zhang, Haonan Dong, Yuchen Zhang, Zhixun Li et al.NeurIPS 2024 · 9 citations
Builds on20
- Open Graph Benchmark: Datasets for Machine Learning on GraphsWeihua Hu, Matthias Fey, Marinka Zitnik, Yuxiao Dong et al.NeurIPS 2020 · 3,935 citations
- Rigging the Lottery: Making All Tickets WinnersUtku Evci, Trevor Gale, Jacob Menick, Pablo Samuel Castro et al.ICML 2020 · 723 citations
- Robust Graph Representation Learning via Neural SparsificationCheng Zheng, Bo Zong, Wei Cheng, Dongjin Song et al.ICML 2020 · 330 citations
- Inductive Matrix Completion Based on Graph Neural NetworksMuhan Zhang, Yixin ChenICLR 2020 · 273 citations
- Graph Structure Learning with Variational Information BottleneckQingyun Sun, Jianxin Li, Hao Peng, Jia Wu et al.AAAI 2022 · 224 citations
Related papers
- Rethinking Graph Lottery Tickets: Graph Sparsity MattersBo Hui, Da Yan, Xiaolong Ma, Wei-Shinn KuICLR 2023 · 2 citations
- Searching Lottery Tickets in Graph Neural Networks: A Dual PerspectiveKun Wang, Yuxuan Liang, Pengkun Wang, Xu Wang et al.ICLR 2023
- A Unified Lottery Ticket Hypothesis for Graph Neural NetworksTianlong Chen, Yongduo Sui, Xuxi Chen, Aston Zhang et al.ICML 2021 · 208 citations
- Pre-Training Identification of Graph Winning Tickets in Adaptive Spatial-Temporal Graph Neural NetworksWenying Duan, Tianxiang Fang, Hong Rao, Xiaoxi HeKDD 2024 · 1 citation
- Early-Bird GCNs: Graph-Network Co-optimization towards More Efficient GCN Training and Inference via Drawing Early-Bird Lottery TicketsHaoran You, Zhihan Lu, Zijian Zhou, Yonggan Fu et al.AAAI 2022 · 37 citations
