Abstract:Hyperparameter optimization (HPO) is an important step in machine learning (ML) model development, but common practices are archaic -- primarily relying on manual or grid searches. This is partly because adopting advanced HPO algorithms introduces added complexity to the workflow, leading to longer computation times. This poses a notable challenge to ML applications, as suboptimal hyperparameter selections curtail the potential of ML model performance, ultimately obstructing the full exploitation of ML techniques. In this article, we present a two-step HPO method as a strategic solution to curbing computational demands and wait times, gleaned from practical experiences in applied ML parameterization work. The initial phase involves a preliminary evaluation of hyperparameters on a small subset of the training dataset, followed by a re-evaluation of the top-performing candidate models post-retraining with the entire training dataset. This two-step HPO method is universally applicable across HPO search algorithms, and we argue it has attractive efficiency gains. As a case study, we present our recent application of the two-step HPO method to the development of neural network emulators for aerosol activation. Although our primary use case is a data-rich limit with many millions of samples, we also find that using up to 0.0025% of the data (a few thousand samples) in the initial step is sufficient to find optimal hyperparameter configurations from much more extensive sampling, achieving up to 135-times speedup. The benefits of this method materialize through an assessment of hyperparameters and model performance, revealing the minimal model complexity required to achieve the best performance. The assortment of top-performing models harvested from the HPO process allows us to choose a high-performing model with a low inference cost for efficient use in global climate models (GCMs).

Two-step hyperparameter optimization method: Accelerating hyperparameter search by using a fraction of a training dataset

Hyperparameter optimization: Classics, acceleration, online, multi-objective, and tools

Multi-level Training and Bayesian Optimization for Economical Hyperparameter Optimization

Hyper-Parameter Optimization: A Review of Algorithms and Applications

HomOpt: A Homotopy-Based Hyperparameter Optimization Method

Practical and sample efficient zero-shot HPO

Transferable Neural Processes for Hyperparameter Optimization

Efficient Hyper-parameter Optimization for NLP Applications.

Hippo: Taming Hyper-parameter Optimization of Deep Learning with Stage Trees

HyperTuner: A Cross-Layer Multi-Objective Hyperparameter Auto-Tuning Framework for Data Analytic Services

AutoML with Parallel Genetic Algorithm for Fast Hyperparameters Optimization in Efficient IoT Time Series Prediction

Optimizing Large-Scale Hyperparameters via Automated Learning Algorithm

Stage-based Hyper-parameter Optimization for Deep Learning

Automatic Hyper-Parameter Optimization Based on Mapping Discovery from Data to Hyper-Parameters

Methodology for Hyperparameter Tuning of Deep Neural Networks for Efficient and Accurate Molecular Property Prediction

AntTune: An Efficient Distributed Hyperparameter Optimization System for Large-Scale Data.

Auptimizer -- an Extensible, Open-Source Framework for Hyperparameter Tuning

BTTackler: A Diagnosis-based Framework for Efficient Deep Learning Hyperparameter Optimization

HyperQ-Opt: Q-learning for Hyperparameter Optimization

Using Known Information to Accelerate HyperParameters Optimization Based on SMBO

Hyperopt: a Python library for model selection and hyperparameter optimization