December 2023

Journal

Distributed Bayesian optimization of deep reinforcement learning algorithms

By:
Young, Michael T; Hinkle, Jacob D; Kannan, Ramakrishnan ; Ramanathan, Arvind
Journal Name:
Journal of Parallel and Distributed Computing
Page Number:
43-52
Volume:
139
Issue Number:
NA
Publication Date:
December 2023
View DOI Listing:
https://doi.org/10.1016/j.jpdc.2019.07.008

Abstract

Significant strides have been made in supervised learning settings thanks to the successful application of deep learning. Now, recent work has brought the techniques of deep learning to bear on sequential decision processes in the area of deep reinforcement learning (DRL). Currently, little is known regarding hyperparameter optimization for DRL algorithms. Given that DRL algorithms are computationally intensive to train, and are known to be sample inefficient, optimizing model hyperparameters for DRL presents significant challenges to established techniques. We provide an open source, distributed Bayesian model-based optimization algorithm, HyperSpace, and show that it consistently outperforms standard hyperparameter optimization techniques across three DRL algorithms.