An Improved Finite-time Analysis of Temporal Difference Learning with Deep Neural Networks

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Ke, Zhifa, Wen, Zaiwen, Zhang, Junyu
Natura: Preprint
Pubblicazione: 2024
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866929337116131328
author Ke, Zhifa
Wen, Zaiwen
Zhang, Junyu
author_facet Ke, Zhifa
Wen, Zaiwen
Zhang, Junyu
contents Temporal difference (TD) learning algorithms with neural network function parameterization have well-established empirical success in many practical large-scale reinforcement learning tasks. However, theoretical understanding of these algorithms remains challenging due to the nonlinearity of the action-value approximation. In this paper, we develop an improved non-asymptotic analysis of the neural TD method with a general $L$-layer neural network. New proof techniques are developed and an improved new $\tilde{\mathcal{O}}(ε^{-1})$ sample complexity is derived. To our best knowledge, this is the first finite-time analysis of neural TD that achieves an $\tilde{\mathcal{O}}(ε^{-1})$ complexity under the Markovian sampling, as opposed to the best known $\tilde{\mathcal{O}}(ε^{-2})$ complexity in the existing literature.
format Preprint
id arxiv_https___arxiv_org_abs_2405_04017
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle An Improved Finite-time Analysis of Temporal Difference Learning with Deep Neural Networks
Ke, Zhifa
Wen, Zaiwen
Zhang, Junyu
Machine Learning
Artificial Intelligence
Optimization and Control
Temporal difference (TD) learning algorithms with neural network function parameterization have well-established empirical success in many practical large-scale reinforcement learning tasks. However, theoretical understanding of these algorithms remains challenging due to the nonlinearity of the action-value approximation. In this paper, we develop an improved non-asymptotic analysis of the neural TD method with a general $L$-layer neural network. New proof techniques are developed and an improved new $\tilde{\mathcal{O}}(ε^{-1})$ sample complexity is derived. To our best knowledge, this is the first finite-time analysis of neural TD that achieves an $\tilde{\mathcal{O}}(ε^{-1})$ complexity under the Markovian sampling, as opposed to the best known $\tilde{\mathcal{O}}(ε^{-2})$ complexity in the existing literature.
title An Improved Finite-time Analysis of Temporal Difference Learning with Deep Neural Networks
topic Machine Learning
Artificial Intelligence
Optimization and Control
url https://arxiv.org/abs/2405.04017