Accelerating Value Iteration with Anchoring

Abstract

Value Iteration (VI) is foundational to the theory and practice of modern reinforcement learning, and it is known to converge at a $\mathcal{O}(\gamma^k)$-rate. Surprisingly, however, the optimal rate for the VI setup was not known, and finding a general acceleration mechanism has been an open problem. In this paper, we present the first accelerated VI for both the Bellman consistency and optimality operators. Our method, called Anc-VI, is based on an \emph{anchoring} mechanism (distinct from Nesterov's acceleration), and it reduces the Bellman error faster than standard VI. In particular, Anc-VI exhibits a $\mathcal{O}(1/k)$-rate for $\gamma\approx 1$ or even $\gamma=1$, while standard VI has rate $\mathcal{O}(1)$ for $\gamma\ge 1-1/k$, where $k$ is the iteration count. We also provide a complexity lower bound matching the upper bound up to a constant factor of $4$, thereby establishing optimality of the accelerated rate of Anc-VI. Finally, we show that the anchoring mechanism provides the same benefit in the approximate VI and Gauss--Seidel VI setups as well.

Cite

Text

Lee and Ryu. "Accelerating Value Iteration with Anchoring." Neural Information Processing Systems, 2023.

Markdown

[Lee and Ryu. "Accelerating Value Iteration with Anchoring." Neural Information Processing Systems, 2023.](https://mlanthology.org/neurips/2023/lee2023neurips-accelerating/)

BibTeX

@inproceedings{lee2023neurips-accelerating,
  title     = {{Accelerating Value Iteration with Anchoring}},
  author    = {Lee, Jongmin and Ryu, Ernest},
  booktitle = {Neural Information Processing Systems},
  year      = {2023},
  url       = {https://mlanthology.org/neurips/2023/lee2023neurips-accelerating/}
}