TY - JOUR AU - Akella, Ravi Tej AU - Azizzadenesheli, Kamyar AU - Ghavamzadeh, Mohammad AU - Anandkumar, Animashree AU - Yue, Yisong PY - 2021/05/18 Y2 - 2024/03/28 TI - Deep Bayesian Quadrature Policy Optimization JF - Proceedings of the AAAI Conference on Artificial Intelligence JA - AAAI VL - 35 IS - 8 SE - AAAI Technical Track on Machine Learning I DO - 10.1609/aaai.v35i8.16817 UR - https://ojs.aaai.org/index.php/AAAI/article/view/16817 SP - 6600-6608 AB - We study the problem of obtaining accurate policy gradient estimates using a finite number of samples. Monte-Carlo methods have been the default choice for policy gradient estimation, despite suffering from high variance in the gradient estimates. On the other hand, more sample efficient alternatives like Bayesian quadrature methods have received little attention due to their high computational complexity. In this work, we propose deep Bayesian quadrature policy gradient (DBQPG), a computationally efficient high-dimensional generalization of Bayesian quadrature, for policy gradient estimation. We show that DBQPG can substitute Monte-Carlo estimation in policy gradient methods, and demonstrate its effectiveness on a set of continuous control benchmarks. In comparison to Monte-Carlo estimation, DBQPG provides (i) more accurate gradient estimates with a significantly lower variance, (ii) a consistent improvement in the sample complexity and average return for several deep policy gradient algorithms, and, (iii) the uncertainty in gradient estimation that can be incorporated to further improve the performance. ER -