Multi-relational graph contrastive learning with learnable graph augmentation

Xian Mo,Jun Pang,Binyuan Wan,Rui Tang,Hao Liu,Shuyu Jiang
DOI: https://doi.org/10.1016/j.neunet.2024.106757
IF: 7.8
2024-10-03
Neural Networks
Abstract:Multi-relational graph learning aims to embed entities and relations in knowledge graphs into low-dimensional representations, which has been successfully applied to various multi-relationship prediction tasks, such as information retrieval, question answering, and etc. Recently, contrastive learning has shown remarkable performance in multi-relational graph learning by data augmentation mechanisms to deal with highly sparse data. In this paper, we present a M ulti- R elational G raph C ontrastive L earning architecture (MRGCL) for multi-relational graph learning. More specifically, our MRGCL first proposes a M ulti-relational G raph H ierarchical A ttention N etworks (MGHAN) to identify the importance between entities, which can learn the importance at different levels between entities for extracting the local graph dependency. Then, two graph augmented views with adaptive topology are automatically learned by the variant MGHAN, which can automatically adapt for different multi-relational graph datasets from diverse domains. Moreover, a subgraph contrastive loss is designed, which generates positives per anchor by calculating strongly connected subgraph embeddings of the anchor as the supervised signals. Comprehensive experiments on multi-relational datasets from three application domains indicate the superiority of our MRGCL over various state-of-the-art methods. Our datasets and source code are published at https://github.com/Legendary-L/MRGCL .
computer science, artificial intelligence,neurosciences
What problem does this paper attempt to address?