arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2404.09809cs.LGcs.CV
收藏

用于改进图表示学习的邻居级消息交互编码

Neighbour-level Message Interaction Encoding for Improved Representation Learning on Graphs

School of Electrical and Data Engineering, Faculty of Engineering and Information Technology, University of Technology Sydney(悉尼科技大学工程与信息技术学院工程与数据工程学院)

Haimin Zhang, Min Xu

AI总结:

针对图消息传递框架中未编码邻居级消息交互导致信息损失的问题,提出通用的邻居级消息交互编码方法,可集成到图卷积网络中,在6个基准数据集的4项任务上提升了基础模型性能,刷新图表示学习SOTA。

摘要:

消息传递已成为图表示学习中的主流框架。消息传递框架的核心思想是基于从局部邻居聚合的信息来更新节点嵌入。然而,现有大多数聚合方法未将邻居级的消息交互编码到聚合后的消息中,导致嵌入生成过程中出现信息损失。并且随着图网络模型层数增加,这种信息损失会不断累积并变得更加严重。为解决这一问题,我们提出一种邻居级消息交互信息编码方法,用于改进图表示学习。对于在某个节点处聚合的所有消息,我们使用编码函数显式生成每条消息与其余所有消息之间的编码。随后我们聚合这些学习到的编码,并将聚合后的编码与聚合消息相加,以此更新该节点的嵌入。通过这种方式,邻居级的消息交互信息被整合到生成的节点嵌入中。所提出的编码方法是一种通用方法,可集成到消息传递型图卷积网络中。我们在6个常用基准数据集上针对4项高需求任务开展了大量实验。结果表明,整合邻居级消息交互可提升基础模型的性能,刷新了图表示学习的当前最优结果。

英文摘要:

Message passing has become the dominant framework in graph representation learning. The essential idea of the message-passing framework is to update node embeddings based on the information aggregated from local neighbours. However, most existing aggregation methods have not encoded neighbour-level message interactions into the aggregated message, resulting in an information lost in embedding generation. And this information lost could be accumulated and become more serious as more layers are added to the graph network model. To address this issue, we propose a neighbour-level message interaction information encoding method for improving graph representation learning. For messages that are aggregated at a node, we explicitly generate an encoding between each message and the rest messages using an encoding function. Then we aggregate these learned encodings and take the sum of the aggregated encoding and the aggregated message to update the embedding for the node. By this way, neighbour-level message interaction information is integrated into the generated node embeddings. The proposed encoding method is a generic method which can be integrated into message-passing graph convolutional networks. Extensive experiments are conducted on six popular benchmark datasets across four highly-demanded tasks. The results show that integrating neighbour-level message interactions achieves improved performance of the base models, advancing the state of the art results for representation learning over graphs.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑