References & Citations
Computer Science > Machine Learning
Title: Graph Neural Convection-Diffusion with Heterophily
(Submitted on 26 May 2023 (v1), last revised 30 May 2023 (this version, v2))
Abstract: 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 \url{this https URL}.
Submission history
From: Kai Zhao [view email][v1] Fri, 26 May 2023 09:47:03 GMT (475kb,D)
[v2] Tue, 30 May 2023 12:25:39 GMT (482kb,D)
Link back to: arXiv, form interface, contact.