OutageDiT: A Generative Foundation Model for Power Outage Forecasting and Scenario Simulation

Yunqin Zhu, Feng Qiu, Yao Xie
9/1/2026
cs.LGcs.AI

No confident code implementation found for this paper.

Abstract

Power-outage planning requires scenarios before an event occurs. These scenarios must represent uncertainty in magnitude, timing, and duration while preserving temporal dependence. However, severe events are rare, and data from any single region contain few examples of extreme outage and restoration patterns. To address this challenge, we introduce OutageDiT, a foundation model for generating seven-day outage trajectories at quarter-hour resolution, trained on outage and weather records across the United States. Specifically, a condition encoder processes the historical context and known future covariates once per forecast, and a shallow flow decoder reuses the resulting horizon-aligned states to generate complete trajectories. The resulting samples support point forecasting, uncertainty quantification, and conditional event simulation within one deep generative model. Across outage forecasting benchmarks, OutageDiT improves forecast accuracy and scenario quality over strong baselines and supports zero-shot transfer to held-out regions. Together, these results position conditional outage simulation as a bridge from outage forecasting to operational planning under uncertainty.

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. Any repos shown below are weak matches — verify before relying on them.

No code implementations found yet.

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

Similar Papers

Cite this paper

@article{zhu2026outagedit,
  title  = {OutageDiT: A Generative Foundation Model for Power Outage Forecasting and Scenario Simulation},
  author = {Yunqin Zhu and Feng Qiu and Yao Xie},
  year   = {2026},
  eprint = {2609.01896},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2609.01896v1}
}

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/2609.01896.svg)](https://codeofpaper.com/paper/2609.01896)

Discussion