Primal-Dual Policy Optimization for Linear CMDPs with Adversarial Losses
Kihyun Yu, Seoungbin Bae, Dabeen Lee
Abstract
Existing work on linear constrained Markov decision processes (CMDPs) has primarily focused on stochastic settings, where the losses and costs are either fixed or drawn from fixed distributions. However, such formulations are inherently vulnerable to adversarially changing environments. To overcome this limitation, we propose a primal-dual policy optimization algorithm for online finite-horizon adversarial linear CMDPs, where the losses are adversarially chosen under full-information feedback and the costs are stochastic under bandit feedback. Our algorithm is the first to achieve sublinear regret and constraint violation bounds in this setting, both bounded by , where denotes the number of episodes. The algorithm introduces and runs with a new class of policies, which we call weighted LogSumExp softmax policies, designed to adapt to adversarially chosen loss functions. Our main result stems from the following key contributions: (i) a new covering number argument for the weighted LogSumExp softmax policies, and (ii) two novel algorithmic components -- periodic policy mixing and a regularized dual update -- which allow us to effectively control both the covering number and the dual variable. We also report numerical results that validate our theoretical findings on the performance of the algorithm.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Builds on30
- Learning Policies with Zero or Bounded Constraint Violation for Constrained MDPsTao Liu, Ruida Zhou, Dileep Kalathil, Panganamala R. Kumar et al.NeurIPS 2021 · 110 citations
- Optimistic Policy Optimization with Bandit FeedbackLior Shani, Yonathan Efroni, Aviv Rosenberg, Shie MannorICML 2020 · 100 citations
- Upper Confidence Primal-Dual Reinforcement Learning for CMDP with Adversarial LossShuang Qiu, Xiaohan Wei, Zhuoran Yang, Jieping Ye et al.NeurIPS 2020 · 65 citations
- Near-optimal Regret Bounds for Stochastic Shortest PathAviv Rosenberg, Alon Cohen, Yishay Mansour, Haim KaplanICML 2020 · 63 citations
- Policy Optimization in Adversarial MDPs: Improved Exploration via Dilated BonusesHaipeng Luo, Chen-Yu Wei, Chung-Wei LeeNeurIPS 2021 · 59 citations
Related papers
- Learning Adversarial MDPs with Stochastic Hard ConstraintsFrancesco Emanuele Stradi, Matteo Castiglioni, Alberto Marchesi, Nicola GattiICML 2025
- Policy Optimization for CMDPs with Bandit Feedback: Learning Stochastic and Adversarial ConstraintsFrancesco Emanuele Stradi, Anna Lunghi, Matteo Castiglioni, Alberto Marchesi et al.ICML 2025
- Optimal Strong Regret and Violation in Constrained MDPs via Policy OptimizationFrancesco Emanuele Stradi, Matteo Castiglioni, Alberto Marchesi, Nicola GattiICLR 2025
- Taming Adversarial Constraints in CMDPsFrancesco Emanuele Stradi, Anna Lunghi, Matteo Castiglioni, Alberto Marchesi et al.NeurIPS 2025 · 2 citations
- Learning General Parameterized Policies for Infinite Horizon Average Reward Constrained MDPs via Primal-Dual Policy Gradient AlgorithmQinbo Bai, Washim Uddin Mondal, Vaneet AggarwalNeurIPS 2024 · 10 citations
