Unified Heterogeneous Graph Neural Network solver for Power Flow, Optimal Power Flow and State Estimation
eess.SY, cs.LG, cs.SY
Submitted: 2026-09-15
Updated: 2026-09-15
Code: https://github.com/SanPen/VeraGrid
License: http://creativecommons.org/licenses/by-nc-nd/4.0/
The gist: Power Flow (PF), Optimal Power Flow (OPF), and State Estimation (SE) are fundamental problems in power system analysis, but solving them is computationally expensive.
Terminology
Abstract
Power Flow (PF), Optimal Power Flow (OPF), and State Estimation (SE) are fundamental problems in power system analysis, but solving them is computationally expensive. Graph Neural Networks (GNNs) have been proposed as fast surrogates, yet existing solvers are trained for a single problem at a time, producing narrow models that must be rebuilt for each new task. We propose a more general approach: a single Heterogeneous Residual Gated Graph Convolutional Network that solves all three problems with one shared backbone. Rather than learning one mapping, the model learns a reusable representation of how the network behaves, from which PF, OPF, and SE can each be estimated. Trained jointly on the three problems across diverse topologies and loading conditions, and evaluated on the IEEE 14-bus and 118-bus systems, the shared model matches the accuracy of task-specific GNN solvers and stays robust on unseen loading levels and topologies. These results show that a single model can capture the basic operation of a power network and serve several analysis tasks at once, a first step toward a foundation model for power systems.
Sources
- Learning to run a power network challenge for training topology controllers
- Neural Message Passing for Quantum Chemistry
- Relational inductive biases, deep learning, and graph networks
- Physics-Informed Graph Neural Networks for Robust AC-Optimal Power Flow
- How Powerful are Graph Neural Networks?
- Residual Gated Graph ConvNets
- Spectral Networks and Locally Connected Networks on Graphs
- Enhancing Power Flow Estimation with Topology-Aware Gated Graph Neural Networks
- Towards Generalization of Graph Neural Networks for AC Optimal Power Flow
Related papers
- One Request, Multiple Experts: LLM Orchestrates Domain Specific Models via Adaptive Task Routing
- A Geometric Decision Procedure for STL Feasibility and Repair
- Submodular Multi-Agent Policy Learning for Online Distributed Task Allocation in Open Multi-Agent Systems
- Policy-Level Recursive Self-Improvement for Embodied AI with a Criticality World Model
- Minimal Experiments for Robust Stabilization: Information, Spectral Geometry, and Duration
- Decentralized Power-Optimal Coordination for Spacecraft Swarms Using Time-Varying Magnetorquer Actuation