English

Reinforcement-based Simultaneous Algorithm and its Hyperparameters Selection

Machine Learning 2016-11-08 v1 Artificial Intelligence Machine Learning

Abstract

Many algorithms for data analysis exist, especially for classification problems. To solve a data analysis problem, a proper algorithm should be chosen, and also its hyperparameters should be selected. In this paper, we present a new method for the simultaneous selection of an algorithm and its hyperparameters. In order to do so, we reduced this problem to the multi-armed bandit problem. We consider an algorithm as an arm and algorithm hyperparameters search during a fixed time as the corresponding arm play. We also suggest a problem-specific reward function. We performed the experiments on 10 real datasets and compare the suggested method with the existing one implemented in Auto-WEKA. The results show that our method is significantly better in most of the cases and never worse than the Auto-WEKA.

Keywords

Cite

@article{arxiv.1611.02053,
  title  = {Reinforcement-based Simultaneous Algorithm and its Hyperparameters Selection},
  author = {Valeria Efimova and Andrey Filchenkov and Anatoly Shalyto},
  journal= {arXiv preprint arXiv:1611.02053},
  year   = {2016}
}
R2 v1 2026-06-22T16:44:11.808Z