Graph Neural Networks (GNNs) have become the de-facto standard tool for modeling relational data. However, while many real-world graphs are directed, the majority of today's GNN models discard this information altogether by simply making the graph undirected. The reasons for this are historical: 1) many early variants of spectral GNNs explicitly required undirected graphs, and 2) the first benchmarks on homophilic graphs did not find significant gain from using direction. In this paper, we show that in heterophilic settings, treating the graph as directed increases the effective homophily of the graph, suggesting a potential gain from the correct use of directionality information. To this end, we introduce Directed Graph Neural Network (Dir-GNN), a novel general framework for deep learning on directed graphs. Dir-GNN can be used to extend any Message Passing Neural Network (MPNN) to account for edge directionality information by performing separate aggregations of the incoming and outgoing edges. We prove that Dir-GNN matches the expressivity of the Directed Weisfeiler-Lehman test, exceeding that of conventional MPNNs. In extensive experiments, we validate that while our framework leaves performance unchanged on homophilic datasets, it leads to large gains over base models such as GCN, GAT and GraphSage on heterophilic benchmarks, outperforming much more complex methods and achieving new state-of-the-art results.
翻译:图神经网络(GNNs)已成为建模关系数据的事实标准工具。然而,尽管许多真实世界图是有向的,当前大多数GNN模型通过简单地将图转换为无向图而完全丢弃了这一信息。其原因具有历史性:1)许多早期谱域GNN变体明确要求无向图,2)首个同质图上的基准测试并未发现利用方向性带来的显著收益。本文证明,在异质图场景中,将有向图处理为有向形式可提升图的有效同质性,表明正确利用方向性信息具有潜在增益。为此,我们提出有向图神经网络(Dir-GNN),一种用于有向图深度学习的新型通用框架。Dir-GNN可通过分别聚合入边和出边信息,将任意消息传递神经网络(MPNN)扩展为能处理边方向性的模型。我们证明Dir-GNN的表达能力等价于有向Weisfeiler-Lehman测试,超越了传统MPNN。在大量实验中,我们验证了该框架在同质数据集上保持性能不变的同时,在异质基准测试中相较于GCN、GAT和GraphSage等基础模型取得显著提升,超越更复杂的方法并达到最新最优结果。