arXiv:2605.23194cs.LGcs.AI2026-05

构建可扩展的异构图模型,高效逼近智能电网最优潮流问题。

Scalable Heterogeneous Graph Foundation Models for Data-Driven Optimal Power Flow in Smart Grids

  • 保留电网中母线、发电机、线路等多类型节点与边结构
  • 在超算上训练出160万至170万参数的紧凑模型,误差最低
  • 预训练模型微调后,在小数据下仍能提升精度与收敛速度

快速可靠的最优潮流(OPF)近似对智能电网稳定运行至关重要。现有学习型代理模型常忽略电网固有的异构结构,或仅适用于有限拓扑,且缺乏可扩展的图基础模型(GFM)训练框架。本文提出基于HydraGNN的可扩展异构图神经网络工作流,用于数据驱动的OPF代理建模与OPF-GFM开发。该工作流保留电网中母线、发电机、负荷、并联电容、交流线路、变压器及设备-母线连接等多类型节点与边,并支持在超算上分布式预处理、训练、超参数优化(HPO)及下游微调。基于涵盖14至13,659个母线的十组PGLib-OPF案例,共三百万个异构图实例,我们在奥克兰前沿超算上采用DeepHyper驱动的HPO。实验识别出参数量约160万至170万的紧凑模型,验证损失最低。下游任务在可行性分类与N-1扰动回归中表明,使用预训练的OPF GFM进行微调,可提升低数据场景下的准确率,稳定训练过程,加速收敛,并降低适配成本,尤其在部分微调或仅头部微调时效果显著。

原文摘要 · Abstract (English)

Fast and reliable optimal power flow (OPF) approximation is essential for reliable smart-grid operation, yet many learning-based surrogates either flatten the native heterogeneous structure of power networks, target a limited set of grid topologies, or lack scalable infrastructure for graph foundation model (GFM) training. This paper presents a scalable heterogeneous graph neural network (GNN) workflow, built on HydraGNN, for data-driven OPF surrogate modeling and OPF-GFM development. The workflow preserves the distinct node and edge types of power grids -- buses, generators, loads, shunts, AC lines, transformers, and device-to-bus couplings -- and supports distributed preprocessing, training, hyperparameter optimization (HPO), and downstream fine-tuning on leadership-class supercomputers. Using three million heterogeneous graph instances spanning ten PGLib-OPF cases, from 14 to 13,659 buses, we conduct DeepHyper-driven HPO on the ORNL Frontier supercomputer. The campaign identifies compact models ($\sim$1.6--1.7M parameters) with the lowest validation losses. Downstream experiments on feasibility classification and N-1 contingency regression show that fine-tuning pretrained OPF GFM improves low-data accuracy, stabilizes training, accelerates convergence, and reduces adaptation cost when partial or head-only fine-tuning is used.

电力系统异构图超算图模型

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。