多任务联合训练让电网模型在新拓扑上表现更稳,避免过拟合。
MxGPS: Multiplex Graph Transformers for a Power Grid Foundation Model

- 用多分支图变换器联合训练状态估计与潮流计算,共享节点编码器。
- 零样本测试下四种新电网拓扑的边界违规率为0%,性能下降仅39%。
- 参数量少12倍,适合电力系统建模且对拓扑变化不敏感的场景。
针对电网图神经网络单任务微调存在的系统性失效问题——即模型在分布内误差最低时,拓扑变化下性能退化最严重,我们提出MxGPS(多路图变换器)。该模型通过共享节点编码器,运行K个任务专用的GPS分支,在静态状态估计(SSE)和交流潮流(PF)上进行自监督预训练与多任务微调,并引入跨分支注意力模块。联合目标迫使共享编码器同时满足互补梯度信号,避免对训练拓扑的结构过拟合。在覆盖四个未见拓扑(14、24、162、300节点)的三重滑动窗口交叉验证中,MxGPS在所有零样本潮流任务中实现0%边界违规率(BVR)。相比之下,内部误差更低的模型在拓扑变化下性能退化达190%至1400%,而MxGPS仅退化39%,直接证明拓扑过拟合是根本失败原因。模型仅需160万参数(比基准GridFM少12倍),表明多任务联合训练是实现电网基础模型拓扑无关泛化的有效且高效方法。
原文摘要 · Abstract (English)
Single-task fine-tuning of graph neural networks (GNNs) for power grid problems exhibits a systematic failure mode: models that achieve the lowest in-distribution error degrade the most under topology shift. We term this topology overfitting: the tendency of task-specific gradient signals to encode relational structure particular to the training topologies rather than the underlying physics, causing models to fail on unseen grids despite strong in-distribution performance. To expose and address this failure mode, we introduce MxGPS (Multiplex GPS), a multiplex graph transformer that runs K task-specialised GPS branches over a shared node encoder, jointly trained on Static State Estimation (SSE) and AC Power Flow (PF) via a self-supervised pre-training and multi-task fine-tuning protocol, with a cross-branch attention module evaluated in ablation. The joint SSE+PF objective forces the shared encoder to simultaneously satisfy complementary gradient signals, preventing it from overfitting to topology-specific relational structure. Under a 3-fold sliding-window cross-validation spanning four unseen topologies (14-, 24-, 162-, and 300-bus), MxGPS attains 0% boundary violation rate (BVR) on all four zero-shot Power Flow topologies. Critically, models with substantially lower in-distribution PF error degrade by 190% to 1400% under topology shift, whereas MxGPS degrades by only 39%, an inversion that directly implicates topology overfitting as the failure mechanism rather than insufficient model capacity. With only 1.6M parameters (12x fewer than the GridFM reference baseline), MxGPS demonstrates that multi-task joint training is a principled and parameter-efficient mechanism for topology-agnostic generalisation in power grid foundation models.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。