DeeperGCN: Training Deeper GCNs with Generalized Aggregation Functions

Guohao Li, Chenxin Xiong, Guocheng Qian, Ali Kassem Thabet, Bernard Ghanem

Research output: Contribution to journalArticlepeer-review

Abstract

Graph Neural Networks (GNNs) have been drawing significant attention to representation learning on graphs. Recent works developed frameworks to train very deep GNNs and showed impressive results in tasks like point cloud learning and protein interaction prediction. In this work, we study the performance of such deep models in large-scale graphs. In particular, we look at the effect of adequately choosing an aggregation function on deep models. We find that GNNs are very sensitive to the choice of aggregation functions (e.g. mean, max, and sum) when applied to different datasets. We systematically study and propose to alleviate this issue by introducing a novel class of aggregation functions named Generalized Aggregation Functions. The proposed functions extend beyond commonly used aggregation functions to a wide range of new permutation-invariant functions. Generalized Aggregation Functions are fully differentiable, where their parameters can be learned in an end-to-end fashion to yield a suitable aggregation function for each task. We show that equipped with the proposed aggregation functions, deep residual GNNs outperform state-of-the-art in several benchmarks from Open Graph Benchmark (OGB) across tasks and domains.
Original languageEnglish (US)
Pages (from-to)1-12
Number of pages12
JournalIEEE Transactions on Pattern Analysis and Machine Intelligence
DOIs
StatePublished - Aug 21 2023

ASJC Scopus subject areas

  • Artificial Intelligence
  • Computational Theory and Mathematics
  • Software
  • Applied Mathematics
  • Computer Vision and Pattern Recognition

Fingerprint

Dive into the research topics of 'DeeperGCN: Training Deeper GCNs with Generalized Aggregation Functions'. Together they form a unique fingerprint.

Cite this