NOSMOG: Learning Noise-robust and Structure-aware MLPs on Graphs
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 multi-hop data dependency. Existing methods attempt to address this scalability issue by training multi-layer perceptrons (MLPs) exclusively on node content features using labels derived from trained GNNs. Even though the performance of MLPs can be significantly improved, two issues prevent MLPs from outperforming GNNs and being used in practice: the ignorance of graph structural information and the sensitivity to node feature noises. In this paper, we propose to learn ise-robust tructure-aware LPs n raphs (NOSMOG) to overcome the challenges. Specifically, we first complement node content with position features to help MLPs capture graph structural information. We then design a novel representational similarity distillation strategy to inject structural node similarities into MLPs. Finally, we introduce the adversarial feature augmentation to ensure stable learning against feature noises and further improve performance. Extensive experiments demonstrate that NOSMOG outperforms GNNs and the state-of-the-art method in both transductive and inductive settings across seven datasets, while maintaining a competitive inference efficiency. Codes are available at.
Benchmark Results
Tested on 2 datasets
Tables show presented results in global context (best 3 papers + GCN + MLP).
Classic
| # | Model | Accuracy ↑ | Paper |
|---|---|---|---|
| 1 | GraphSAGE+CNA | 0.9418
±0.0033 | |
| 2 | GCN+CNA | 0.9366
±0.0048 | |
| 3 | TransformerConv | 0.9350
±0.0058 | |
| 21 | GCN | 0.9004 | |
| 348 | GLNN | 0.8360
±0.0022 best reported | best at pub: Graph-less Neural Networks: Teaching Old MLPs New Tricks via Distillation (ICLR 2021) best today: Frameless Graph Knowledge Distillation (2023) |
| 424 | NOSMOG | 0.8304
±0.0126 | This paper |
| 724 | MLP | 0.7768 |
| # | Model | Accuracy ↑ | Paper |
|---|---|---|---|
| 1 | CNA | 0.9575
±0.0058 | |
| 2 | IS-GIB | 0.9390
±0.0187 | |
| 3 | EERM | 0.9112
±0.0145 | |
| 44 | GCN | 0.8033
±0.0088 | |
| 229 | GLNN | 0.7400
±0.0002 best reported | best at pub: Graph-less Neural Networks: Teaching Old MLPs New Tricks via Distillation (ICLR 2021) best today: Frameless Graph Knowledge Distillation (2023) |
| 246 | NOSMOG | 0.7378
±0.0154 | This paper |
| 259 | SAGE | 0.7361
±0.0190 best reported | best at pub: Extract the Knowledge of Graph Neural Networks and Go Beyond it: An Effective Knowledge Distillation Framework (WWW 2021) best today: Efficient Tuning and Inference for Large Language Models on Textual Graphs (IJCAI 2024) |
| 350 | MLP | 0.7289
±0.0010 |
