Graph Neural Convection-Diffusion with Heterophily

Graph Neural Convection-Diffusion with Heterophily

Kai Zhao, Qiyu Kang, Yang Song, Rui She, Sijie Wang, Wee Peng Tay

Proceedings of the Thirty-Second International Joint Conference on Artificial Intelligence
Main Track. Pages 4656-4664. https://doi.org/10.24963/ijcai.2023/518

Graph neural networks (GNNs) have shown promising results across various graph learning tasks, but they often assume homophily, which can result in poor performance on heterophilic graphs. The connected nodes are likely to be from different classes or have dissimilar features on heterophilic graphs. In this paper, we propose a novel GNN that incorporates the principle of heterophily by modeling the flow of information on nodes using the convection-diffusion equation (CDE). This allows the CDE to take into account both the diffusion of information due to homophily and the ``convection'' of information due to heterophily. We conduct extensive experiments, which suggest that our framework can achieve competitive performance on node classification tasks for heterophilic graphs, compared to the state-of-the-art methods. The code is available at https://github.com/zknus/Graph-Diffusion-CDE.
Keywords:
Machine Learning: ML: Sequence and graph learning
Machine Learning: ML: Classification