A System for Massively Parallel Hyperparameter Tuning
Not on the map. Why?
Abstract (arXiv)
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.
Key facts. Cited by 201 papers in the map (data through 7 Oct 2026). First posted on arXiv 13 Oct 2018.
Builds on
No reference list yet.
Built on by
Citing papers per year: 2021: 30; 2022: 35; 2023: 37; 2024: 35; 2025: 33; 2026: 31.
201 papers.
Sort:
By area:
- 2021Advances in Neural Renderingcited by 365
- 2022Tensor Programs V: Tuning Large Neural Networks via Zero-Shot Hyperparameter Transfercited by 272
- 2023Neural Architecture Search: Insights from 1000 Paperscited by 136
- 2021The Benchmark Lotterycited by 133
- 2021HPOBench: A Collection of Reproducible Multi-Fidelity Benchmark Problems for HPOcited by 111
- 2021How Powerful are Performance Predictors in Neural Architecture Search?cited by 88
- 2022Zeus: Understanding and Optimizing GPU Energy Consumption of DNN Trainingcited by 78
- 2022Efficient Methods for Natural Language Processing: A Surveycited by 78
- 2021YAHPO Gym -- An Efficient Multi-Objective Multi-Fidelity Benchmark for Hyperparameter Optimizationcited by 61
- 2023Scalable Membership Inference Attacks via Quantile Regressioncited by 61
- 2021Model Zoo: A Growing "Brain" That Learns Continuallycited by 53
- 2022Model Zoos: A Dataset of Diverse Populations of Neural Network Modelscited by 38
38 more
- 2022Multi-Objective Hyperparameter Optimization in Machine Learning -- An Overviewcited by 37
- 2023Cross-Modal Fine-Tuning: Align then Refinecited by 34
- 2021NAS-Bench-360: Benchmarking Neural Architecture Search on Diverse Taskscited by 32
- 2023PriorBand: Practical Hyperparameter Optimization in the Age of Deep Learningcited by 31
- 2023CUDA: Curriculum of Data Augmentation for Long-Tailed Recognitioncited by 30
- 2024In-Context Freeze-Thaw Bayesian Optimization for Hyperparameter Optimizationcited by 26
- 2022Revisiting Simple Regret: Fast Rates for Returning a Good Armcited by 25
- 2023FairTune: Optimizing Parameter Efficient Fine Tuning for Fairness in Medical Image Analysiscited by 23
- 2024Specialized Foundation Models Struggle to Beat Supervised Baselinescited by 23
- 2022AUTOMATA: Gradient Based Data Subset Selection for Compute-Efficient Hyper-parameter Tuningcited by 22
- 2021Automatic Termination for Hyperparameter Optimizationcited by 21
- 2021NAS-Bench-x11 and the Power of Learning Curvescited by 20
- 2021Bag of Baselines for Multi-objective Joint Neural Architecture Search and Hyperparameter Optimizationcited by 19
- 2022Efficient Architecture Search for Diverse Taskscited by 18
- 2022Supervising the Multi-Fidelity Race of Hyperparameter Configurationscited by 15
- 2021AutoLRS: Automatic Learning-Rate Schedule by Bayesian Optimization on the Flycited by 14
- 2022Machine Learning for Particle Flow Reconstruction at CMScited by 14
- 2024ImageNot: A contrast with ImageNet preserves model rankingscited by 14
- 2022Hyper-Tune: Towards Efficient Hyper-parameter Tuning at Scalecited by 13
- 2022PASHA: Efficient HPO and NAS with Progressive Resource Allocationcited by 13
- 2023Event-Based Contrastive Learning for Medical Time Seriescited by 13
- 2021An Empirical Study on Hyperparameter Optimization for Fine-Tuning Pre-trained Language Modelscited by 12
- 2024Efficient Data Subset Selection to Generalize Training Across Models: Transductive and Inductive Networkscited by 12
- 2021Tune It or Don't Use It: Benchmarking Data-Efficient Image Classificationcited by 11
- 2024Koopman-Assisted Reinforcement Learningcited by 11
- 2024HW-GPT-Bench: Hardware-Aware Architecture Benchmark for Language Modelscited by 10
- 2022Open-source FPGA-ML codesign for the MLPerf Tiny Benchmarkcited by 9
- 2023MILO: Model-Agnostic Subset Selection Framework for Efficient Model Training and Tuningcited by 9
- 2023Machine learning detects terminal singularitiescited by 9
- 2024Hyperband-based Bayesian Optimization for Black-box Prompt Selectioncited by 9
- 2021Self-attention Presents Low-dimensional Knowledge Graph Embeddings for Link Predictioncited by 8
- 2021Scalable One-Pass Optimisation of High-Dimensional Weight-Update Hyperparameters by Implicit Differentiationcited by 7
- 2024Input-to-State Stable Coupled Oscillator Networks for Closed-form Model-based Control in Latent Spacecited by 7
- 2021Machine learning classification of non-Markovian noise disturbing quantum dynamicscited by 6
- 2021RIANN -- A Robust Neural Network Outperforms Attitude Estimation Filterscited by 6
- 2021FairAutoML: Embracing Unfairness Mitigation in AutoMLcited by 6
- 2022Asynchronous Decentralized Bayesian Optimization for Large Scale Hyperparameter Optimizationcited by 6
- 2022Speeding up NAS with Adaptive Subset Selectioncited by 6
50 most cited of 201 All 201 (JSON)
- 2026EdgeCraft: Automated Model Crafting for Edge IoT28 Sep 2026
- 2026Rank Collapse Is Recoverable, Growing |Q| Is Not: Out-of-Sample Early Warning for Value Divergence in High-UTD Soft Actor-Critic26 Sep 2026
- 2026Revisiting Multi-Object Tracking Baselines: Hyperparameter Optimization with Multi-Fidelity Greedy Coordinate Search10 Sep 2026
- 2026Breadth Beats Depth: Improving GCG-Based Jailbreak Optimization with Breadth-Oriented Suffix Search2 Sep 2026
- 2026Neural Symbollic Regression Using Deep Learning and Sparse Modelling1 Sep 2026
- 2026Knowing When Not to Reuse: Conditional Experience Transfer in Autonomous LLM Post-Training27 Aug 2026
- 2026Steering Recurrent Reasoners at Inference Time with Readout Feedback25 Aug 2026
- 2026Differential Learning for Robust Prediction of Thermal Stability with Application to Energetic Materials24 Aug 2026
- 2026Candidate-Fate Accounting for Transparent Sensor Diagnostic Pipeline Search19 Aug 2026
- 2026A Convolutional Layer Activation Dimensionality Reduction for Out-of-Distribution and Adversarial Attack Detection Methods10 Aug 2026
- 2026Automated Discovery Has No Universally Superior Harness20 Jul 2026
- 2026LLMZero: Discovering Adaptive Training Strategies for RL Post-Training via LLM Agents16 Jun 2026
38 more
- 2026BBOmix: A Tabular Benchmark for Hyperparameter Optimization of Unsupervised Biological Representation Learning3 Jun 2026
- 2026Two-Fidelity Best-Action Identification for Stochastic Minimax Tree1 Jun 2026
- 2026EEG-FM-Audit: A Systematic Evaluation and Analysis Pipeline for EEG Foundation Models26 May 2026
- 2026KernelBrain: Coarse-to-Fine, Budget-Aware Search for Agentic GPU Kernel Optimization19 May 2026
- 2026Active Budget Allocation for Efficient Scaling Law Estimation via Surrogate-Guided Pruning17 May 2026
- 2026BoLT: A Benchmark to Democratize Black-box Optimization Research for Expensive LLM Tasks16 May 2026
- 2026Spectral Lens: Activation and Gradient Spectra as Diagnostics of LLM Optimization7 May 2026
- 2026Looking for the Bottleneck in Fine-grained Temporal Relation Classification27 Apr 2026
- 2026Forecasting Individual NetFlows using a Predictive Masked Graph Autoencoder22 Apr 2026
- 2026QuickScope: Certifying Hard Questions in Dynamic LLM Benchmarks20 Apr 2026
- 2026ALTO: Adaptive LoRA Tuning and Orchestration for Heterogeneous LoRA Training Workloads7 Apr 2026
- 2026System-Anchored Knee Estimation for Low-Cost Context Window Selection in PDE Forecasting26 Mar 2026
- 2026Translation Invariance of Neural Operators for the FitzHugh-Nagumo Model18 Mar 2026
- 2026Using the SEKF to Transfer NN Models of Dynamical Systems with Limited Data2 Mar 2026
- 2026Learning with a Budget: Identifying the Best Arm with Resource Constraints27 Feb 2026
- 2026Dynamic Hyperparameter Importance for Efficient Multi-Objective Optimization6 Jan 2026
- 2026The Machine Learning Canvas: Empirical Findings on Why Strategy Matters More Than AI Code Generation5 Jan 2026
- 2026BSAT: B-Spline Adaptive Tokenizer for Long-Term Time Series Forecasting2 Jan 2026
- 2026Integrating Multi-Armed Bandit, Active Learning, and Distributed Computing for Scalable Optimization2 Jan 2026
- 2025The Seismic Wavefield Common Task Framework22 Dec 2025
- 2025Efficient Hyperparameter Search for Non-Stationary Model Training1 Dec 2025
- 2025Multi-objective Hyperparameter Optimization in the Age of Deep Learning11 Nov 2025
- 2025Provable Accelerated Bayesian Optimization with Knowledge Transfer5 Nov 2025
- 2025Cost-Sensitive Freeze-thaw Bayesian Optimization for Efficient Hyperparameter Tuning24 Oct 2025
- 2025Automated Machine Learning for Unsupervised Tabular Tasks8 Oct 2025
- 2025Tuning the Tuner: Introducing Hyperparameter Optimization for Auto-Tuning30 Sep 2025
- 2025Quantile-Scaled Bayesian Optimization Using Rank-Only Feedback28 Sep 2025
- 2025Fast Linear Solvers via AI-Tuned Markov Chain Monte Carlo-based Matrix Inversion22 Sep 2025
- 2025Tune My Adam, Please!27 Aug 2025
- 2025GRAID: Synthetic Data Generation with Geometric Constraints and Multi-Agentic Reflection for Harmful Content Detection23 Aug 2025
- 2025Adaptive Model-Predictive Control of a Soft Continuum Robot Using a Physics-Informed Neural Network Based on Cosserat Rod Theory18 Aug 2025
- 2025Towards Generalized Parameter Tuning in Coherent Ising Machines: A Portfolio-Based Approach27 Jul 2025
- 2025Taming Data Challenges in ML-based Security Tasks Using Generative AI8 Jul 2025
- 2025Missing-Modality-Aware Graph Neural Network for Cancer Classification28 Jun 2025
- 2025T-SHRED: Symbolic Regression for Regularization and Model Discovery with Transformer Shallow Recurrent Decoders18 Jun 2025
- 2025carps: A Framework for Comparing N Hyperparameter Optimizers on M Benchmarks6 Jun 2025
- 2025You Only Train Once4 Jun 2025
- 2025Learning High-dimensional Ionic Model Dynamics Using Fourier Neural Operators20 May 2025
50 newest of 201 All 201 (JSON)
- 2021Advances in Neural Renderingcited by 365
- 2022Tensor Programs V: Tuning Large Neural Networks via Zero-Shot Hyperparameter Transfercited by 272
- 2023Neural Architecture Search: Insights from 1000 Paperscited by 136
- 2021The Benchmark Lotterycited by 133
- 2021HPOBench: A Collection of Reproducible Multi-Fidelity Benchmark Problems for HPOcited by 111
- 2021How Powerful are Performance Predictors in Neural Architecture Search?cited by 88
- 2022Zeus: Understanding and Optimizing GPU Energy Consumption of DNN Trainingcited by 78
- 2022Efficient Methods for Natural Language Processing: A Surveycited by 78
- 2021YAHPO Gym -- An Efficient Multi-Objective Multi-Fidelity Benchmark for Hyperparameter Optimizationcited by 61
- 2021Model Zoo: A Growing "Brain" That Learns Continuallycited by 53
- 2022Model Zoos: A Dataset of Diverse Populations of Neural Network Modelscited by 38
- 2022Multi-Objective Hyperparameter Optimization in Machine Learning -- An Overviewcited by 37
12 most cited of 195 All 195 (JSON)
- 2026EdgeCraft: Automated Model Crafting for Edge IoT28 Sep 2026
- 2026Rank Collapse Is Recoverable, Growing |Q| Is Not: Out-of-Sample Early Warning for Value Divergence in High-UTD Soft Actor-Critic26 Sep 2026
- 2026Revisiting Multi-Object Tracking Baselines: Hyperparameter Optimization with Multi-Fidelity Greedy Coordinate Search10 Sep 2026
- 2026Neural Symbollic Regression Using Deep Learning and Sparse Modelling1 Sep 2026
- 2026Knowing When Not to Reuse: Conditional Experience Transfer in Autonomous LLM Post-Training27 Aug 2026
- 2026Steering Recurrent Reasoners at Inference Time with Readout Feedback25 Aug 2026
- 2026Differential Learning for Robust Prediction of Thermal Stability with Application to Energetic Materials24 Aug 2026
- 2026Candidate-Fate Accounting for Transparent Sensor Diagnostic Pipeline Search19 Aug 2026
- 2026A Convolutional Layer Activation Dimensionality Reduction for Out-of-Distribution and Adversarial Attack Detection Methods10 Aug 2026
- 2026Automated Discovery Has No Universally Superior Harness20 Jul 2026
- 2026BBOmix: A Tabular Benchmark for Hyperparameter Optimization of Unsupervised Biological Representation Learning3 Jun 2026
- 2026Two-Fidelity Best-Action Identification for Stochastic Minimax Tree1 Jun 2026
12 newest of 195 All 195 (JSON)