Functional Acceleration for Policy Mirror Descent

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: Chelu, Veronica, Precup, Doina
Formato: Preprint
Publicado: 2024
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866916661298200576
author Chelu, Veronica
Precup, Doina
author_facet Chelu, Veronica
Precup, Doina
contents We apply functional acceleration to the Policy Mirror Descent (PMD) general family of algorithms, which cover a wide range of novel and fundamental methods in Reinforcement Learning (RL). Leveraging duality, we propose a momentum-based PMD update. By taking the functional route, our approach is independent of the policy parametrization and applicable to large-scale optimization, covering previous applications of momentum at the level of policy parameters as a special case. We theoretically analyze several properties of this approach and complement with a numerical ablation study, which serves to illustrate the policy optimization dynamics on the value polytope, relative to different algorithmic design choices in this space. We further characterize numerically several features of the problem setting relevant for functional acceleration, and lastly, we investigate the impact of approximation on their learning mechanics.
format Preprint
id arxiv_https___arxiv_org_abs_2407_16602
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Functional Acceleration for Policy Mirror Descent
Chelu, Veronica
Precup, Doina
Machine Learning
Artificial Intelligence
We apply functional acceleration to the Policy Mirror Descent (PMD) general family of algorithms, which cover a wide range of novel and fundamental methods in Reinforcement Learning (RL). Leveraging duality, we propose a momentum-based PMD update. By taking the functional route, our approach is independent of the policy parametrization and applicable to large-scale optimization, covering previous applications of momentum at the level of policy parameters as a special case. We theoretically analyze several properties of this approach and complement with a numerical ablation study, which serves to illustrate the policy optimization dynamics on the value polytope, relative to different algorithmic design choices in this space. We further characterize numerically several features of the problem setting relevant for functional acceleration, and lastly, we investigate the impact of approximation on their learning mechanics.
title Functional Acceleration for Policy Mirror Descent
topic Machine Learning
Artificial Intelligence
url https://arxiv.org/abs/2407.16602