Diffusion-Based Scenario Tree Generation for Multivariate Time Series Prediction and Multistage Stochastic Optimization

2509.14832v1 cs.LG, cs.AI, cs.SY, eess.SY, I.2.6; I.5.1 2025-09-20
Авторы:

Stelios Zarifis, Ioannis Kordonis, Petros Maragos

Резюме на русском

## Контекст В оптимизации деcision-making в системах с неопределенностью, таких как энергетические рынки и финансовые рынки, ключевой роль играет статистическое прогнозирование. Наиболее важной задачей является построение сценариев будущих событий, которые могут стать условиями решения. Одним из главных ограничений является то, что прогнозы должны соблюдать принцип непредвиденности (non-anticipativity), что означает, что любое решение должно быть зависимым только от ранее наблюдавшегося данных. На сегодняшний день существуют методы, которые могут генерировать сценарии для похожих задач, однако они часто имеют ограниченную гибкость или сложность в реализации. Мы предлагаем Diffusion Scenario Tree (DST), многомерную модель для генерации сценариев, которая объединяет гибкость и точность в построении прогнозов при соблюдении непредвиденности. ## Метод DST использует идею diffusion-based probabilistic forecasting для построения сценариев. Основоположником является идея, что каждый сценарий представляется как последовательность событий, которые происходят в пространстве достаточно далекого будущего. Эти сценарии рекурсивно собираются в дерево с помощью метода clustering, позволяя выделять схожие траектории и создавать уникальные сценарии. Каждый узел дерева представляет собой наблюдаемое событие, которое может случиться в будущем, чтобы избежать противоречий с принципом непредвиденности. Мы также используем многомерные данные, чтобы учитывать сложности во взаимодействии нескольких переменных, что позволяет генерировать более точные сценарии. ## Результаты Мы проводили эксперименты на задаче энергетического арбитража в энергетических рынках Нью-Йорка. Мы сравнили DST с другими подходами, включая классические модели, такие как ARIMA, и модели Model-Free Reinforcement Learning. Результаты показали, что DST постоянно показывает более высокую точность прогноза и более эффективное управление неопределенностью. Мы также показали, что DST улучшает результаты задач стохастической оптимизации, таких как energy arbitrage, благодаря лучшему учету неопределенности и более точному представлении сценариев. ## Значимость Мы считаем, что наш подход может иметь широкое применение в различных областях, где необходимо учитывать неопределенность, такие как энергетические системы, финансовые рынки, транспортные системы. Наш алгоритм предлагает значительное улучшение по сравнению с существующими моделями, благодаря тому, что он может более точно предсказывать сложные сценарии в многомерных задачах. Мы видим потенциал DST в улучшении деcision-making, особенно в задачах стохастической оптимизации, где точность и эффективность ре

Abstract

Stochastic forecasting is critical for efficient decision-making in uncertain systems, such as energy markets and finance, where estimating the full distribution of future scenarios is essential. We propose Diffusion Scenario Tree (DST), a general framework for constructing scenario trees for multivariate prediction tasks using diffusion-based probabilistic forecasting models. DST recursively samples future trajectories and organizes them into a tree via clustering, ensuring non-anticipativity (decisions depending only on observed history) at each stage. We evaluate the framework on the optimization task of energy arbitrage in New York State's day-ahead electricity market. Experimental results show that our approach consistently outperforms the same optimization algorithms that use scenario trees from more conventional models and Model-Free Reinforcement Learning baselines. Furthermore, using DST for stochastic optimization yields more efficient decision policies, achieving higher performance by better handling uncertainty than deterministic and stochastic MPC variants using the same diffusion-based forecaster.

Ссылки и действия