🤖 AI Summary
Voltage forecasting in distribution systems suffers from poor generalization under data scarcity, jeopardizing grid stability. To address this, we propose a physics-informed graph neural network framework that enhances out-of-distribution robustness in few-shot settings via three physically grounded inductive biases: (1) a physics-constrained loss derived from power flow equations; (2) complex-domain modeling to naturally represent voltage phasors; and (3) residual task reconstruction to enforce physical consistency. Ablation studies on the ENGAGE multi-scenario power grid dataset demonstrate that our method significantly improves prediction accuracy—reducing mean absolute error by 23.6% on average—and substantially enhances cross-configuration generalization. Moreover, the framework retains interpretability and computational efficiency. This work establishes a novel paradigm for AI-driven power system modeling under data-limited conditions.
📝 Abstract
Voltage prediction in distribution grids is a critical yet difficult task for maintaining power system stability. Machine learning approaches, particularly Graph Neural Networks (GNNs), offer significant speedups but suffer from poor generalization when trained on limited or incomplete data. In this work, we systematically investigate the role of inductive biases in improving a model's ability to reliably learn power flow. Specifically, we evaluate three physics-informed strategies: (i) power-flow-constrained loss functions, (ii) complex-valued neural networks, and (iii) residual-based task reformulation. Using the ENGAGE dataset, which spans multiple low- and medium-voltage grid configurations, we conduct controlled experiments to isolate the effect of each inductive bias and assess both standard predictive performance and out-of-distribution generalization. Our study provides practical insights into which model assumptions most effectively guide learning for reliable and efficient voltage prediction in modern distribution networks.