English

Augment-Reinforce-Merge Policy Gradient for Binary Stochastic Policy

Machine Learning 2019-03-14 v1 Artificial Intelligence Machine Learning

Abstract

Due to the high variance of policy gradients, on-policy optimization algorithms are plagued with low sample efficiency. In this work, we propose Augment-Reinforce-Merge (ARM) policy gradient estimator as an unbiased low-variance alternative to previous baseline estimators on tasks with binary action space, inspired by the recent ARM gradient estimator for discrete random variable models. We show that the ARM policy gradient estimator achieves variance reduction with theoretical guarantees, and leads to significantly more stable and faster convergence of policies parameterized by neural networks.

Keywords

Cite

@article{arxiv.1903.05284,
  title  = {Augment-Reinforce-Merge Policy Gradient for Binary Stochastic Policy},
  author = {Yunhao Tang and Mingzhang Yin and Mingyuan Zhou},
  journal= {arXiv preprint arXiv:1903.05284},
  year   = {2019}
}
R2 v1 2026-06-23T08:06:31.920Z