MixSp: A Framework for Embedding Heterogeneous Information Networks with Arbitrary Number of Node and Edge Types

Linchuan Xu, Jing Wang, Lifang He, Jiannong Cao, Xiaokai Wei, Philip S. Yu, Kenji Yamanishi

Research output: Journal article publicationJournal articleAcademic researchpeer-review

1 Citation (Scopus)

Abstract

Heterogeneous information network (HIN) embedding is to encode network structure into node representations with the heterogeneous semantics of different node and edge types considered. However, since each HIN may have a unique nature, e.g., a unique set of node and edge types, a model designed for one type of networks may not be applicable to or effective on another type. In this article, we thus attempt to propose a framework for HINs with arbitrary number of node and edge types. The proposed framework constructs a novel mixture-split representation of an HIN, and hence is named as MixSp. The mixture sub-representation and the split sub-representation serve as two different views of the network. Compared with existing models which only learn from the original view, MixSp thus may exploit more comprehensive information. Node representations in each view are learned by embedding the respective network structure. Moreover, the node representations are further refined through cross-view co-regularization. The framework is instantiated in three models which differ from each other in the co-regularization. Extensive experiments on three real-world datasets show MixSp outperforms several recent models in both node classification and link prediction tasks even though MixSp is not designed for a particular type of HINs.

Original languageEnglish
Article number8913597
Pages (from-to)2627-2639
Number of pages13
JournalIEEE Transactions on Knowledge and Data Engineering
Volume33
Issue number6
DOIs
Publication statusPublished - 1 Jun 2021

Keywords

  • heterogeneous information networks
  • link prediction
  • multi-label classification
  • Network embedding

ASJC Scopus subject areas

  • Information Systems
  • Computer Science Applications
  • Computational Theory and Mathematics

Cite this