论文标题

VR-GNN:用于对同质和异质化建模的变异关系矢量图神经网络

VR-GNN: Variational Relation Vector Graph Neural Network for Modeling both Homophily and Heterophily

论文作者

Shi, Fengzhao, Li, Ren, Cao, Yanan, Shang, Yanmin, Zhang, Lanxue, Zhou, Chuan, Wu, Jia, Pan, Shirui

论文摘要

图形神经网络(GNN)在不同的现实世界应用中取得了显着成功。传统的GNN是基于同质性设计的,这在异性场景下导致性能差。当前的解决方案主要是通过混合高阶邻居或传递签名消息来处理异性的。但是,混合高阶邻居会破坏原始的图形结构并传递签名的消息,利用了不灵活的消息通话机制,这很容易产生不令人满意的效果。为了克服上述问题,我们提出了一个基于关系矢量翻译的新型GNN模型,称为变分关系矢量图神经网络(VR-GNN)。 VR-GNN将关系生成和图形聚集模型为基于变异自动编码器的端到端模型。编码器利用结构,功能和标签来生成适当的关系向量。解码器通过将关系翻译合并到消息通话框架中来实现上级节点表示。 VR-GNN可以在节点之间在节点之间完全捕获同质和异质性,这是因为在建模邻居关系中的关系转换的灵活性。我们对具有不同同质性特性的八个现实世界数据集进行了广泛的实验,以验证我们的模型的有效性。实验结果表明,VR-GNN在异性疾病下对最新的GNN方法以及同质性的竞争性能取得了一致和显着的改进。

Graph Neural Networks (GNNs) have achieved remarkable success in diverse real-world applications. Traditional GNNs are designed based on homophily, which leads to poor performance under heterophily scenarios. Current solutions deal with heterophily mainly by mixing high-order neighbors or passing signed messages. However, mixing high-order neighbors destroys the original graph structure and passing signed messages utilizes an inflexible message-passing mechanism, which is prone to producing unsatisfactory effects. To overcome the above problems, we propose a novel GNN model based on relation vector translation named Variational Relation Vector Graph Neural Network (VR-GNN). VR-GNN models relation generation and graph aggregation into an end-to-end model based on Variational Auto-Encoder. The encoder utilizes the structure, feature and label to generate a proper relation vector. The decoder achieves superior node representation by incorporating the relation translation into the message-passing framework. VR-GNN can fully capture the homophily and heterophily between nodes due to the great flexibility of relation translation in modeling neighbor relationships. We conduct extensive experiments on eight real-world datasets with different homophily-heterophily properties to verify the effectiveness of our model. The experimental results show that VR-GNN gains consistent and significant improvements against state-of-the-art GNN methods under heterophily, and competitive performance under homophily.

扫码加入交流群

加入微信交流群

微信交流群二维码

扫码加入学术交流群,获取更多资源