atlas

Research map / Graph learning / GNN architectures

Mixture of Weak and Strong Experts on Graphs

ICLR 2024 · Graph learning › GNN architectures · approach: GNN generalization theory · 0 citations (OpenAlex)

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 Mowst, a mixture of a weak MLP expert and a strong off-the-shelf GNN expert, where a confidence gate based on the MLP's prediction dispersion decides per node whether to use the MLP or fall back to the GNN.

Problem

GNNs apply global, uniform aggregation to all nodes even though nodes differ in whether self-features or neighborhood structure drive their labels; heterophilous neighbors can add noise, and MoE over GNNs is costly and hard to optimize.

Approach

Mix an MLP (weak expert) and a GNN (strong expert) at the output level, with a confidence C = G(D(p)) computed from the dispersion (variance or negative entropy) of only the MLP's logits; the MLP is used with probability C, else the GNN is run. Training alternates expert updates or jointly minimizes the expected loss, and the design is biased toward the GNN.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Add an MLP expert with confidence-based gating to GCNGCN · Flickr, Products, arXiv, Penn, Pokec, Twitchaccuracy: +0.76; +0.85; +0.64; +1.02; +0.29; +0.83Helped
Add an MLP expert with confidence-based gating to GINGIN · Flickr, arXiv, Penn, Pokec, Twitchaccuracy: +1.77; +2.04; +1.88; +22.74; +2.56Helped
Add an MLP expert with confidence-based gating to GIN-skipGIN-skip · Flickr, arXiv, Penn, Pokec, Twitchaccuracy: +0.49; +0.51; +0.88; +3.41; +0.64Helped
Add an MLP expert with confidence-based gating to GraphSAGEGraphSAGE · Flickr, Products, arXiv, Penn, Pokec, Twitchaccuracy: +0.39; +0.88; +0.55; +2.03; +1.33; +1.05Helped
Add an MLP expert with confidence-based gating to H2GCNH2GCN · Penn, Pokec, Twitchaccuracy: +0.68; +2.13; +0.33Helped
Train both experts simultaneously instead of using in-turn training (ablation)Mowst-GCN · Flickr, Pokec, Twitchaccuracy: 54.62 -> 53.47; 77.12 -> 76.62; 63.74 -> 63.44Hurt
Use the Mowst+ prediction-mixture loss instead of the Mowst loss (ablation)Mowst-GCN · Flickr, Pokec, Twitchaccuracy: 54.62 vs. 53.94; 77.12 vs. 77.28; 63.74 vs. 63.59Mixed
Replace the weak MLP expert with a shallow GCN expert (ablation)Mowst-GCN · Flickr, Products, arXiv, Penn, Pokec, Twitchaccuracy: 53.86 -> 54.19; 75.64 -> 74.47; 71.74 -> 71.92; 82.17 -> 82.38; 76.01 -> 76.51; 62.42 -> 63.36Mixed
Add skip connections to GCN layers (ablation)GCN · Flickr, arXiv, Penn, Pokec, Twitchaccuracy: 53.86 -> 52.98; 71.74 -> 69.56; 82.17 -> 76.58; 76.01 -> 73.46; 62.42 -> 61.05Hurt
Add skip connections to GIN layers (ablation)GIN · arXiv, Pokec, Twitchaccuracy: 69.39 -> 71.28; 53.37 -> 76.29; 61.76 -> 64.27Mixed
Add an MLP expert to GIN-skip (ablation)GIN-skip · arXiv, Pokec, Twitchaccuracy: 71.28 -> 71.79; 76.29 -> 79.70; 64.27 -> 64.91Mixed

Surprising findings

Limitations

Experimental setup

Datasets: Flickr ogbn-arxiv ogbn-products Penn94 Pokec Twitch

Benchmarks: Flickr ogbn-arxiv ogbn-products Penn94 Pokec Twitch

Metrics: accuracy

Baselines: MLP GCN GraphSAGE GAT GIN GPR-GNN H2GCN AdaGCN GraphMoE GCN-skip GIN-skip

Builds on: Mixture of Experts GraphMoE GPR-GNN H2GCN GLNN

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