Message Passing Does More with Less for In-Context Learning on Graphs
cs.LG, cs.SI
Submitted: 2026-09-29
Updated: 2026-09-29
Code: https://github.com/nums-ai/ephris
License: http://creativecommons.org/licenses/by-nc-sa/4.0/
The gist: Achieving strong performance with graph neural networks (GNNs) typically requires training and hyperparameter tuning for each dataset, incurring repeated costs and effort.
Terminology
Abstract
Achieving strong performance with graph neural networks (GNNs) typically requires training and hyperparameter tuning for each dataset, incurring repeated costs and effort. Graph in-context learning (ICL) avoids this by using a single pretrained model to predict unknown node labels directly from labeled context nodes. Existing approaches, however, rely on dense attention across nodes, making inference increasingly expensive as graphs grow. In this work, we present Ephris, a new graph in-context learner built on sparse message passing, scaling linearly with the number of node-feature entries and graph edges. Ephris is pretrained entirely on synthetic graphs generated from structural causal models with diverse graph structures and relational dynamics, exposing the model to varied dependencies among topology, features, and labels. We evaluate Ephris on 51 node-classification datasets against 15 extensively tuned GNNs and existing graph ICL methods under both high- and low-label train/validation/test splits. Across both settings, Ephris ranks first on all four aggregate measures: Elo, improvability, average rank, and accuracy. Its inference cost remains comparable to training a single GNN once, while being over 10 times faster than previous graph ICL models. Together, these results advance the performance-runtime Pareto frontier, demonstrating that strong graph ICL does not require dense attention. Code and model weights are available at https://github.com/nums-ai/ephris.
Sources
- Deep Gaussian Embedding of Graphs: Unsupervised Inductive Learning via Ranking
- How Attentive are Graph Attention Networks?
- NAGphormer: A Tokenized Graph Transformer for Node Classification in Large Graphs
- Adaptive Universal Generalized PageRank Graph Neural Network
- You are AllSet: A Multiset Function Framework for Hypergraph Neural Networks
- Causilo Technical Report
- Turning Tabular Foundation Models into Graph Foundation Models
- Predict then Propagate: Graph Neural Networks meet Personalized PageRank
- TabPFN-3: Technical Report
- Tabular foundation models for in-context prediction of molecular properties
- TabPFN-3.5: Technical Report
- Decoupled Weight Decay Regularization
- Wiki-CS: A Wikipedia-Based Benchmark for Graph Neural Networks
- Geom-GCN: Geometric Graph Convolutional Networks
- A critical look at the evaluation of GNNs under heterophily: Are we really making progress?
- Pitfalls of Graph Neural Network Evaluation
- Masked Label Prediction: Unified Message Passing Model for Semi-Supervised Classification
- Towards Graph Foundation Models: Learning Generalities Across Graphs via Task-Trees
- Anti-Money Laundering in Bitcoin: Experimenting with Graph Convolutional Networks for Financial Forensics
- GraphSAINT: Graph Sampling Based Inductive Learning Method
Related papers
- Polynomial-Augmented Neural Networks (PANNs) with Weak Orthogonality Constraints for Enhanced Function and PDE Approximation
- AIRL-S: Unifying Reinforcement Learning and Search-Based Test-Time Scaling via Adversarial Inverse Reinforcement Learning
- Transformers as Bayesian In-Context Experimenters: Smoothness-Adaptive Efficient ATE Estimation
- Convergence issues in Relational Concept Analysis based on AOC-posets
- Beliefs Beyond Posteriors: The Predictive Cost of ELBO Binding
- Understanding Diffusion Models via Ratio-Based Function Approximation with SignReLU Networks