Homophily modulates double descent generalization in graph convolution networks

Cheng Shi, Liming Pan, Hong Hu, Ivan Dokmanić

Research output: Contribution to journalArticlepeer-review

3 Scopus citations

Abstract

Graph neural networks (GNNs) excel in modeling relational data such as biological, social, and transportation networks, but the underpinnings of their success are not well understood. Traditional complexity measures from statistical learning theory fail to account for observed phenomena like the double descent or the impact of relational semantics on generalization error. Motivated by experimental observations of “transductive” double descent in key networks and datasets, we use analytical tools from statistical physics and random matrix theory to precisely characterize generalization in simple graph convolution networks on the contextual stochastic block model. Our results illuminate the nuances of learning on homophilic versus heterophilic data and predict double descent whose existence in GNNs has been questioned by recent work. We show how risk is shaped by the interplay between the graph noise, feature noise, and the number of training labels. Our findings apply beyond stylized models, capturing qualitative trends in real-world GNNs and datasets. As a case in point, we use our analytic insights to improve performance of state-of-the-art graph convolution networks on heterophilic datasets.

Original languageEnglish
Article numbere2309504121
JournalProceedings of the National Academy of Sciences of the United States of America
Volume121
Issue number8
DOIs
StatePublished - 2024

Keywords

  • double descent
  • graph neural network
  • homophily
  • statistical mechanics
  • stochastic block model

Fingerprint

Dive into the research topics of 'Homophily modulates double descent generalization in graph convolution networks'. Together they form a unique fingerprint.

Cite this