论文

GridSFM:用基础模型求解大规模交流最优潮流问题

GridSFM: A Foundation Model for Solving AC Optimal Power Flow

精选理由

电网调度方向的研究者可以看看,1500万参数的GNN预训练后零样本跑10000节点电网,误差2.45%,代码全开源。

GridSFM 是一个 1500 万参数的物理启发图神经网络,在 54 种电网拓扑、500 到 4000 节点的电网上预训练,用于求解 AC-OPF(交流最优潮流)。在 10000 节点的 held-out 运行工况上,零样本发电成本误差为 2.45%,且不随系统规模增大而退化。结合基于牛顿法的物理信息微调,只需 100 个已求解样本即可适配到最多 10000 节点的未见电网。作者还证明了对数罚松弛后的弹性可行集是可缩的,并给出将近似解投影回原始可行集的良定性。模型、数据和代码全部开源。

原文 · arXiv cs.LG

GridSFM: A Foundation Model for Solving AC Optimal Power Flow

We introduce GridSFM, a framework that combines a pretrained foundation model across grid topologies with physics-informed fine-tuning for solving AC Optimal Power Flow (AC-OPF) at scale. It is a $15$ million parameter physics-inspired graph neural network pretrained across $54$ topologies of $500$ to $4{,}000$ buses. Our model attains a $2.45\%$ zero-shot generation-cost error on a $10{,}000$ bus case held-out operating conditions with no degradation as system size grows. Building on this, we pair the pretrained backbone with a physics-informed fine-tuning design based on Newton's method for power flow. With only $100$ solved instances, GridSFM adapts to unseen grids up to $10{,}000$ buses. We show it out performs single topology, dedicated neural network models that are trained more data, both in terms of cost and solver iterations when deployed as warm starting points. In designing this foundation model, we overcome the fact that the feasible set for AC-OPF can be disconnected. This is an obstruction that prevents any continuous neural network from approximating the solution map. To do so, we lift the problem and relax its constraints with logarithmically penalized slacks. We prove that the resulting elastic feasible set is contractible, that the AC-OPF minimizers remain minimizers of the elastic problem above an explicit penalty threshold, and that projecting an approximate solution back onto the AC-OPF feasible set is well posed. We release all models, data, and code so that the community can build on a shared starting point for AC-OPF.