Expected Policy Gradients for Reinforcement Learning

Abstract

We propose expected policy gradients (EPG), which unify stochastic policy gradients (SPG) and deterministic policy gradients (DPG) for reinforcement learning. Inspired by expected sarsa, EPG integrates (or sums) across actions when estimating the gradient, instead of relying only on the action in the sampled trajectory. For continuous action spaces, we first derive a practical result for Gaussian policies and quadratic critics and then extend it to a universal analytical method, covering a broad class of actors and critics, including Gaussian, exponential families, and policies with bounded support. For Gaussian policies, we introduce an exploration method that uses covariance proportional to the matrix exponential of the scaled Hessian of the critic with respect to the actions. For discrete action spaces, we derive a variant of EPG based on softmax policies. We also establish a new general policy gradient theorem, of which the stochastic and deterministic policy gradient theorems are special cases. Furthermore, we prove that EPG reduces the variance of the gradient estimates without requiring deterministic policies and with little computational overhead. Finally, we provide an extensive experimental evaluation of EPG and show that it outperforms existing approaches on multiple challenging control domains.

36 pages, submitted for review to JMLR. This is an extended version of our paper in the AAAI-18 conference (arXiv:1706.05374)

Cited by in corpus (8)

• FedKL: Tackling Data Heterogeneity in Federated Reinforcement Learning by Penalizing KL Divergence
• A reinforcement learning approach to rare trajectory sampling
• VIREL: A Variational Inference Framework for Reinforcement Learning
• Settling the Variance of Multi-Agent Policy Gradients
• Fourier Policy Gradients
• Marginal Policy Gradients: A Unified Family of Estimators for Bounded Action Spaces with Applications
• Co-Adaptation of Algorithmic and Implementational Innovations in Inference-based Deep Reinforcement Learning
• Global Optimality and Finite Sample Analysis of Softmax Off-Policy Actor Critic under State Distribution Mismatch