Hyperparameter tuning for big data using Bayesian optimisation

Citations Over TimeTop 10% of 2016 papers

Abstract

Hyperparameters play a crucial role in the model selection of machine learning algorithms. Tuning these hyperparameters can be exhaustive when the data is large. Bayesian optimisation has emerged as an efficient tool for hyperparameter tuning of machine learning algorithms. In this paper, we propose a novel framework for tuning the hyperparameters for big data using Bayesian optimisation. We divide the big data into chunks and generate hyperparameter configurations for the chunks using the standard Bayesian optimisation. We utilise this information from the chunks for hyperparameter tuning on big data using a transfer learning setting. We evaluate the performance of the proposed method on the task of tuning hyperparameters of two machine learning algorithms. We show that our method achieves the best available hyperparameter configuration within less computational time compared to the state-of-art hyperparameter tuning methods.

Related Papers

• → Initializing Bayesian Hyperparameter Optimization via Meta-Learning(2015)394 cited
• → Deep Neural Network with Hyperparameter Tuning for Detection of Heart Disease(2021)23 cited
• → Evaluation of Hyperparameter-Optimization Approaches in an Industrial Federated Learning System(2022)14 cited
• → Bayesian Hyperparameter Optimization for Ensemble Learning(2016)35 cited
• → Evaluation of Hyperparameter-Optimization Approaches in an Industrial Federated Learning System(2021)