Scale-Adaptive Power Flow Analysis with Local Topology Slicing and Multi-Task Graph Learning

Yongzhe Li, Lin Guan, Zihan Cai, Zuxian Lin, Jiyu Huang +1 more
1/4/2026
cs.LGcs.AI

No confident code implementation found for this paper. 3 possible matches can be shown below.

Abstract

Developing deep learning models with strong adaptability to topological variations is of great practical significance for power flow analysis. To enhance model performance under variable system scales and improve robustness in branch power prediction, this paper proposes a Scale-adaptive Multi-task Power Flow Analysis (SaMPFA) framework. SaMPFA introduces a Local Topology Slicing (LTS) sampling technique that extracts subgraphs of different scales from the complete power network to strengthen the model's cross-scale learning capability. Furthermore, a Reference-free Multi-task Graph Learning (RMGL) model is designed for robust power flow prediction. Unlike existing approaches, RMGL predicts bus voltages and branch powers instead of phase angles. This design not only avoids the risk of error amplification in branch power calculation but also guides the model to learn the physical relationships of phase angle differences. In addition, the loss function incorporates extra terms that encourage the model to capture the physical patterns of angle differences and power transmission, further improving consistency between predictions and physical laws. Simulations on the IEEE 39-bus system and a real provincial grid in China demonstrate that the proposed model achieves superior adaptability and generalization under variable system scales, with accuracy improvements of 4.47% and 36.82%, respectively.

View on arXivView PDF

Code Implementations

No confident code match yet

We couldn't find an author-owned or strongly-evidenced community implementation for this paper. 3 weaker matches are hidden by default — verify before relying on them.

No code implementations found yet.

Know of an implementation? Let us know in the comments below!

Cite this paper

@article{li2026scaleadaptive,
  title  = {Scale-Adaptive Power Flow Analysis with Local Topology Slicing and Multi-Task Graph Learning},
  author = {Yongzhe Li and Lin Guan and Zihan Cai and Zuxian Lin and Jiyu Huang and Liukai Chen},
  year   = {2026},
  eprint = {2601.01387},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2601.01387v1}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2601.01387.svg)](https://codeofpaper.com/paper/2601.01387)

Discussion