Composition-based Multi-Relational Graph Convolutional Networks

TL;DR

CompGCN is a new graph convolutional framework that jointly embeds nodes and relations in multi-relational graphs, improving performance on various tasks and generalizing existing methods.

Contribution

It introduces a novel composition-based approach for embedding both nodes and relations, addressing over-parameterization and scalability issues in multi-relational GCNs.

Findings

Achieves superior results on node classification, link prediction, and graph classification.

Scales effectively with the number of relations.

Generalizes several existing multi-relational GCN methods.

Abstract

Graph Convolutional Networks (GCNs) have recently been shown to be quite successful in modeling graph-structured data. However, the primary focus has been on handling simple undirected graphs. Multi-relational graphs are a more general and prevalent form of graphs where each edge has a label and direction associated with it. Most of the existing approaches to handle such graphs suffer from over-parameterization and are restricted to learning representations of nodes only. In this paper, we propose CompGCN, a novel Graph Convolutional framework which jointly embeds both nodes and relations in a relational graph. CompGCN leverages a variety of entity-relation composition operations from Knowledge Graph Embedding techniques and scales with the number of relations. It also generalizes several of the existing multi-relational GCN methods. We evaluate our proposed method on multiple tasks such as node classification, link prediction, and graph classification, and achieve demonstrably superior results. We make the source code of CompGCN available to foster reproducible research.