Node co-occurrence based graph neural networks for knowledge graph link prediction

Dai Quoc Nguyen, Vinh Tong, Dinh Phung, Dat Quoc Nguyen

Research output: Chapter in Book/Report/Conference proceedingConference PaperOtherpeer-review

26 Citations (Scopus)

Abstract

We introduce a novel embedding model, named NoGE, which aims to integrate co-occurrence among entities and relations into graph neural networks to improve knowledge graph completion (i.e., link prediction). Given a knowledge graph, NoGE constructs a single graph considering entities and relations as individual nodes. NoGE then computes weights for edges among nodes based on the co-occurrence of entities and relations. Next, NoGE proposes Dual Quaternion Graph Neural Networks (DualQGNN) and utilizes DualQGNN to update vector representations for entity and relation nodes. NoGE then adopts a score function to produce the triple scores. Comprehensive experimental results show that NoGE obtains state-of-the-art results on three new and difficult benchmark datasets CoDEx for knowledge graph completion.

Original languageEnglish
Title of host publicationProceedings of the Fifteenth ACM International Conference on Web Search and Data Mining
EditorsParth Nagarkar
Place of PublicationNew York NY USA
PublisherAssociation for Computing Machinery (ACM)
Pages1589-1592
Number of pages4
ISBN (Electronic)9781450391320
DOIs
Publication statusPublished - 2022
EventACM International Conference on Web Search and Data Mining 2022 - Online, United States of America
Duration: 21 Feb 202225 Feb 2022
Conference number: 15th

Conference

ConferenceACM International Conference on Web Search and Data Mining 2022
Abbreviated titleWSDM 2022
Country/TerritoryUnited States of America
CityOnline
Period21/02/2225/02/22

Keywords

  • Graph neural networks
  • Knowledge graph completion
  • Knowledge graph embeddings
  • Quaternion

Cite this