STERLING: Synergistic Representation Learning on Bipartite Graphs

Baoyu Jing, Yuchen Yan, Kaize Ding, Chanyoung Park, Yada Zhu, Huan Liu, Hanghang Tong

Research output: Contribution to journalConference articlepeer-review

4 Scopus citations

Abstract

A fundamental challenge of bipartite graph representation learning is how to extract informative node embeddings. Self-Supervised Learning (SSL) is a promising paradigm to address this challenge. Most recent bipartite graph SSL methods are based on contrastive learning which learns embeddings by discriminating positive and negative node pairs. Contrastive learning usually requires a large number of negative node pairs, which could lead to computational burden and semantic errors. In this paper, we introduce a novel synergistic representation learning model (STERLING) to learn node embeddings without negative node pairs. STERLING preserves the unique local and global synergies in bipartite graphs. The local synergies are captured by maximizing the similarity of the inter-type and intra-type positive node pairs, and the global synergies are captured by maximizing the mutual information of co-clusters. Theoretical analysis demonstrates that STERLING could improve the connectivity between different node types in the embedding space. Extensive empirical evaluation on various benchmark datasets and tasks demonstrates the effectiveness of STERLING for extracting node embeddings.

Original languageEnglish (US)
Pages (from-to)12976-12984
Number of pages9
JournalProceedings of the AAAI Conference on Artificial Intelligence
Volume38
Issue number12
DOIs
StatePublished - Mar 25 2024
Event38th AAAI Conference on Artificial Intelligence, AAAI 2024 - Vancouver, Canada
Duration: Feb 20 2024Feb 27 2024

Funding

This work is partly supported by NSF (#2229461), DARPA (HR001121C0165), NIFA (2020-67021-32799), MIT-IBM Watson AI Lab, and IBM-Illinois Discovery Accelerator Institute. The content of the information in this document does not necessarily reflect the position or the policy of the Government or Amazon, and no official endorsement should be inferred. The U.S. Government is authorized to reproduce and distribute reprints for Government purposes notwithstanding any copyright notation here on. Dr. Chanyoung Park is supported by the IITP grant funded by the Korea government (MSIT) (RS-2023-00216011).

ASJC Scopus subject areas

  • Artificial Intelligence

Fingerprint

Dive into the research topics of 'STERLING: Synergistic Representation Learning on Bipartite Graphs'. Together they form a unique fingerprint.

Cite this