Hyper-parameter optimization based on soft actor critic and hierarchical mixture regularization

Chaoyue Liu,Yulai Zhang
DOI: https://doi.org/10.48550/arXiv.2112.04084
IF: 5.414
2021-12-08
Machine Learning
Abstract:Hyper-parameter optimization is a crucial problem in machine learning as it aims to achieve the state-of-the-art performance in any model. Great efforts have been made in this field, such as random search, grid search, Bayesian optimization. In this paper, we model hyper-parameter optimization process as a Markov decision process, and tackle it with reinforcement learning. A novel hyper-parameter optimization method based on soft actor critic and hierarchical mixture regularization has been proposed. Experiments show that the proposed method can obtain better hyper-parameters in a shorter time.
What problem does this paper attempt to address?