Weisfeiler and Lehman Go Paths: Learning Topological Features via Path Complexes

13 Aug 2023  ·  Quang Truong, Peter Chin ·

Graph Neural Networks (GNNs), despite achieving remarkable performance across different tasks, are theoretically bounded by the 1-Weisfeiler-Lehman test, resulting in limitations in terms of graph expressivity. Even though prior works on topological higher-order GNNs overcome that boundary, these models often depend on assumptions about sub-structures of graphs. Specifically, topological GNNs leverage the prevalence of cliques, cycles, and rings to enhance the message-passing procedure. Our study presents a novel perspective by focusing on simple paths within graphs during the topological message-passing process, thus liberating the model from restrictive inductive biases. We prove that by lifting graphs to path complexes, our model can generalize the existing works on topology while inheriting several theoretical results on simplicial complexes and regular cell complexes. Without making prior assumptions about graph sub-structures, our method outperforms earlier works in other topological domains and achieves state-of-the-art results on various benchmarks.

PDF Abstract

Results from the Paper


Task Dataset Model Metric Name Metric Value Global Rank Result Benchmark
Graph Classification IMDb-B PIN Accuracy 76.6% # 9
Graph Classification NCI1 PIN Accuracy 85.1% # 12
Graph Classification NCI109 PIN Accuracy 84.0 # 5
Graph Property Prediction ogbg-molhiv PIN Test ROC-AUC 0.7944 ± 1.40 # 21
Graph Classification PROTEINS PIN Accuracy 78.8% # 11
Graph Regression ZINC PIN MAE 0.096 # 15

Methods


No methods listed for this paper. Add relevant methods here