English

Horizon-free Reinforcement Learning in Adversarial Linear Mixture MDPs

Machine Learning 2023-05-16 v1 Optimization and Control Machine Learning

Abstract

Recent studies have shown that episodic reinforcement learning (RL) is no harder than bandits when the total reward is bounded by 11, and proved regret bounds that have a polylogarithmic dependence on the planning horizon HH. However, it remains an open question that if such results can be carried over to adversarial RL, where the reward is adversarially chosen at each episode. In this paper, we answer this question affirmatively by proposing the first horizon-free policy search algorithm. To tackle the challenges caused by exploration and adversarially chosen reward, our algorithm employs (1) a variance-uncertainty-aware weighted least square estimator for the transition kernel; and (2) an occupancy measure-based technique for the online search of a \emph{stochastic} policy. We show that our algorithm achieves an O~((d+log(S2A))K)\tilde{O}\big((d+\log (|\mathcal{S}|^2 |\mathcal{A}|))\sqrt{K}\big) regret with full-information feedback, where dd is the dimension of a known feature mapping linearly parametrizing the unknown transition kernel of the MDP, KK is the number of episodes, S|\mathcal{S}| and A|\mathcal{A}| are the cardinalities of the state and action spaces. We also provide hardness results and regret lower bounds to justify the near optimality of our algorithm and the unavoidability of logS\log|\mathcal{S}| and logA\log|\mathcal{A}| in the regret bound.

Keywords

Cite

@article{arxiv.2305.08359,
  title  = {Horizon-free Reinforcement Learning in Adversarial Linear Mixture MDPs},
  author = {Kaixuan Ji and Qingyue Zhao and Jiafan He and Weitong Zhang and Quanquan Gu},
  journal= {arXiv preprint arXiv:2305.08359},
  year   = {2023}
}

Comments

34 pages

R2 v1 2026-06-28T10:34:20.337Z