Diffusion-Based Forecasting for Uncertainty-Aware Model Predictive Control

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Zarifis, Stelios, Kordonis, Ioannis, Maragos, Petros
Format: Preprint
Published: 2025
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866915204822990848
author Zarifis, Stelios
Kordonis, Ioannis
Maragos, Petros
author_facet Zarifis, Stelios
Kordonis, Ioannis
Maragos, Petros
contents We propose Diffusion-Informed Model Predictive Control (D-I MPC), a generic framework for uncertainty-aware prediction and decision-making in partially observable stochastic systems by integrating diffusion-based time series forecasting models in Model Predictive Control algorithms. In our approach, a diffusion-based time series forecasting model is used to probabilistically estimate the evolution of the system's stochastic components. These forecasts are then incorporated into MPC algorithms to estimate future trajectories and optimize action selection under the uncertainty of the future. We evaluate the framework on the task of energy arbitrage, where a Battery Energy Storage System participates in the day-ahead electricity market of the New York state. Experimental results indicate that our model-based approach with a diffusion-based forecaster significantly outperforms both implementations with classical forecasting methods and model-free reinforcement learning baselines.
format Preprint
id arxiv_https___arxiv_org_abs_2503_15095
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Diffusion-Based Forecasting for Uncertainty-Aware Model Predictive Control
Zarifis, Stelios
Kordonis, Ioannis
Maragos, Petros
Machine Learning
Artificial Intelligence
Systems and Control
I.2.6; I.5.1
We propose Diffusion-Informed Model Predictive Control (D-I MPC), a generic framework for uncertainty-aware prediction and decision-making in partially observable stochastic systems by integrating diffusion-based time series forecasting models in Model Predictive Control algorithms. In our approach, a diffusion-based time series forecasting model is used to probabilistically estimate the evolution of the system's stochastic components. These forecasts are then incorporated into MPC algorithms to estimate future trajectories and optimize action selection under the uncertainty of the future. We evaluate the framework on the task of energy arbitrage, where a Battery Energy Storage System participates in the day-ahead electricity market of the New York state. Experimental results indicate that our model-based approach with a diffusion-based forecaster significantly outperforms both implementations with classical forecasting methods and model-free reinforcement learning baselines.
title Diffusion-Based Forecasting for Uncertainty-Aware Model Predictive Control
topic Machine Learning
Artificial Intelligence
Systems and Control
I.2.6; I.5.1
url https://arxiv.org/abs/2503.15095