HyperSpace: Distributed Bayesian Hyperparameter Optimization

• M. Todd Young Oak Ridge National Laboratory
• Jacob Hinkle Oak Ridge National Laboratory
• Arvind Ramanathan Oak Ridge National Laboratory
• Ramakrishnan Kannan Oak Ridge National Laboratory

Resumo

As machine learning models continue to increase in complexity, so does the potential number of free model parameters commonly known as hyperparameters. While there has been considerable progress toward finding optimal configurations of these hyperparameters, many optimization procedures are treated as black boxes. We believe optimization methods should not only return a set of optimized hyperparameters, but also give insight into the effects of model hyperparameter settings. To this end, we present HyperSpace, a parallel implementation of Bayesian sequential model-based optimization. HyperSpace leverages high performance computing (HPC) resources to better understand unknown, potentially non-convex hyperparameter search spaces. We show that it is possible to learn the dependencies between model hyperparameters through the optimization process. By partitioning large search spaces and running many optimization procedures in parallel, we also show that it is possible to discover families of good hyperparameter settings over a variety of models including unsupervised clustering, regression, and classification tasks.

Palavras-chave: Optimization, Bayes methods, Computational modeling, Gaussian processes, Linear programming, Machine learning algorithms, Kernel, Bayesian optimization, SMBO, parallel computing, HPC