Provably Efficient Infinite-Horizon Average-Reward Reinforcement Learning with Linear Function Approximation

Fuente: arXiv
Enregistré dans:
Détails bibliographiques
Auteurs principaux: Chae, Woojin, Lee, Dabeen
Format: Preprint
Publié: 2024
Sujets:
Accès en ligne:
Tags: Ajouter un tag
Pas de tags, Soyez le premier à ajouter un tag!
_version_ 1866909323758665728
author Chae, Woojin
Lee, Dabeen
author_facet Chae, Woojin
Lee, Dabeen
contents This paper proposes a computationally tractable algorithm for learning infinite-horizon average-reward linear Markov decision processes (MDPs) and linear mixture MDPs under the Bellman optimality condition. While guaranteeing computational efficiency, our algorithm for linear MDPs achieves the best-known regret upper bound of $\widetilde{\mathcal{O}}(d^{3/2}\mathrm{sp}(v^*)\sqrt{T})$ over $T$ time steps where $\mathrm{sp}(v^*)$ is the span of the optimal bias function $v^*$ and $d$ is the dimension of the feature mapping. For linear mixture MDPs, our algorithm attains a regret bound of $\widetilde{\mathcal{O}}(d\cdot\mathrm{sp}(v^*)\sqrt{T})$. The algorithm applies novel techniques to control the covering number of the value function class and the span of optimistic estimators of the value function, which is of independent interest.
format Preprint
id arxiv_https___arxiv_org_abs_2409_10772
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Provably Efficient Infinite-Horizon Average-Reward Reinforcement Learning with Linear Function Approximation
Chae, Woojin
Lee, Dabeen
Machine Learning
Data Structures and Algorithms
Optimization and Control
This paper proposes a computationally tractable algorithm for learning infinite-horizon average-reward linear Markov decision processes (MDPs) and linear mixture MDPs under the Bellman optimality condition. While guaranteeing computational efficiency, our algorithm for linear MDPs achieves the best-known regret upper bound of $\widetilde{\mathcal{O}}(d^{3/2}\mathrm{sp}(v^*)\sqrt{T})$ over $T$ time steps where $\mathrm{sp}(v^*)$ is the span of the optimal bias function $v^*$ and $d$ is the dimension of the feature mapping. For linear mixture MDPs, our algorithm attains a regret bound of $\widetilde{\mathcal{O}}(d\cdot\mathrm{sp}(v^*)\sqrt{T})$. The algorithm applies novel techniques to control the covering number of the value function class and the span of optimistic estimators of the value function, which is of independent interest.
title Provably Efficient Infinite-Horizon Average-Reward Reinforcement Learning with Linear Function Approximation
topic Machine Learning
Data Structures and Algorithms
Optimization and Control
url https://arxiv.org/abs/2409.10772