atlas

Research map / Graph learning / GNN architectures

Multi-Track Message Passing: Tackling Oversmoothing and Oversquashing in Graph Learning via Preventing Heterophily Mixing

ICML 2024 · Graph learning › GNN architectures · approach: Expressive equivariant message passing

Read the paperCheck your idea against 108,967 papers

AI-extracted summary. Written by AI from the paper's full text. Numbers and claims can be wrong or missing context, so check the paper before you cite or build on it.

Proposes MTGCN, a multi-track GCN whose message passing propagates node features separately in category-specific tracks to prevent heterophily mixing, tackling oversmoothing and oversquashing.

Problem

Deep GNNs suffer from oversmoothing and oversquashing, which the authors attribute to information loss from mixing messages of different category semantics (heterophily mixing) during aggregation.

Approach

Separate messages into multiple isomorphic graph tracks, one per category semantic, with soft node-track affiliations computed by dot-product attention against track prototypes; messages propagate independently within each track, with initial residuals and no per-layer weights or activations, and a multi-stage self-training pipeline refines affiliations and pseudo-labels.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Use multi-track message passing with stage 3MTGCN · Cora (semi-supervised)accuracy: 85.30 -> 86.40Helped
Use multi-track message passing with stage 3MTGCN · Citeseer (semi-supervised)accuracy: 75.60 -> 74.60Hurt
Use multi-track message passing with stage 2MTGCN · Pubmed (semi-supervised)accuracy: 80.60 -> 81.10Helped
Use multi-track message passing with stage 4MTGCN · Coauthor CS (semi-supervised)accuracy: 90.01 -> 92.54Helped
Use multi-track message passing with stage 4MTGCN · Coauthor Physics (semi-supervised)accuracy: 94.55 -> 94.72Helped
Use multi-track message passing with stage 1MTGCN · Cora (fully supervised)accuracy: 88.49 -> 90.61Helped
Use multi-track message passing with stage 3MTGCN · Citeseer (fully supervised)accuracy: 77.99 -> 77.36Hurt
Use multi-track message passing with stage 1MTGCN · Pubmed (fully supervised)accuracy: 90.30 -> 88.43Hurt
Use multi-track message passing with stage 4MTGCN · Cornell (fully supervised)accuracy: 85.68 -> 89.47Helped
Use multi-track message passing with stage 2MTGCN · Texas (fully supervised)accuracy: 88.38 -> 92.10Helped
Use multi-track message passing with stage 1MTGCN · Wisconsin (fully supervised)accuracy: 89.41 -> 90.20Helped
Increase message-passing depth from 2 to 64 layers (ablation)MTGCN · Cora (semi-supervised)accuracy: 80.5 -> 86.4Helped

Surprising findings

Experimental setup

Datasets: Cora Citeseer Pubmed Coauthor CS Coauthor Physics Cornell Texas Wisconsin Tree-NeighborsMatch

Benchmarks: Cora Citeseer Pubmed Coauthor CS Coauthor Physics Cornell Texas Wisconsin Tree-NeighborsMatch

Metrics: classification accuracy group distance ratio Rg training accuracy node-track affiliation accuracy

Baselines: GCN GAT Self-Train DisenGCN GCNII EGNN GRAND++ PDE-GCN GraphCON ACMP GREAD H2GCN

Builds on: GCN GCNII SGC DisenGCN

Closest papers by meaning

Writing a paper in this area? scime draft is a free LaTeX editor with AI included, real-time co-authors and a check of your references and venue rules before you submit. It also imports OpenAI Prism projects.

Open scime draft

Get the next scime Dispatch and a monthly field update

Free. We send a confirmation link first; every email has a one-click unsubscribe. Privacy