NHGMI: Heterogeneous graph multi-view infomax with node-wise contrasting samples selection

Qing Li, Hang Ni, Yuanchun Wang

Research output: Contribution to journalArticlepeer-review

3 Scopus citations

Abstract

Heterogeneous Graph Neural Networks (HGNNs) have consistently demonstrated exceptional performance in the context of Heterogeneous Information Networks (HINs). However, the majority of existing methods are constrained by their applicability to semi-supervised learning scenarios, which entail substantial labor and resources. Recently, the advent of contrastive learning has catalyzed the development of self-supervised HGNN approaches. In this study, we introduce NHGMI, a novel unsupervised method for heterogeneous graph representation learning, rooted in the principles of mutual information maximization. NHGMI comprehensively harnesses the inherent characteristics of HINs and adeptly amalgamates diverse semantic information emanating from multiple contrasting perspectives. In addition to inter-view contrasts, NHGMI also incorporates intra-view contrasts. Diverging from conventional contrastive learning approaches, NHGMI meticulously selects contrasting samples based on similarity metrics, thereby achieving a noise-free contrastive learning paradigm. Furthermore, we propose an extension of NHGMI that focuses on the generation of negative samples. We conduct extensive experiments across various real-world datasets, and NHGMI consistently exhibits marked superiority when compared to relevant methods, even surpassing semi-supervised approaches. The source code for NHGMI is publicly available at https://github.com/Frederick-the-Fox/NHGMI.

Original languageEnglish
Article number111520
JournalKnowledge-Based Systems
Volume289
DOIs
StatePublished - 8 Apr 2024

Keywords

  • Contrastive learning
  • Graph neural network
  • Heterogeneous graph
  • Infomax
  • Node-wise contrasting sample

Fingerprint

Dive into the research topics of 'NHGMI: Heterogeneous graph multi-view infomax with node-wise contrasting samples selection'. Together they form a unique fingerprint.

Cite this