arXiv:2411.05119cs.LGeess.SP2024-11被引 1

用双图结构提升GNN在多图场景下的表现

Exploiting the Structure of Two Graphs with Graph Neural Networks

  • 设计三模块架构,处理输入输出分别定义在不同图上的信号
  • 在真实与合成数据上均优于传统深度学习模型
  • 适合需要跨图关系建模的任务,如多源数据融合

图神经网络(GNN)在处理非结构化数据方面表现出色,超越了传统深度学习架构。然而,现有大多数GNN模型仅适用于单一图,限制了其在涉及多个图的现实场景中的应用。为此,我们提出一种新型基于图的深度学习架构,用于处理两个不同图上定义的信号。首先考虑输入信号定义在一个图(输入图)上,输出信号定义在另一个图(输出图)上。我们设计了一个三块式架构:先用作用于输入图的GNN处理输入数据;再通过潜空间中的变换函数将信号从输入图映射到输出图;最后在输出图上运行第二个GNN。该方法不固定每个模块的具体形式,而是提供一种灵活的双图任务解决方案。第二部分引入自监督设定,关注潜在表示而非输出空间,受典型相关分析启发,寻找可用于下游任务的有信息量的表示。通过利用两个图的信息,所提架构能捕捉更复杂的实体间关系。我们在合成与真实数据集上进行多组实验,结果表明该架构优于传统深度学习方法,证明了双图信息融合的重要性。

原文摘要 · Abstract (English)

Graph neural networks (GNNs) have emerged as a promising solution to deal with unstructured data, outperforming traditional deep learning architectures. However, most of the current GNN models are designed to work with a single graph, which limits their applicability in many real-world scenarios where multiple graphs may be involved. To address this limitation, we propose a novel graph-based deep learning architecture to handle tasks where two sets of signals exist, each defined on a different graph. First we consider the setting where the input is represented as a signal on top of one graph (input graph) and the output is a graph signal defined over a different graph (output graph). For this setup, we propose a three-block architecture where we first process the input data using a GNN that operates over the input graph, then apply a transformation function that operates in a latent space and maps the signals from the input to the output graph, and finally implement a second GNN that operates over the output graph. Our goal is not to propose a single specific definition for each of the three blocks, but rather to provide a flexible approach to solve tasks involving data defined on two graphs. The second part of the paper addresses a self-supervised setup, where the focus is not on the output space but on the underlying latent space and, inspired by Canonical Correlation Analysis, we seek informative representations of the data that can be leveraged to solve a downstream task. By leveraging information from multiple graphs, the proposed architecture can capture more intricate relationships between different entities in the data. We test this in several experimental setups using synthetic and real world datasets, and observe that the proposed architecture works better than traditional deep learning architectures, showcasing the importance of leveraging the information of the two graphs.

图神经网络双图建模自监督学习

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。