A Classifier-Guided Diffusion Model-Based Key Sample Augmentation Method for Power System Transient Stability

Yangjin Wu, Junhao Zhao, Xiaodong Shen (), Shixiong Fan, Shicong Ma and Junyong Liu
Additional contact information
Yangjin Wu: College of Electrical Engineering, Sichuan University, Chengdu 610065, China
Junhao Zhao: College of Electrical Engineering, Sichuan University, Chengdu 610065, China
Xiaodong Shen: College of Electrical Engineering, Sichuan University, Chengdu 610065, China
Shixiong Fan: China Electric Power Research Institute, Beijing 100192, China
Shicong Ma: China Electric Power Research Institute, Beijing 100192, China
Junyong Liu: College of Electrical Engineering, Sichuan University, Chengdu 610065, China

Energies, 2025, vol. 18, issue 18, 1-22

Abstract: Modern power systems are increasingly complex, and the risk of transient instability is rising accordingly. Data-driven transient stability assessment (TSA) is attractive for its efficiency, yet in practice the number of unstable events is much smaller than that of stable ones, leading to severe class imbalance and degraded accuracy. This paper proposes a SHAP-guided, classifier-controlled diffusion augmentation framework to mitigate imbalance and enhance TSA. First, SHAP analysis identifies critical unstable and near-boundary samples, ensuring that augmentation targets the most informative regions of the state space. Then, a classifier-guided conditional diffusion model—with a Transformer-based denoising network—generates class-faithful synthetic trajectories that capture long-range temporal dependencies and inter-variable couplings. Case studies on the IEEE 10-machine 39-bus system show that the proposed method consistently surpasses traditional over-sampling (e.g., SMOTE/ADASYN) and deep generative baselines (e.g., CGAN/TimeGAN) in terms of accuracy, precision, recall, and F 1-score. Moreover, the approach maintains strong performance under small-sample settings and shortened time-series inputs, demonstrating favorable adaptability and robustness. These results indicate that the proposed augmentation framework offers a practical and effective solution for TSA under severe class imbalance.

Keywords: transient stability assessment; power system; sample imbalance; data augmentation; diffusion models; SHAP; classifier guidance (search for similar items in EconPapers)
JEL-codes: Q Q0 Q4 Q40 Q41 Q42 Q43 Q47 Q48 Q49 (search for similar items in EconPapers)
Date: 2025
References: Add references at CitEc
Citations:

Downloads: (external link)
https://www.mdpi.com/1996-1073/18/18/4848/pdf (application/pdf)
https://www.mdpi.com/1996-1073/18/18/4848/ (text/html)

Related works:
This item may be available elsewhere in EconPapers: Search for items with the same title.

Export reference: BibTeX RIS (EndNote, ProCite, RefMan) HTML/Text

Persistent link: https://EconPapers.repec.org/RePEc:gam:jeners:v:18:y:2025:i:18:p:4848-:d:1747668

Access Statistics for this article

Energies is currently edited by Ms. Cassie Shen

More articles in Energies from MDPI
Bibliographic data for series maintained by MDPI Indexing Manager ().