271

Nearly Optimal Regret for Learning Adversarial MDPs with Linear Function Approximation

International Conference on Artificial Intelligence and Statistics (AISTATS), 2021
Abstract

We study the reinforcement learning for finite-horizon episodic Markov decision processes with adversarial reward and full information feedback, where the unknown transition probability function is a linear function of a given feature mapping. We propose an optimistic policy optimization algorithm with Bernstein bonus and show that it can achieve O~(dHT)\tilde{O}(dH\sqrt{T}) regret, where HH is the length of the episode, TT is the number of interaction with the MDP and dd is the dimension of the feature mapping. Furthermore, we also prove a matching lower bound of Ω~(dHT)\tilde{\Omega}(dH\sqrt{T}) up to logarithmic factors. To the best of our knowledge, this is the first computationally efficient, nearly minimax optimal algorithm for adversarial Markov decision processes with linear function approximation.

View on arXiv
Comments on this paper