Learning MLPs on Graphs: A Unified View of Effectiveness, Robustness, and Efficiency
Yijun Tian, Chuxu Zhang, Zhichun Guo, Xiangliang Zhang, Nitesh V. Chawla
Abstract
While Graph Neural Networks (GNNs) have demonstrated their efficacy in dealing with non-Euclidean structural data, they are difficult to be deployed in real applications due to the scalability constraint imposed by the multi-hop data dependency. Existing methods attempt to address this scalability issue by training student multi-layer perceptrons (MLPs) exclusively on node content features using labels derived from the teacher GNNs. However, the trained MLPs are neither effective nor robust. In this paper, we ascribe the lack of effectiveness and robustness to three significant challenges: 1) the misalignment between content feature and label spaces, 2) the strict hard matching to teacher's output, and 3) the sensitivity to node feature noises. To address the challenges, we propose NOSMOG, a novel method to learn NOise-robust Structure-aware MLPs On Graphs, with remarkable effectiveness, robustness, and efficiency. Specifically, we first address the misalignment by complementing node content with position features to capture the graph structural information. We then design an innovative representational similarity distillation strategy to inject soft node similarities into MLPs. Finally, we introduce adversarial feature augmentation to ensure stable learning against feature noises. Extensive experiments and theoretical analyses demonstrate the superiority of NOSMOG by comparing it to GNNs and the state-of-the-art method in both transductive and inductive settings across seven datasets. Codes are available at https://github.com/meettyj/NOSMOG.
Ask Lune
Ask anything. The answer shows its sources.
Lune reads the papers around it, then shows you the passages behind every answer.
Lune answers from
its abstract · 25 top-tier papers that cite it · related work
Free to start. No credit card required.
Cited by top-tier papers25
- Logit Standardization in Knowledge DistillationShangquan Sun, Wenqi Ren, Jingzhi Li, Rui Wang et al.CVPR 2024 · 183 citations
- Graph Neural Prompting with Large Language ModelsYijun Tian, Huan Song, Zichen Wang, Haozhu Wang et al.AAAI 2024 · 90 citations
- VQGraph: Rethinking Graph Representation Space for Bridging GNNs and MLPsLing Yang, Ye Tian, Minkai Xu, Zhongyi Liu et al.ICLR 2024 · 48 citations
- Fair Graph Representation Learning via Diverse Mixture-of-ExpertsZheyuan Liu, Chunhui Zhang, Yijun Tian, Erchi Zhang et al.WWW 2023 · 43 citations
- S3GCL: Spectral, Swift, Spatial Graph Contrastive LearningGuancheng Wan, Yijun Tian, Wenke Huang, Nitesh V. Chawla et al.ICML 2024 · 26 citations
Related papers
- NRGNN: Learning a Label Noise Resistant Graph Neural Network on Sparsely and Noisily Labeled GraphsEnyan Dai, Charu Aggarwal, Suhang WangKDD 2021 · 80 citations
- Leap of FAITH from GNN-to-MLP: Fairness Aware Inference via DisTillation of GrapH KnowledgeVipul Kumar Singh, Jyotismita Barman, Sandeep Kumar, Tapan K. Gandhi et al.AAAI 2026
- TAG2M- A Task-Agnostic Knowledge Distillation Framework for Distilling GNN to MLPRam Ganesh V, Ayush Singh, Aditi Rai, Harsh Pal et al.KDD 2025
- Quantifying the Knowledge in GNNs for Reliable Distillation into MLPsLirong Wu, Haitao Lin, Yufei Huang, Stan Z. LiICML 2023 · 48 citations
- AdaGMLP: AdaBoosting GNN-to-MLP Knowledge DistillationWeigang Lu, Ziyu Guan, Wei Zhao, Yaming YangKDD 2024 · 10 citations
