RHOASo: An Early Stop Hyper-Parameter Optimization Algorithm

• Main Authors: Ángel Luis Muñoz Castañeda, Noemí DeCastro-García, David Escudero García
• Format: Article
• Language: English
• Published: MDPI AG 2021-09-01
• Series: Mathematics
• Subjects: hyperparameters; machine learning; optimization; inference
• Online Access: https://www.mdpi.com/2227-7390/9/18/2334

This work proposes a new algorithm for optimizing hyper-parameters of a machine learning algorithm, RHOASo, based on conditional optimization of concave asymptotic functions. A comparative analysis of the algorithm is presented, giving particular emphasis to two important properties: the capability of the algorithm to work efficiently with a small part of a dataset and to finish the tuning process automatically, that is, without making explicit, by the user, the number of iterations that the algorithm must perform. Statistical analyses over 16 public benchmark datasets comparing the performance of seven hyper-parameter optimization algorithms with RHOASo were carried out. The efficiency of RHOASo presents the positive statistically significant differences concerning the other hyper-parameter optimization algorithms considered in the experiments. Furthermore, it is shown that, on average, the algorithm needs around <inline-formula><math xmlns="http://www.w3.org/1998/Math/MathML" display="inline"><semantics><mrow><mn>70</mn><mo>%</mo></mrow></semantics></math></inline-formula> of the iterations needed by other algorithms to achieve competitive performance. The results show that the algorithm presents significant stability regarding the size of the used dataset partition.