arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2403.03465cs.LGcs.SI

用于结构学习和节点嵌入的自注意力增强图卷积网络

Self-Attention Empowered Graph Convolutional Network for Structure Learning and Node Embedding

School of Electronic and Information Engineering, Xi’an Jiaotong University(西安交通大学电子与信息工程学院);College of Geomatics, Xi’an University of Science and Technology(西安科技大学测绘学院)

Mengying Jiang, Guizhong Liu, Yuanchao Su, Xinliang Wu

AI总结:

针对图神经网络难捕捉长程依赖、异配图下性能下降的问题,提出融合自注意力的GCN-SA框架,通过图结构学习模块与改进transformer块捕捉长程依赖,在不同同配性图的节点表示学习上表现优异。

摘要:

在图结构数据的表示学习中,许多主流图神经网络(GNNs)无法捕捉长程依赖关系,导致性能下降。此外,当所研究的图具有异配性(低同配性)特征时,这一缺陷会被放大。为解决该问题,本文提出了一种名为带自注意力的图卷积网络(GCN-SA)的新型图学习框架。该方案在节点级表示学习中展现出卓越的泛化能力。所提出的GCN-SA包含分别对应边和节点特征的两项改进。针对边,我们利用自注意力机制设计了一个稳定有效的图结构学习模块,该模块能够捕捉任意节点对之间的内在关联,可从全图范围内为每个节点识别可靠的邻居。针对节点特征,我们对transformer块进行了修改,使其更适用于让GCN融合来自全图的有价值信息。这两项改进以不同方式助力GCN-SA捕捉长程依赖,使其能够在同配性水平各异的图上开展表示学习。在基准数据集上的实验结果证明了所提GCN-SA的有效性,与其他优秀的GNN同类方法相比,该模型具备竞争力。

英文摘要:

In representation learning on graph-structured data, many popular graph neural networks (GNNs) fail to capture long-range dependencies, leading to performance degradation. Furthermore, this weakness is magnified when the concerned graph is characterized by heterophily (low homophily). To solve this issue, this paper proposes a novel graph learning framework called the graph convolutional network with self-attention (GCN-SA). The proposed scheme exhibits an exceptional generalization capability in node-level representation learning. The proposed GCN-SA contains two enhancements corresponding to edges and node features. For edges, we utilize a self-attention mechanism to design a stable and effective graph-structure-learning module that can capture the internal correlation between any pair of nodes. This graph-structure-learning module can identify reliable neighbors for each node from the entire graph. Regarding the node features, we modify the transformer block to make it more applicable to enable GCN to fuse valuable information from the entire graph. These two enhancements work in distinct ways to help our GCN-SA capture long-range dependencies, enabling it to perform representation learning on graphs with varying levels of homophily. The experimental results on benchmark datasets demonstrate the effectiveness of the proposed GCN-SA. Compared to other outstanding GNN counterparts, the proposed GCN-SA is competitive.

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

↑