Distributionally Robust Control with End-to-End Statistically Guaranteed Metric Learning
Wu, Ning, Shi
Wasserstein distributionally robust control (DRC) recently emerges as a principled paradigm for handling uncertainty in stochastic dynamical systems. However, it constructs data-driven ambiguity sets via uniform distribution shifts before sequentially incorporating them into downstream control synthesis. This segregation between ambiguity set construction and control objectives inherently introduces a structural misalignment, which undesirably leads to conservative control policies with sub-optimal performance. To address this limitation, we propose a novel end-to-end finite-horizon Wasserstein DRC framework that integrates the learning of anisotropic Wasserstein metrics with downstream control tasks in a closed-loop manner, thus enabling ambiguity sets to be systematically adjusted along performance-critical directions and yielding more effective control policies. This framework is formulated as a bilevel program: the inner level characterizes dynamical system evolution under DRC, while the outer level refines the anisotropic metric leveraging control-performance feedback across a range of initial conditions. To solve this program efficiently, we develop a stochastic augmented Lagrangian algorithm tailored to the bilevel structure. Theoretically, we prove that the learned ambiguity sets preserve statistical finite-sample guarantees under a novel radius adjustment mechanism, and we establish the well-posedness of the bilevel formulation by demonstrating its continuity with respect to the learnable metric. Furthermore, we show that the algorithm converges to stationary points of the outer level problem, which are statistically consistent with the optimal metric at a non-asymptotic convergence rate. Experiments on both numerical and inventory control tasks verify that the proposed framework achieves superior closed-loop performance and robustness compared against state-of-the-art methods.
academic
Распределительно-робастное управление с гарантированным статистически метрическим обучением "конец-в-конец"
Распределительно-робастное управление (DRC) на основе расстояния Вассерштейна в последнее время привлекает значительное внимание как принципиальная парадигма для обработки неопределённости в стохастических динамических системах. Однако существующие методы строят управляемые множества на основе данных посредством равномерного смещения распределения, а затем последовательно интегрируют их в синтез управления. Такое разделение между построением управляемого множества и целями управления по своей природе вводит структурное рассогласование, приводящее к консервативным стратегиям управления и субоптимальной производительности. Для решения этого ограничения в статье предлагается новая структура распределительно-робастного управления Вассерштейна на конечном горизонте "конец-в-конец", которая интегрирует обучение анизотропной метрике Вассерштейна с задачей управления в замкнутом цикле, позволяя управляемому множеству систематически адаптироваться вдоль критических по производительности направлений, обеспечивая более эффективные стратегии управления.
Стохастическое управление широко применяется в робототехнике, энергетических системах и финансах для принятия решений в условиях неопределённости. Классическая теория предполагает полное знание вероятностного распределения неопределённости управления, однако на практике такое распределение редко доступно, и контроллер должен быть разработан на основе приблизительной информации, выведенной из ограниченных данных. Эта неполнота приближения приводит к расхождению между оценённым и истинным распределениями, что значительно снижает производительность управления.
Консервативность традиционного DRC Вассерштейна: Существующие методы используют изотропные шары Вассерштейна, рассматривая смещения распределения во всех направлениях как одинаково важные, игнорируя их неоднородное влияние на производительность управления
Структурное рассогласование последовательной обработки: Разделение между построением управляемого множества и синтезом управления приводит к проектированию управляемого множества, независимому от задачи, что порождает чрезмерно консервативные стратегии управления
Ограничения обобщения в управлении "конец-в-конец": Существующие методы управления "конец-в-конец" обычно обучаются на одном начальном условии, что приводит к переобучению и ограничивает практическую применимость
Данная работа направлена на преодоление разделения между проектированием управляемого множества и производительностью управления путём интеграции DRC на основе Вассерштейна с обучением "конец-в-конец", адаптируя метрику Вассерштейна на основе обратной связи производительности управления при сохранении статистических гарантий на конечной выборке.
Первая структура DRC Вассерштейна "конец-в-конец": Предложена первая структура, которая посредством двухуровневой оптимизации связывает проектирование управляемого множества и управление в режиме обратной связи с обобщением на разнообразные начальные условия
Анизотропное управляемое множество Вассерштейна, ориентированное на задачу управления: Предложен механизм адаптации радиуса с теоретически установленными статистическими гарантиями на конечной выборке
Рассмотрим линейную систему с аддитивным возмущением:
xt+1=Axt+But+wt
где xt∈Rnx, ut∈Rnu, wt∈Rnx — состояние системы, управляющее воздействие и неопределённое возмущение соответственно. Система подчинена ограничениям:
FxTxt+FuTut+f≤0
Теорема 1: При предположении о лёгких хвостах анизотропное управляемое множество Вассерштейна удовлетворяет тем же гарантиям на конечной выборке:
PN{P∈Bε(Λ)Λ(P^N)}≥{1−c1exp(−c2Nεmax{m,2}),1−c1exp(−c2Nεa),ε≤1ε>1
Данная работа открывает новое направление в области распределительно-робастного управления и окажет значительное влияние на кросс-дисциплинарные исследования в теории управления и машинном обучении. Идея обучения "конец-в-конец" может быть обобщена на другие задачи робастного управления.
Статья ссылается на 45 соответствующих работ, охватывающих распределительно-робастную оптимизацию, предсказательное управление моделью, обучение с подкреплением и другие важные области, обеспечивая прочную теоретическую базу для исследования.