arXiv cs.AI· Yunqin Zhu, Feng Qiu, Yao Xie·· 14 小时前AI 评分29
OutageDiT:面向停电预测与场景生成的基础模型
OutageDiT: A Foundation Model for Power Outage Forecasting and Scenario Generation
AI 导读
OutageDiT 是一个基于全美停电与天气记录训练的基础模型,可生成 15 分钟分辨率、为期七天的停电轨迹。它用条件编码器处理历史上下文与已知未来协变量,再由浅层流匹配解码器复用对齐状态采样完整轨迹,在单一深度生成模型中同时实现点预测、不确定性量化和条件场景生成。在停电预测基准上,其预测精度与场景质量均优于强基线,并支持向未见过区域的零样本迁移。
正文 · 原文
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 15-minute resolution, trained on outage and weather records across the United States. Specifically, a condition encoder processes historical context and known future covariates once per forecast, and a shallow flow-matching decoder reuses the resulting horizon-aligned states to sample complete outage trajectories. This enables point forecasting, uncertainty quantification, and conditional scenario generation 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 scenario generation as a bridge from outage forecasting to operational planning under uncertainty.
| Comments: | 11 pages, 3 figures, 3 tables. Accepted at the NeurIPS 2026 Second Workshop on MLxOR |
| Subjects: | Machine Learning (cs.LG); Artificial Intelligence (cs.AI) |
| Cite as: | arXiv:2609.01896 [cs.LG] |
| (or arXiv:2609.01896v2 [cs.LG] for this version) | |
| https://doi.org/10.48550/arXiv.2609.01896 arXiv-issued DOI via DataCite |
Submission history
From: Yunqin Zhu [view email]
[v1]
Tue, 1 Sep 2026 21:54:17 UTC (172 KB)
[v2]
Tue, 27 Oct 2026 21:59:08 UTC (357 KB)
来源:arXiv cs.AI · arxiv.org