Learning to Model Graph Structural Information on MLPs via Graph Structure Self-Contrasting

Lirong Wu, Haitao Lin, Guojiang Zhao, Cheng Tan, Stan Z. Li
9/9/2024
cs.LGcs.AI

No official code found. Best implementation: LirongWu/GSSC (1 star, MIT, last commit 2024-12-11).

Abstract

Recent years have witnessed great success in handling graph-related tasks with Graph Neural Networks (GNNs). However, most existing GNNs are based on message passing to perform feature aggregation and transformation, where the structural information is explicitly involved in the forward propagation by coupling with node features through graph convolution at each layer. As a result, subtle feature noise or structure perturbation may cause severe error propagation, resulting in extremely poor robustness. In this paper, we rethink the roles played by graph structural information in graph data training and identify that message passing is not the only path to modeling structural information. Inspired by this, we propose a simple but effective Graph Structure Self-Contrasting (GSSC) framework that learns graph structural information without message passing. The proposed framework is based purely on Multi-Layer Perceptrons (MLPs), where the structural information is only implicitly incorporated as prior knowledge to guide the computation of supervision signals, substituting the explicit message propagation as in GNNs. Specifically, it first applies structural sparsification to remove potentially uninformative or noisy edges in the neighborhood, and then performs structural self-contrasting in the sparsified neighborhood to learn robust node representations. Finally, structural sparsification and self-contrasting are formulated as a bi-level optimization problem and solved in a unified framework. Extensive experiments have qualitatively and quantitatively demonstrated that the GSSC framework can produce truly encouraging performance with better generalization and robustness than other leading competitors.

View on arXivView PDF

Code Implementations(1)

LirongWu/GSSCCommunity100%
MIT

Code for TNNLS paper "Learning to Model Graph Structural Information on MLPs via Graph Structure Self-Contrasting"

10PythonSep 9, 20241 years agoMIT
graph-algorithmsgraph-contrastive-learninggraph-modelinggraph-neural-networks

Similar Papers

Cite this paper

@article{wu2024learning,
  title  = {Learning to Model Graph Structural Information on MLPs via Graph Structure Self-Contrasting},
  author = {Lirong Wu and Haitao Lin and Guojiang Zhao and Cheng Tan and Stan Z. Li},
  year   = {2024},
  eprint = {2409.05573},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2409.05573v1}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2409.05573.svg)](https://codeofpaper.com/paper/2409.05573)

Discussion