Graph Neural Networks (GNNs) have proven to be powerful in many graph-based applications. However, they fail to generalize well under heterophilic setups, where neighbor nodes have different labels. To address this challenge, we employ a confidence ratio as a hyper-parameter, assuming that some of the edges are disassortative (heterophilic). Here, we propose a two-phased algorithm. Firstly, we determine edge coefficients through subgraph matching using a supplementary module. Then, we apply GNNs with a modified label propagation mechanism to utilize the edge coefficients effectively. Specifically, our supplementary module identifies a certain proportion of task-irrelevant edges based on a given confidence ratio. Using the remaining edges, we employ the widely used optimal transport to measure the similarity between two nodes with their subgraphs. Finally, using the coefficients as supplementary information on GNNs, we improve the label propagation mechanism which can prevent two nodes with smaller weights from being closer. The experiments on benchmark datasets show that our model alleviates over-smoothing and improves performance.
翻译:----
图神经网络在许多基于图的应用中表现出了强大的能力。但是,在异质设置下,GNN很难实现良好的泛化,其中邻居节点具有不同的标签。为了解决这个问题,我们利用置信度比作为超参数,假设一些边是异构的。本文提出了一个两阶段的算法。首先,我们通过子图匹配确定边的系数,使用辅助模块。然后,我们应用GNN,并修改标签传播机制以有效地利用边的系数。具体来说,我们的辅助模块基于给定的置信度比率,确定某些比例的与任务无关的边。使用剩余的边,我们采用广泛使用的最优传输方法来测量两个节点及其子图之间的相似性。最后,使用边的系数作为GNN的补充信息,我们改进了标签传播机制,可以防止两个权重较小的节点被认为很接近。在基准数据集上的实验证明,我们的模型缓解了过度平滑,并提高了性能。