arXiv:2506.02619cs.LGcs.AI2025-06ICML被引 3

无需数据增强,用最优传输提升异构图自监督学习效果

HGOT: Self-supervised Heterogeneous Graph Neural Network with Optimal Transport

  • 通过最优传输机制自动匹配不同元路径视图与中心视图
  • 节点分类平均准确率提升超6%,优于当前最佳方法
  • 适合无标签异构图数据,避免人工设计正负样本

异构图神经网络在处理异构信息网络方面表现出色。当缺乏标签时,基于对比的自监督学习策略展现出巨大潜力,但需精心设计图增强方法并选择正负样本,而样本间相似度的精确判定极具挑战。为此,我们提出一种新型自监督异构图神经网络(HGOT),无需图增强策略即可实现高效自监督学习。不同于传统对比学习,HGOT采用最优传输机制,简化正负样本采样过程。具体而言,设计一个聚合视图(中心视图)以整合由不同元路径(分支视图)表示的语义信息,并引入最优传输计划,识别分支视图与中心视图间的语义传输关系。该传输计划使图间表示对齐,迫使编码器学习更贴近图空间、质量更高的节点表示。在四个真实世界数据集上的大量实验表明,所提HGOT模型在多种下游任务中达到领先性能,尤其在节点分类任务中,平均准确率较现有最优方法提升超过6%。

原文摘要 · Abstract (English)

Heterogeneous Graph Neural Networks (HGNNs), have demonstrated excellent capabilities in processing heterogeneous information networks. Self-supervised learning on heterogeneous graphs, especially contrastive self-supervised strategy, shows great potential when there are no labels. However, this approach requires the use of carefully designed graph augmentation strategies and the selection of positive and negative samples. Determining the exact level of similarity between sample pairs is non-trivial.To solve this problem, we propose a novel self-supervised Heterogeneous graph neural network with Optimal Transport (HGOT) method which is designed to facilitate self-supervised learning for heterogeneous graphs without graph augmentation strategies. Different from traditional contrastive self-supervised learning, HGOT employs the optimal transport mechanism to relieve the laborious sampling process of positive and negative samples. Specifically, we design an aggregating view (central view) to integrate the semantic information contained in the views represented by different meta-paths (branch views). Then, we introduce an optimal transport plan to identify the transport relationship between the semantics contained in the branch view and the central view. This allows the optimal transport plan between graphs to align with the representations, forcing the encoder to learn node representations that are more similar to the graph space and of higher quality. Extensive experiments on four real-world datasets demonstrate that our proposed HGOT model can achieve state-of-the-art performance on various downstream tasks. In particular, in the node classification task, HGOT achieves an average of more than 6% improvement in accuracy compared with state-of-the-art methods.

异构图自监督最优传输节点分类

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