Analyzing Heterogeneous Networks With Missing Attributes by Unsupervised Contrastive Learning

Dongxiao He, Chundong Liang, Cuiying Huo, Zhiyong Feng, Di Jin, Liang Yang, Weixiong Zhang

Research output: Contribution to journalArticlepeer-review

20 Scopus citations

Abstract

Heterogeneous information networks (HINs) are potent models of complex systems. In practice, many nodes in an HIN have their attributes unspecified, resulting in significant performance degradation for supervised and unsupervised representation learning. We developed an unsupervised heterogeneous graph contrastive learning approach for analyzing HINs with missing attributes (HGCA). HGCA adopts a contrastive learning strategy to unify attribute completion and representation learning in an unsupervised heterogeneous framework. To deal with a large number of missing attributes and the absence of labels in unsupervised scenarios, we proposed an augmented network to capture the semantic relations between nodes and attributes to achieve a fine-grained attribute completion. Extensive experiments on three large real-world HINs demonstrated the superiority of HGCA over several state-of-the-art methods. The results also showed that the complemented attributes by HGCA can improve the performance of existing HIN models.

Original languageEnglish
Pages (from-to)4438-4450
Number of pages13
JournalIEEE Transactions on Neural Networks and Learning Systems
Volume35
Issue number4
DOIs
StatePublished - Apr 1 2024

Keywords

  • Contrastive learning
  • heterogeneous information networks (HINs)
  • missing data
  • unsupervised learning

Fingerprint

Dive into the research topics of 'Analyzing Heterogeneous Networks With Missing Attributes by Unsupervised Contrastive Learning'. Together they form a unique fingerprint.

Cite this