Skip to main content

Momentum-Based policy gradient methods

Publication ,  Conference
Huang, F; Gao, S; Pei, J; Huang, H
Published in: 37th International Conference on Machine Learning, ICML 2020
January 1, 2020

In the paper, we propose a class of efficient momentum-based policy gradient methods for the model-free reinforcement learning, which use adaptive learning rates and do not require any large batches. Specifically, we propose a fast important-sampling momentum-based policy gradient (IS-MBPG) method based on a new momentum-based variance reduced technique and the importance sampling technique. We also propose a fast Hessian-aided momentum-based policy gradient (HA-MBPG) method based on the momentum-based variance reduced technique and the Hessian-aided technique. Moreover, we prove that both the IS-MBPG and HA-MBPG methods reach the best known sample complexity of O(ϵ -3) for finding an ϵ-stationary point of the nonconcave performance function, which only require one trajectory at each iteration. In particular, we present a non-adaptive version of IS-MBPG method, i.e., IS-MBPG*, which also reaches the best known sample complexity of O(ϵ -3) without any large batches. In the experiments, we apply four benchmark tasks to demonstrate the effectiveness of our algorithms.

Duke Scholars

Published In

37th International Conference on Machine Learning, ICML 2020

Publication Date

January 1, 2020

Volume

PartF168147-6

Start / End Page

4372 / 4383
 

Citation

APA
Chicago
ICMJE
MLA
NLM
Huang, F., Gao, S., Pei, J., & Huang, H. (2020). Momentum-Based policy gradient methods. In 37th International Conference on Machine Learning, ICML 2020 (Vol. PartF168147-6, pp. 4372–4383).
Huang, F., S. Gao, J. Pei, and H. Huang. “Momentum-Based policy gradient methods.” In 37th International Conference on Machine Learning, ICML 2020, PartF168147-6:4372–83, 2020.
Huang F, Gao S, Pei J, Huang H. Momentum-Based policy gradient methods. In: 37th International Conference on Machine Learning, ICML 2020. 2020. p. 4372–83.
Huang, F., et al. “Momentum-Based policy gradient methods.” 37th International Conference on Machine Learning, ICML 2020, vol. PartF168147-6, 2020, pp. 4372–83.
Huang F, Gao S, Pei J, Huang H. Momentum-Based policy gradient methods. 37th International Conference on Machine Learning, ICML 2020. 2020. p. 4372–4383.

Published In

37th International Conference on Machine Learning, ICML 2020

Publication Date

January 1, 2020

Volume

PartF168147-6

Start / End Page

4372 / 4383