Feasible-First Exploration for Constrained ML Deployment Optimization in Crash-Prone Hierarchical Search Spaces
Not on the map. Why?
Abstract (arXiv)
Deploying machine learning models under production constraints requires joint optimization over model family, quantization scheme, runtime backend, and serving configuration. This induces a hierarchical mixed-variable search space in which many configurations are invalid: evaluations may crash, exceed memory limits, or violate latency constraints. Standard black-box optimizers such as Tree-structured Parzen Estimators (TPE) and constrained Bayesian optimization are effective when valid configurations are common, but they can spend a large fraction of a small evaluation budget on invalid or uninformative trials in hostile deployment spaces. This paper studies that regime and asks whether optimization should be decomposed into an explicit exploration stage followed by model-guided exploitation. We propose Thermal Budget Annealing (TBA), a feasible-first exploration procedure that maps valid and feasible regions before warm-starting TPE. The method includes two robustness mechanisms for hostile hardware: trial timeouts that abort clearly infeasible evaluations early, and subspace blacklisting that temporarily suppresses categorical subspaces after repeated failures. We also introduce DeployBench, a benchmark suite for deployment optimization with hierarchical structure, hidden crash zones, hard constraints, and unequal evaluation costs. On synthetic benchmarks and real GPU deployment with five pre-trained vision models across five GPU targets (NVIDIA H100, A100, RTX 5080, L4, and T4), the proposed hybrid improves model-family discovery under tight constraints while reducing wasted budget relative to cold-start TPE.
Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 14 of its 37 references that are in the map. First posted on arXiv 27 Apr 2026.
Builds on
14 of 37 references.
References · 14
- 2019Optuna: A Next-generation Hyperparameter Optimization Frameworkcited by 2,501
- 2016Hyperband: A Novel Bandit-Based Approach to Hyperparameter Optimizationcited by 699
- 2018BOHB: Robust and Efficient Hyperparameter Optimization at Scalecited by 470
- 2021SMAC3: A Versatile Bayesian Optimization Package for Hyperparameter Optimizationcited by 230
- 2020Ansor: Generating High-Performance Tensor Programs for Deep Learningcited by 201
- 2019MLPerf Inference Benchmarkcited by 199
- 2024Vidur: A Large-Scale Simulation Framework For LLM Inferencecited by 117
- 2021HPOBench: A Collection of Reproducible Multi-Fidelity Benchmark Problems for HPOcited by 112
- 2014Bayesian Optimization with Unknown Constraintscited by 109
- 2021YAHPO Gym -- An Efficient Multi-Objective Multi-Fidelity Benchmark for Hyperparameter Optimizationcited by 62
- 2015A General Framework for Constrained Bayesian Optimization using Information-based Searchcited by 50
- 2022Edge Impulse: An MLOps Platform for Tiny Machine Learningcited by 24
Built on by
Not cited yet.
Similar, not cited
Close in wording, no citation either way
- 2026Constraint-Driven Model Optimization: An Industry Framework for Selecting Compression and Acceleration Techniques in Modern Machine Learning Systems
- 2026Optimizing AI Inference Across the Deployment Stack
- 2025Optimizing the Unknown: Black Box Bayesian Optimization with Energy-Based Model and Reinforcement Learning
- 2025Reinforcement Learning for Machine Learning Model Deployment: Evaluating Multi-Armed Bandits in ML Ops Environments
- 2026HARBOR: Automated Harness Optimization
- 2026StarOR: Synergizing Tree Search and Test-Time Reinforcement Learning for Optimization Modeling
- 2026SLO-Guard: Crash-Aware, Budget-Consistent Autotuning for SLO-Constrained LLM Serving
- 2026In-Context Batch Black-box Optimization via Flow Matching