Improving the Effective Receptive Field of Message-Passing Neural Networks

Research output: Contribution to journalConference articlepeer-review

Abstract

Message-Passing Neural Networks (MPNNs) have become a cornerstone for processing and an-alyzing graph-structured data. However, their ef-fectiveness is often hindered by phenomena such as over-squashing, where long-range dependen-cies or interactions are inadequately captured and expressed in the MPNN output. This limitation mirrors the challenges of the Effective Receptive Field (ERF) in Convolutional Neural Networks (CNNs), where the theoretical receptive field is underutilized in practice. In this work, we show and theoretically explain the limited ERF prob-lem in MPNNs. Furthermore, inspired by re-cent advances in ERF augmentation for CNNs, we propose an Interleaved Multiscale Message-Passing Neural Networks (IM-MPNN) architec-ture to address these problems in MPNNs. Our method incorporates a hierarchical coarsening of the graph, enabling message-passing across multiscale representations and facilitating long-range interactions without excessive depth or pa-rameterization. Through extensive evaluations on benchmarks such as the Long-Range Graph Benchmark (LRGB), we demonstrate substantial improvements over baseline MPNNs in captur-ing long-range dependencies while maintaining computational efficiency.

Original languageEnglish
Pages (from-to)17203-17220
Number of pages18
JournalProceedings of Machine Learning Research
Volume267
StatePublished - 1 Jan 2025
Event42nd International Conference on Machine Learning, ICML 2025 - Vancouver, Canada
Duration: 13 Jul 202519 Jul 2025

ASJC Scopus subject areas

  • Software
  • Control and Systems Engineering
  • Statistics and Probability
  • Artificial Intelligence

Fingerprint

Dive into the research topics of 'Improving the Effective Receptive Field of Message-Passing Neural Networks'. Together they form a unique fingerprint.

Cite this