You Only Spectralize Once: Taking a Spectral Detour to Accelerate Graph Neural Network
Yi Li, Zhichun Guo, Guanpeng Li, Bingzhe Li
Abstract
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.
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 87f4def3-bf05-4c1c-9094-c935066a9305Builds on16
- Open Graph Benchmark: Datasets for Machine Learning on GraphsWeihua Hu, Matthias Fey, Marinka Zitnik, Yuxiao Dong et al.NeurIPS 2020 · 3,935 citations
- GraphSAINT: Graph Sampling Based Inductive Learning MethodHanqing Zeng, Hongkuan Zhou, Ajitesh Srivastava, Rajgopal Kannan et al.ICLR 2020 · 1,155 citations
- Graph Structure Learning for Robust Graph Neural NetworksWei Jin, Yao Ma, Xiaorui Liu, Xianfeng Tang et al.KDD 2020 · 604 citations
- Few-Shot Graph Learning for Molecular Property PredictionZhichun Guo, Chuxu Zhang, Wenhao Yu, John Herr et al.WWW 2021 · 213 citations
- Graph Condensation for Graph Neural NetworksWei Jin, Lingxiao Zhao, Shichang Zhang, Yozen Liu et al.ICLR 2022 · 203 citations
Related papers
- CompressGNN: Accelerating Graph Neural Network Training via Hierarchical CompressionZheng Chen, Feng Zhang, Yifei Xia, Wentao Zhang et al.KDD 2025 · 1 citation
- EXACT: Scalable Graph Neural Networks Training via Extreme Activation CompressionZirui Liu, Kaixiong Zhou, Fan Yang, Li Li et al.ICLR 2022 · 72 citations
- Serving Graph Compression for Graph Neural NetworksSi Si, Felix X. Yu, Ankit Singh Rawat, Cho-Jui Hsieh et al.ICLR 2023
- GraphNorm: A Principled Approach to Accelerating Graph Neural Network TrainingTianle Cai, Shengjie Luo, Keyulu Xu, Di He et al.ICML 2021 · 224 citations
- Scaling Up Graph Neural Networks Via Graph CoarseningZengfeng Huang, Shengzhong Zhang, Chong Xi, Tang Liu et al.KDD 2021 · 78 citations
