Rate-Optimal Policy Optimization for Linear Markov Decision Processes
Uri Sherman, Alon Cohen, Tomer Koren, Yishay Mansour
Abstract
We study regret minimization in online episodic linear Markov Decision Processes, and propose a policy optimization algorithm that is computationally efficient, and obtains rate optimal O( √ K) regret where K denotes the number of episodes. Our work is the first to establish the optimal rate (in terms of K) of convergence in the stochastic setting with bandit feedback using a policy optimization based approach, and the first to establish the optimal rate in the adversarial setup with full information feedback, for which no algorithm with an optimal rate guarantee was previously known.
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.
Your agent calls
Luneget_paper_fulltext
Free to start. No credit card required.
Terminal
Install the CLIlune papers fulltext e1018ffd-5afd-479f-91cd-b1225a752bd4Cited by top-tier papers13
- Greedy Sampling Is Provably Efficient For RLHFDi Wu, Chengshuai Shi, Jing Yang, Cong ShenNeurIPS 2025 · 11 citations
- Imitation Learning in Discounted Linear MDPs without exploration assumptionsLuca Viano, Stratis Skoulakis, Volkan CevherICML 2024 · 10 citations
- Warm-up Free Policy Optimization: Improved Regret in Linear Markov Decision ProcessesAsaf B. Cassel, Aviv RosenbergNeurIPS 2024 · 6 citations
- Near-Optimal Regret in Linear MDPs with Aggregate Bandit FeedbackAsaf B. Cassel, Haipeng Luo, Aviv Rosenberg, Dmitry SotnikovICML 2024 · 6 citations
- Rethinking Model-based, Policy-based, and Value-based Reinforcement Learning via the Lens of Representation ComplexityGuhao Feng, Han ZhongNeurIPS 2024 · 5 citations
Builds on22
- Training language models to follow instructions with human feedbackLong Ouyang, Jeffrey Wu, Xu Jiang, Diogo Almeida et al.NeurIPS 2022 · 24,707 citations
- Model-Based Reinforcement Learning with Value-Targeted RegressionAlex Ayoub, Zeyu Jia, Csaba Szepesvári, Mengdi Wang et al.ICML 2020 · 324 citations
- Reinforcement Learning in Feature Space: Matrix Bandit, Kernels, and Regret BoundLin Yang, Mengdi WangICML 2020 · 308 citations
- Provably Efficient Exploration in Policy OptimizationQi Cai, Zhuoran Yang, Chi Jin, Zhaoran WangICML 2020 · 304 citations
- Bellman Eluder Dimension: New Rich Classes of RL Problems, and Sample-Efficient AlgorithmsChi Jin, Qinghua Liu, Sobhan MiryoosefiNeurIPS 2021 · 264 citations
Related papers
- Towards Optimal Regret in Adversarial Linear MDPs with Bandit FeedbackHaolin Liu, Chen-Yu Wei, Julian ZimmertICLR 2024 · 11 citations
- Learning Adversarial Markov Decision Processes with Delayed FeedbackTal Lancewicki, Aviv Rosenberg, Yishay MansourAAAI 2022 · 40 citations
- Optimistic Policy Optimization with Bandit FeedbackLior Shani, Yonathan Efroni, Aviv Rosenberg, Shie MannorICML 2020 · 100 citations
- Learning Adversarial Markov Decision Processes with Bandit Feedback and Unknown TransitionChi Jin, Tiancheng Jin, Haipeng Luo, Suvrit Sra et al.ICML 2020 · 117 citations
- Near-Optimal Regret for Adversarial MDP with Delayed Bandit FeedbackTiancheng Jin, Tal Lancewicki, Haipeng Luo, Yishay Mansour et al.NeurIPS 2022 · 29 citations
