English

Learning Optimal Distributionally Robust Stochastic Control in Continuous State Spaces

Machine Learning 2025-11-18 v2 Machine Learning

Abstract

We study data-driven learning of robust stochastic control for infinite-horizon systems with potentially continuous state and action spaces. In many managerial settings--supply chains, finance, manufacturing, services, and dynamic games--the state-transition mechanism is determined by system design, while available data capture the distributional properties of the stochastic inputs from the environment. For modeling and computational tractability, a decision maker often adopts a Markov control model with i.i.d. environment inputs, which can render learned policies fragile to internal dependence or external perturbations. We introduce a distributionally robust stochastic control paradigm that promotes policy reliability by introducing adaptive adversarial perturbations to the environment input, while preserving the modeling, statistical, and computational tractability of the Markovian formulation. From a modeling perspective, we examine two adversary models--current-action-aware and current-action-unaware--leading to distinct dynamic behaviors and robust optimal policies. From a statistical learning perspective, we characterize optimal finite-sample minimax rates for uniform learning of the robust value function across a continuum of states under ambiguity sets defined by the fkf_k-divergence and Wasserstein distance. To efficiently compute the optimal robust policies, we further propose algorithms inspired by deep reinforcement learning methodologies. Finally, we demonstrate the applicability of the framework to real managerial problems.

Keywords

Cite

@article{arxiv.2406.11281,
  title  = {Learning Optimal Distributionally Robust Stochastic Control in Continuous State Spaces},
  author = {Shengbo Wang and Jason Meng and Nian Si and Jose Blanchet and Zhengyuan Zhou},
  journal= {arXiv preprint arXiv:2406.11281},
  year   = {2025}
}
R2 v1 2026-06-28T17:08:15.792Z