Proceedings

SALE-MLP: Structure Aware Latent Embeddings for GNN to Graph-free MLP Distillation

Download BibTeX Cite

Proceedings of the Thirty-Fourth International Joint Conference on Artificial Intelligence. Main Track. Pages 6003-6011.

https://doi.org/10.24963/ijcai.2025/668

Abstract

Graph Neural Networks (GNNs), with their ability to effectively handle non-Euclidean data structures, have demonstrated state-of-the-art performance in learning node and graph-level representations. However, GNNs face significant computational overhead due to their message-passing mechanisms, making them impractical for real-time large-scale applications. Recently, Graph-to-MLP (G2M) knowledge distillation has emerged as a promising solution, utilizing MLPs to reduce inference latency. However, existing methods often lack structural awareness (SA), limiting their ability to capture essential graph-specific information. Moreover, some methods require access to large-scale graphs, undermining their scalability. To address these issues, we propose SALE-MLP (Structure-Aware Latent Embeddings for GNN-to-Graph-Free MLP Distillation), a novel graph-free and structure-aware approach that leverages unsupervised structural losses to align the MLP feature space with the underlying graph structure. SALE-MLP does not rely on precomputed GNN embeddings nor require graph during inference, making it efficient for real-world applications. Extensive experiments demonstrate that SALE-MLP outperforms existing G2M methods across tasks and datasets, achieving 3–4% improvement in node classification for inductive settings while maintaining strong transductive performance.

  1. [Bianchi et al., 2020] Filippo Maria Bianchi, Daniele Grat- tarola, and Cesare Alippi. Spectral clustering with graph neural networks for graph pooling. In International con- ference on machine learning, pages 874-883, 2020.
  2. [Chaurasiya et al., 2022] Deepak Chaurasiya, Anil Surisetty, Nitish Kumar, Alok Singh, Vikrant Dey, Aakarsh Malho- tra, Gaurav Dhama, and Ankur Arora. Entity alignment for knowledge graphs: progress, challenges, and empirical studies. arXiv preprint arXiv:2205.08777, 2022.
  3. [Chen et al., 2021] Yuzhao Chen, Yatao Bian, Xi Xiao, Yu Rong, Tingyang Xu, and Junzhou Huang. On self- distilling graph neural network. In Zhi-Hua Zhou, editor, International Joint Conference on Artificial Intelligence, pages 2278-2284, 2021.
  4. [Chen et al., 2022] Jie Chen, Shouzhen Chen, Mingyuan Bai, Junbin Gao, Junping Zhang, and Jian Pu. Sa-mlp: Distilling graph knowledge from gnns into structure-aware mlp. arXiv preprint arXiv:2210.09609, 2022.
  5. Kenneth Ward Church. Word2vec. Natural Language Engineering, 23:155-162, 2017.
  6. [Dhillon et al., 2004] Inderjit S Dhillon, Yuqiang Guan, and Brian Kulis. Kernel k-means: spectral clustering and nor- malized cuts. In ACM international conference on Knowl- edge discovery and data mining, pages 551-556, 2004.
  7. [Ding et al., 2021] Mucong Ding, Kezhi Kong, Jingling Li, Chen Zhu, John Dickerson, Furong Huang, and Tom Gold- stein. Vq-gnn: A universal framework to scale up graph neural networks using vector quantization. Neural Infor- mation Processing Systems, 34:6733-6746, 2021.
  8. [Feng et al., 2022] Wenzheng Feng, Yuxiao Dong, Tinglin Huang, Ziqi Yin, Xu Cheng, Evgeny Kharlamov, and Jie Tang. Grand+: Scalable graph random neural networks. In ACM Web Conference, pages 3248-3258, 2022.
  9. [Grover and Leskovec, 2016] Aditya Grover and Jure Leskovec. node2vec: Scalable feature learning for net- works. In ACM International Conference on Knowledge Discovery and Data mining, pages 855-864, 2016.
  10. [Hamilton et al., 2017] Will Hamilton, Zhitao Ying, and Jure Leskovec. Inductive representation learning on large graphs. Neural Information Processing Systems, 30, 2017.
  11. [Han et al., 2022] Xiaotian Han, Zhimeng Jiang, Ninghao Liu, and Xia Hu. G-mixup: Graph data augmentation for graph classification. In International Conference on Ma- chine Learning, pages 8230-8248, 2022.
  12. [Hinton, 2015] Geoffrey Hinton. Distilling the knowledge in a neural network. arXiv preprint arXiv:1503.02531, 2015.
  13. [Hu et al., 2020] Weihua Hu, Matthias Fey, Marinka Zitnik, Yuxiao Dong, Hongyu Ren, Bowen Liu, Michele Catasta, and Jure Leskovec. Open graph benchmark: Datasets for machine learning on graphs. Neural Information Process- ing Systems, 33:22118-22133, 2020.
  14. [Huang et al., 2024] Yuyang Huang, Wenjing Lu, and Yang Yang. An efficient prototype-based clustering approach for edge pruning in graph neural networks to battle over- smoothing. In International Joint Conference on Artificial Intelligence, pages 4201-4209, 2024.
  15. [Jha et al., 2022] Kanchan Jha, Sriparna Saha, and Hiteshi Singh. Prediction of protein-protein interaction using graph neural networks. Nature Publishing Group UK Lon- don, 12:8360, 2022.
  16. [Joshi et al., 2022] Chaitanya K Joshi, Fayao Liu, Xu Xun, Jie Lin, and Chuan Sheng Foo. On representation knowl- edge distillation for graph neural networks. IEEE transac- tions on neural networks and learning systems, 2022.
  17. [Kipf and Welling, 2022] Thomas N Kipf and Max Welling. Semi-supervised classification with graph convolutional networks. In International Conference on Learning Rep- resentations, 2022.
  18. [Lassance et al., 2020] Carlos Lassance, Myriam Bontonou, Ghouthi Hacene, Vincent Gripon, Jian Tang, and Anto- nio Ortega. Deep geometric knowledge distillation with graphs. In IEEE International Conference on Acoustics, Speech and Signal Processing, pages 8484-8488, 2020.
  19. [Li et al., 2018] Qimai Li, Zhichao Han, and Xiao-Ming Wu. Deeper insights into graph convolutional networks for semi-supervised learning. In AAAI conference on artificial intelligence, volume 32, 2018.
  20. [Liu et al., 2022] Xin Liu, Mingyu Yan, Lei Deng, Guoqi Li, Xiaochun Ye, Dongrui Fan, Shirui Pan, and Yuan Xie. Sur- vey on graph neural network acceleration: An algorithmic perspective. In International Joint Conference on Artificial Intelligence, pages 5521-5529, 2022. IJCAI
  21. [Lu et al., 2024] Weigang Lu, Ziyu Guan, Wei Zhao, and Yaming Yang. Adagmlp: Adaboosting gnn-to-mlp knowl- edge distillation. In ACM Conference on Knowledge Dis- covery and Data Mining, pages 2060-2071, 2024.
  22. [Malik et al., 2024] Sarthak Malik, Aditi Rai, Himank Seh- gal, Akshay Sethi, Aakarsh Malhotra, et al. Grated-mlp: Efficient node classification via graph transformer distilla- tion to mlp. In Learning on Graphs Conference, 2024.
  23. [Mikolov et al., 2013] Tomas Mikolov, Ilya Sutskever, Kai Chen, Greg S Corrado, and Jeff Dean. Distributed rep- resentations of words and phrases and their composition- ality. Neural Information Processing Systems, 26, 2013.
  24. [Modak et al., 2024] Sudipta Modak, Aakarsh Malhotra, Sarthak Malik, Anil Surisetty, and Esam Abdel- Raheem. Cpa-wac: constellation partitioning-based scal- able weighted aggregation composition for knowledge graph embedding. In International Joint Conference on Artificial Intelligence, pages 3504-3512, 2024. IJCAI
  25. [Perozzi et al., 2014] Bryan Perozzi, Rami Al-Rfou, and Steven Skiena. Deepwalk: Online learning of social rep- resentations. In ACM international conference on Knowl- edge discovery and data mining, pages 701-710, 2014.
  26. [Postȃvaru et al., 2020] S ¸tefan Postȃvaru, Anton Tsitsulin, Filipe Miguel Gonc ¸alves de Almeida, Yingtao Tian, Sil- vio Lattanzi, and Bryan Perozzi. Instantembedding: Efficient local node representations. arXiv preprint arXiv:2010.06992, 2020.
  27. [Ren et al., 2021] Yating Ren, Junzhong Ji, Lingfeng Niu, and Minglong Lei. Multi-task self-distillation for graph-based semi-supervised learning. arXiv preprint arXiv:2112.01174, 2021.
  28. [Sen et al., 2008] Prithviraj Sen, Galileo Namata, Mustafa Bilgic, Lise Getoor, Brian Galligher, and Tina Eliassi-Rad. Collective classification in network data. AI magazine, 29:93-93, 2008.
  29. [Surisetty et al., 2022] Anil Surisetty, Deepak Chaurasiya, Nitish Kumar, Alok Singh, Gaurav Dhama, Aakarsh Mal- hotra, Ankur Arora, and Vikrant Dey. Reps: Relation, position and structure aware entity alignment. In ACM World Wide Web Conference Workshop on Graph Learn- ing, pages 1083-1091, 2022.
  30. [Szegedy, 2013] C Szegedy. Intriguing properties of neural networks. arXiv preprint arXiv:1312.6199, 2013.
  31. [Tailor et al., 2021] Shyam A. Tailor, Javier F.-Marques, and Nicholas Donald Lane. Degree-quant: Quantization-aware training for graph neural networks. In International Con- ference on Learning Representations, 2021.
  32. [Tang et al., 2009] Jie Tang, Jimeng Sun, Chi Wang, and Zi Yang. Social influence analysis in large-scale networks. In ACM international conference on Knowledge discovery and data mining, pages 807-816, 2009.
  33. [Tang et al., 2015] Jian Tang, Meng Qu, Mingzhe Wang, Ming Zhang, Jun Yan, and Qiaozhu Mei. Line: Large- scale information network embedding. In international conference on world wide web, pages 1067-1077, 2015.
  34. [Tian et al., 2022] Yijun Tian, Chuxu Zhang, Zhichun Guo, Xiangliang Zhang, and Nitesh Chawla. Learning mlps on graphs: A unified view of effectiveness, robustness, and efficiency. In International Conference on Learning Rep- resentations, 2022.
  35. [Veličković et al., 2018] Petar Veličković, Guillem Cucurull, Arantxa Casanova, Adriana Romero, Pietro Liò, and Yoshua Bengio. Graph attention networks. In Interna- tional Conference on Learning Representations, 2018.
  36. [Wang et al., 2024] Zehong Wang, Zheyuan Zhang, Chuxu Zhang, and Yanfang Ye. Graph inference acceleration by learning mlps on graphs without supervision. arXiv preprint arXiv:2402.08918, 2024.
  37. [Winter et al., 2024] Daniel Winter, Niv Cohen, and Yedid Hoshen. Classifying nodes in graphs without gnns. arXiv preprint arXiv:2402.05934, 2024.
  38. [Wu et al., 2020] Zonghan Wu, Shirui Pan, F. Chen, G. Long, Chengqi Zhang, and S Yu Philip. A comprehen- sive survey on graph neural networks. IEEE transactions on neural networks and learning systems, 32:4-24, 2020.
  39. [Wu et al., 2022a] Lirong Wu, Haitao Lin, Yufei Huang, and Stan Z Li. Knowledge distillation improves graph struc- ture augmentation for graph neural networks. Neural In- formation Processing Systems, 35:11815-11827, 2022.
  40. [Wu et al., 2022b] Lirong Wu, Jun Xia, Haitao Lin, Zhangyang Gao, Zicheng Liu, Guojiang Zhao, and Stan Z Li. Teaching yourself: Graph self-distillation on neighborhood for node classification. arXiv preprint arXiv:2210.02097, 2022.
  41. [Wu et al., 2023a] Lirong Wu, Haitao Lin, Yufei Huang, Tianyu Fan, and Stan Z Li. Extracting low-/high-frequency knowledge from graph neural networks and injecting it into mlps: An effective gnn-to-mlp distillation framework. In AAAI Conference on Artificial Intelligence, volume 37, pages 10351-10360, 2023.
  42. [Wu et al., 2023b] Lirong Wu, Haitao Lin, Yufei Huang, and Stan Z Li. Quantifying the knowledge in gnns for reli- able distillation into mlps. In International Conference on Machine Learning, pages 37571-37581, 2023.
  43. [Wu et al., 2023c] Taiqiang Wu, Zhe Zhao, Jiahao Wang, Xingyu Bai, Lei Wang, Ngai Wong, and Yujiu Yang. Edge-free but structure-aware: Prototype-guided knowl- edge distillation from gnns to mlps. arXiv preprint arXiv:2303.13763, 2023.
  44. [Xu et al., 2019] Keyulu Xu, Weihua Hu, Jure Leskovec, and Stefanie Jegelka. How powerful are graph neural net- works? In International Conference on Learning Rep- resentations, 2019.
  45. [Yan et al., 2020] Bencheng Yan, C. Wang, G. Guo, and Yunkai Lou. Tinygnn: Learning efficient graph neural net- works. In ACM International Conference on Knowledge Discovery & Data Mining, pages 1848-1856, 2020.
  46. [Yang et al., 2020] Yiding Yang, Jiayan Qiu, Mingli Song, Dacheng Tao, and Xinchao Wang. Distilling knowledge from graph convolutional networks. In conference on com- puter vision and pattern recognition, pages 7074-7083, 2020.
  47. [Zhang et al., 2020] Wentao Zhang, Xupeng Miao, Yingxia Shao, Jiawei Jiang, Lei Chen, Olivier Ruas, and Bin Cui. Reliable data distillation on graph convolutional network. In ACM SIGMOD international conference on manage- ment of data, pages 1399-1414, 2020.
  48. [Zhang et al., 2022a] Shichang Zhang, Yozen Liu, Yizhou Sun, and Neil Shah. Graph-less neural networks: Teach- ing old mlps new tricks via distillation. In International Conference on Learning Representations, 2022.
  49. [Zhang et al., 2022b] Yanfu Zhang, Shangqian Gao, Jian Pei, and Heng Huang. Improving social network embed- ding via new second-order continuous graph neural net- works. In ACM conference on knowledge discovery and data mining, pages 2515-2523, 2022.
  50. [Zhao et al., 2020] Yiren Zhao, Duo Wang, Daniel Bates, Robert Mullins, Mateja Jamnik, and Pietro Lio. Learned low precision graph neural networks. arXiv preprint arXiv:2009.09232, 2020.

References were extracted automatically from the PDF and may contain errors.

Welcome Future Conferences AI Journal Division IJCAI Proceedings About IJCAI IJCAI Awards