PowerModelsGAT-AI: Physics-Informed Graph Attention for Multi-System Power Flow with Continual Learning
Not on the map. Why?
Abstract (arXiv)
Solving the alternating current power flow equations in real time is essential for secure grid operation, yet classical Newton-Raphson solvers can be slow under stressed conditions. Existing graph neural networks for power flow are typically trained on a single system and often degrade on different systems. We present PowerModelsGAT-AI, a physics-informed graph attention network that predicts bus voltages and generator injections. The model uses bus-type-aware masking to handle different bus types and balances multiple loss terms, including a power-mismatch penalty, using learned weights. We evaluate the model on 14 benchmark systems (4 to 6,470 buses) and train a unified model on 13 of these under N-2 (two-branch outage) conditions, achieving an average normalized mean absolute error of 0.89% for voltage magnitudes and R^2 > 0.99 for voltage angles. We also show continual learning: when adapting a base model to a new 1,354-bus system, standard fine-tuning causes severe forgetting with error increases exceeding 1000% on base systems, while our experience replay and elastic weight consolidation strategy keeps error increases below 2% and in some cases improves base-system performance. Interpretability analysis shows that learned attention weights correlate with physical branch parameters (susceptance: r = 0.38; thermal limits: r = 0.22), and feature importance analysis supports that the model captures established power flow relationships.
Key facts. No citing papers in the map yet (data through 7 Oct 2026). Builds on 27 of its 63 references that are in the map. First posted on arXiv 24 Feb 2026.
Builds on
27 of 63 references.
References · 27
- 2019PyTorch: An Imperative Style, High-Performance Deep Learning Librarycited by 20,160
- 2016SGDR: Stochastic Gradient Descent with Warm Restartscited by 4,993
- 2016Overcoming catastrophic forgetting in neural networkscited by 4,718
- 2017Axiomatic Attribution for Deep Networkscited by 3,350
- 2019A Comprehensive Survey on Graph Neural Networkscited by 2,906
- 2017Multi-Task Learning Using Uncertainty to Weigh Losses for Scene Geometry and Semanticscited by 1,345
- 2019Attention is not Explanationcited by 916
- 2021How Attentive are Graph Attention Networks?cited by 818
- 2019GNNExplainer: Generating Explanations for Graph Neural Networkscited by 735
- 2019Attention is not not Explanationcited by 629
- 2020GraphNorm: A Principled Approach to Accelerating Graph Neural Network Trainingcited by 116
- 2017pandapower - an Open Source Python Tool for Convenient Modeling, Analysis and Optimization of Electric Power Systemscited by 79
15 more
- 2021A Review of Graph Neural Networks and Their Applications in Power Systemscited by 62
- 2025PyG 2.0: Scalable Learning on Real World Graphscited by 49
- 2023PowerFlowNet: Power Flow Approximation Using Message Passing Graph Neural Networkscited by 19
- 2024PowerGraph: A power grid benchmark dataset for graph neural networkscited by 19
- 2016AC Power Flow Data in MATPOWER and QCQP Format: iTesla, RTE Snapshots, and PEGASEcited by 15
- 2022Unsupervised Optimal Power Flow Using Graph Neural Networkscited by 13
- 2022Solving AC Power Flow with Graph Neural Networks under Realistic Constraintscited by 11
- 2018A Survey on State Estimation Techniques and Challenges in Smart Distribution Systemscited by 10
- 2021Neural Network-based Power Flow Modelcited by 10
- 2025Towards Generalization of Graph Neural Networks for AC Optimal Power Flowcited by 5
- 2023Graph Neural Networks for Fast Contingency Analysis of Power Systemscited by 2
- 2024Improved Physics-Informed Neural Network based AC Power Flow for Distribution Networkscited by 2
- 2024Beyond the Neural Fog: Interpretable Learning for AC Optimal Power Flowcited by 1
- 2025Graph Neural Networks for Efficient AC Power Flow Prediction in Power Gridscited by 1
- 2025A Hybrid GNN-LSE Method for Fast, Robust, and Physically-Consistent AC Power Flowcited by 1
Built on by
Not cited yet.