[1810.05934] A System for Massively Parallel Hyperparameter Tuning
Abstract:Modern learning models are characterized by large hyperparameter spaces and long training times. These properties, coupled with the rise of parallel computing and the growing demand to productionize machine learning workloads, motivate the need to develop mature hyperparameter optimization functionality in distributed computing settings. We address this challenge by first introducing a simple and robust hyperparameter optimization algorithm called ASHA, which exploits parallelism and aggressive early-stopping to tackle large-scale hyperparameter optimization problems. Our extensive empirical results show that ASHA outperforms existing state-of-the-art hyperparameter optimization methods; scales linearly with the number of workers in distributed settings; and is suitable for massive parallelism, as demonstrated on a task with 500 workers. We then describe several design decisions we encountered, along with our associated solutions, when integrating ASHA in Determined AI's end-to-end production-quality machine learning system that offers hyperparameter tuning as a service.
View PDF HTML (experimental) Abstract:Modern learning models are characterized by large hyperparameter spaces and long training times. These properties, coupled with the rise of parallel computing and the growing demand to productionize machine learning workloads, motivate the need to develop mature hyperparameter optimization functionality in distributed computing settings. We address this challenge by first introducing a simple and robust hyperparameter optimization algorithm called ASHA, which exploits parallelism and aggressive early-stopping to tackle large-scale hyperparameter…
saved by
related reading
- [2510.13786] The Art of Scaling Reinforcement Learning Compute for LLMsarxiv.org
- GitHub - google-research/tuning_playbook: A playbook for systematically maximizing the performance of deep learning models.github.com
- IsoCompute Playbook: Optimally Scaling Sampling Compute for RL Training of LLMscompute-optimal-rl-llm-scaling.github.io
- Pipeline-Parallelism: Distributed Training via Model Partitioningsiboehm.com
- Hyperparameter optimization - Wikipediaen.wikipedia.org
- [2509.14786] Pre-training under infinite computearxiv.org
- [2602.17616] Stable Asynchrony: Variance-Controlled Off-Policy RL for LLMsarxiv.org
- irhum.github.io - Tensor Parallelism with jax.pjitirhum.github.io
- Paradigms of Parallelism | Colossal-AIcolossalai.org
- [2601.08393] Controlled LLM Training on Spectral Spherearxiv.org
- Parallelism in Distributed Deep Learning · Better Tomorrow with Computer Scienceinsujang.github.io
- The Practitioner’s Guide to the Maximal Update Parameterization - Cerebrascerebras.ai