You Only Spectralize Once: Taking a Spectral Detour to Accelerate Graph Neural Network
Yi Li, Zhichun Guo, Guanpeng Li, Bingzhe Li
摘要
Training Graph Neural Networks (GNNs) often relies on repeated, irregular, and expensive message-passing operations over all nodes (e.g., N ), leading to high computational overhead. To alleviate this inefficiency, we revisit the GNNs training from the spectral perspective. Node features and embeddings in many real-world graph exhibit sparse representation in Graph Fourier domain. This inherent sparsity aligns well with the Compressed Sensing principles, which posits that sparse signals can be accurately reconstructed from significantly fewer measurements (e.g., M and M ≪ N ). This observation motivates designing efficient GNNs that operates predominantly in a compressed spectral subspace. In this paper, we propose You Only Spectralize Once (YOSO), a GNN training scheme that first performing a single projection of features onto a learnable orthonormal Graph Fourier basis U ℓ , and after compressed sensing is used, retaining only M spectral coefficients where M ≪ N . The entire GNN computation then performs in this reduced/compressed spectral domain. Finally, the full graph embeddings are recovered back to original domain at output layer by solving a compressed sensing bounded ℓ 2 , 1 -regularized optimization problem. Theoretically, drawing upon the compressed sensing theory, we prove that stable recovery by showing that this whole process can satisfy the Restricted Isometry Property when M = O ( k (log N/k )) . Empirically, YOSO achieves an average 74% training time reduction across five benchmark datasets compared to state-of-the-art baseline schemes, while maintaining the competitive model accuracy.
问问这篇 Paper
智能体会读完全文。
Lune 把这篇 Paper 索引到了每一个公式,引用它的顶会 Paper 也一样。你提问,回答直接引用原文。
它引用的顶会 Paper16
- Open Graph Benchmark: Datasets for Machine Learning on GraphsWeihua Hu, Matthias Fey, Marinka Zitnik, Yuxiao Dong 等NeurIPS 2020 · 被引用 3,935 次
- GraphSAINT: Graph Sampling Based Inductive Learning MethodHanqing Zeng, Hongkuan Zhou, Ajitesh Srivastava, Rajgopal Kannan 等ICLR 2020 · 被引用 1,155 次
- Graph Structure Learning for Robust Graph Neural NetworksWei Jin, Yao Ma, Xiaorui Liu, Xianfeng Tang 等KDD 2020 · 被引用 604 次
- Few-Shot Graph Learning for Molecular Property PredictionZhichun Guo, Chuxu Zhang, Wenhao Yu, John Herr 等WWW 2021 · 被引用 213 次
- Graph Condensation for Graph Neural NetworksWei Jin, Lingxiao Zhao, Shichang Zhang, Yozen Liu 等ICLR 2022 · 被引用 203 次
相关 Paper
- CompressGNN: Accelerating Graph Neural Network Training via Hierarchical CompressionZheng Chen, Feng Zhang, Yifei Xia, Wentao Zhang 等KDD 2025 · 被引用 1 次
- EXACT: Scalable Graph Neural Networks Training via Extreme Activation CompressionZirui Liu, Kaixiong Zhou, Fan Yang, Li Li 等ICLR 2022 · 被引用 72 次
- Serving Graph Compression for Graph Neural NetworksSi Si, Felix X. Yu, Ankit Singh Rawat, Cho-Jui Hsieh 等ICLR 2023
- GraphNorm: A Principled Approach to Accelerating Graph Neural Network TrainingTianle Cai, Shengjie Luo, Keyulu Xu, Di He 等ICML 2021 · 被引用 224 次
- Scaling Up Graph Neural Networks Via Graph CoarseningZengfeng Huang, Shengzhong Zhang, Chong Xi, Tang Liu 等KDD 2021 · 被引用 78 次
