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

生成-对比异构图神经网络

Generative-Contrastive Heterogeneous Graph Neural Network

School of Computer Science and Technology, Anhui University(安徽大学计算机科学与技术学院);Key Laboratory of Knowledge Engineering with Big Data (the Ministry of Education of China), Hefei University of Technology(大数据知识工程重点实验室(教育部))

Yu Wang, Lei Sang, Yi Zhang, Yiwen Zhang, Xindong Wu

AI总结:

针对异构图神经网络中数据增强受限和对比判别器存在采样偏差、缺乏局部异构信息的问题,提出生成-对比异构图神经网络GC-HGNN,通过掩码自编码器视图增强、位置与语义感知采样及分层对比学习,在节点分类和链路预测任务上优于最新基线。

摘要:

异构图(HGs)通过多类型节点和边有效地建模现实世界中的复杂关系。近年来,受自监督学习(SSL)启发,基于对比学习(CL)的异构图神经网络(HGNNs)在利用数据增强和对比判别器进行下游任务方面展现出巨大潜力。然而,由于图数据的完整性,数据增强仍然受到限制。此外,对比判别器存在采样偏差,并且缺乏局部异构信息。为了解决上述局限性,我们提出了一种新颖的生成-对比异构图神经网络(GC-HGNN)。具体来说,我们提出了一种异构图生成学习方法,以增强基于对比学习的范式。该范式包括:1)使用掩码自编码器的对比视图增强策略。2)用于生成困难负样本的位置感知和语义感知正样本采样策略。3)旨在捕获局部和全局信息的分层对比学习策略。此外,分层对比学习和采样策略旨在从生成-对比视角构成一个增强的对比判别器。最后,我们在八个真实世界数据集上将我们的模型与十七个基线方法进行了比较。我们的模型在节点分类和链路预测任务上优于最新的基线方法。

英文摘要:

Heterogeneous Graphs (HGs) effectively model complex relationships in the real world through multi-type nodes and edges. In recent years, inspired by self-supervised learning (SSL), contrastive learning (CL)-based Heterogeneous Graphs Neural Networks (HGNNs) have shown great potential in utilizing data augmentation and contrastive discriminators for downstream tasks. However, data augmentation remains limited due to the graph data's integrity. Furthermore, the contrastive discriminators suffer from sampling bias and lack local heterogeneous information. To tackle the above limitations, we propose a novel Generative-Contrastive Heterogeneous Graph Neural Network (GC-HGNN). Specifically, we propose a heterogeneous graph generative learning method that enhances CL-based paradigm. This paradigm includes: 1) A contrastive view augmentation strategy using a masked autoencoder. 2) Position-aware and semantics-aware positive sample sampling strategy for generating hard negative samples. 3) A hierarchical contrastive learning strategy aimed at capturing local and global information. Furthermore, the hierarchical contrastive learning and sampling strategies aim to constitute an enhanced contrastive discriminator under the generative-contrastive perspective. Finally, we compare our model with seventeen baselines on eight real-world datasets. Our model outperforms the latest baselines on node classification and link prediction tasks.

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

↑