HydraGNN_OPF_GFM_2026 - Ensemble of predictive graph foundation models for power grid applications
-
Massimiliano Lupo Pasini | Oak Ridge National Laboratory
Description
This dataset supports research on graph foundation models for optimal power flow (OPF) on electric grids using HydraGNN. It contains heterogeneous graph representations of PGLib-OPF cases spanning systems from 14 to 13,659 buses, together with packed HDF5 datasets for pretraining, feasibility classification, and N-1 contingency analysis. The release includes OPF solution data, downstream fine-tuning datasets, pretrained HeteroSAGE and HeteroHEAT model checkpoints, hyperparameter-optimization summaries across multiple heterogeneous GNN architectures, and aggregated fine-tuning results for sample-efficiency studies. The dataset is designed to enable scalable training, evaluation, and transfer-learning studies for OPF surrogate modeling, including node-level AC-OPF solution prediction, graph-level prediction, feasibility classification, operating-condition generalization, and contingency-response tasks.
Funding Information
DOE Contract Number
AC05-00OR22725Originating Research Organization
Oak Ridge National Laboratory (ORNL), Oak Ridge, TN (United States)Sponsoring Organization
Office of Science (SC)Details
Release Date
July 14, 2026Subject
24 POWER TRANSMISSION AND DISTRIBUTIONKeywords
High-performance computing , Graph Neural Network, Graph Foundation ModelDataset
Dataset Type
ND Numeric DataCite This Dataset:
Lupo Pasini, M. (2026). HydraGNN_OPF_GFM_2026 - Ensemble of predictive graph foundation models for power grid applications. Oak Ridge National Laboratory. https://doi.org/10.13139/OLCF/3239722.
Acknowledgements
This research used resources of the Oak Ridge Leadership Computing Facility at the Oak Ridge National Laboratory, which is supported by the Advanced Scientific Computing Research programs in the Office of Science of the U.S. Department of Energy under Contract No. DE-AC05-00OR22725.