Bayesian hyperparameter optimization for ensemble learning

Bayesian hyperparameter optimization for ensemble learning

Lévesque, Julien Charles and Gagné, Christian and Sabourin, Robert

32nd Conference on Uncertainty in Artificial Intelligence 2016, UAI 2016 2016

Abstract : In this paper, we bridge the gap between hyperparameter optimization and ensemble learning by performing Bayesian optimization of an ensemble with regards to its hyperparameters. Our method consists in building a fixed-size ensemble, optimizing the configuration of one classifier of the ensemble at each iteration of the hyperparameter optimization algorithm, taking into consideration the interaction with the other models when evaluating potential performances. We also consider the case where the ensemble is to be reconstructed at the end of the hyperparameter optimization phase, through a greedy selection over the pool of models generated during the optimization. We study the performance of our proposed method on three different hyperparameter spaces, showing that our approach is better than both the best single model and a greedy ensemble construction over the models produced by a standard Bayesian optimization.