Skip to main content

HydraGNN_OPF_GFM_2026 - Ensemble of predictive graph foundation models for power grid applications

    Massimiliano Lupo Pasini | Oak Ridge National Laboratory
Download Dataset on Globus

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-00OR22725

Originating Research Organization

Oak Ridge National Laboratory (ORNL), Oak Ridge, TN (United States)

Sponsoring Organization

Office of Science (SC)

Details

Release Date

July 14, 2026

Subject

24 POWER TRANSMISSION AND DISTRIBUTION

Keywords

High-performance computing , Graph Neural Network, Graph Foundation Model

Dataset

Dataset Type

ND Numeric Data

Cite 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.