Multi-Attractor GNNs: Set-Valued Expressivity Beyond Unique Equilibria
cs.LG, cs.AI
Submitted: 2026-09-28
Updated: 2026-09-28
License: http://creativecommons.org/licenses/by/4.0/
The gist: Recurrent and equilibrium graph neural networks (GNNs) often enforce a unique fixed point or use one training target per graph.
Terminology
Abstract
Recurrent and equilibrium graph neural networks (GNNs) often enforce a unique fixed point or use one training target per graph. Yet many combinatorial and scientific problems admit multiple valid solutions, with no preferred one. A designated target can then impose an arbitrary selection rule. For tasks invariant to node relabeling, a symmetric graph may have a symmetric solution set but no symmetric solution. We show that multiple equilibria enable one weight-tied message-passing GNN to represent set-valued equivariant maps: different initializations approach different valid solutions. Under stated regularity assumptions, we first construct globally Lipschitz, permutation-equivariant dynamics that converge almost surely to valid solutions and reach every solution branch with positive probability. We then establish approximate realization by recurrent message passing with continuous component maps, with arbitrarily small update and limiting errors and arbitrarily high probability. This goes beyond standard universality arguments: although message passing alone cannot distinguish symmetric nodes, the evolving state keeps nodes distinguishable at every finite step without auxiliary node identifiers. Such dynamics can be learned without solution labels using problem-specific energies. On Ising ground states, structural module detection in protein graphs, and chemical reaction steady states, the learned updates produce multiple high-quality predictions with high numerical convergence rates. They achieve better average solution quality than the tested unique-equilibrium, single-target, and feedforward baselines, while remaining competitive with much larger diffusion-based solvers.
Sources
- The Surprising Power of Graph Neural Networks with Random Node Initialization
- A simple yet effective baseline for non-attributed graph classification
- Expressiveness and Approximation Properties of Graph Neural Networks
- Any-Subgroup Equivariant Networks via Symmetry Breaking
- Universality and Approximation Rates of Graph Neural Networks with Random Features
- Equivariant Flow Matching for Symmetry-Breaking Bifurcation Problems
- Bifurcation Models: Learning Set-Valued Solution Maps with Weight-Tied Dynamics
- Symmetry Breaking and Equivariant Neural Networks
- Gated Graph Sequence Neural Networks
- What graph neural networks cannot learn: depth vs width
- TUDataset: A collection of benchmark datasets for learning with graphs
- Neural Learning of One-of-Many Solutions for Combinatorial Problems in Structured Output Spaces
- Convergent Graph Solvers
- Hopfield Networks is All You Need
- Learning a SAT Solver from Single-Bit Supervision
- Annealed Training for Combinatorial Optimization on Graphs
- Neural Execution of Graph Algorithms
- How Powerful are Graph Neural Networks?
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