Sample complexity of data-driven tuning of model hyperparameters in neural networks with structured parameter-dependent dual function
Maria-Florina Balcan, Anh Nguyen, Dravyansh Sharma
Abstract
Modern machine learning algorithms, especially deep learning-based techniques, typically involve careful hyperparameter tuning to achieve the best performance. Despite the surge of intense interest in practical techniques like Bayesian optimization and random search-based approaches to automating this laborious and compute-intensive task, the fundamental learning-theoretic complexity of tuning hyperparameters for deep neural networks is poorly understood. Inspired by this glaring gap, we initiate the formal study of hyperparameter tuning complexity in deep learning through a recently introduced data-driven setting. We assume that we have a series of learning tasks, and we have to tune hyperparameters to do well on average over the distribution of tasks. A major difficulty is that the utility function as a function of the hyperparameter is very volatile and furthermore, it is given implicitly by an optimization problem over the model parameters. To tackle this challenge, we introduce a new technique to characterize the discontinuities and oscillations of the utility function on any fixed problem instance as we vary the hyperparameter; our analysis relies on subtle concepts including tools from algebraic geometry, differential geometry and constrained optimization. We use this to show that the learning-theoretic complexity of the corresponding family of utility functions is bounded. We instantiate our results and provide sample complexity bounds for concrete applications-tuning a hyperparameter that interpolates neural activation functions and setting the kernel parameter in graph neural networks.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Cited by top-tier papers3
- Generalization Guarantees for Learning Score-Based Branch-and-Cut Policies in Integer ProgrammingHongyu Cheng, Amitabh BasuNeurIPS 2025 · 6 citations
- Provably Data-driven Lagrangian Relaxation for Mixed Integer Linear ProgrammingQUOC TUNG LE, Anh Nguyen, Viet Anh NguyenICML 2026 · 2 citations
- Provably Data-driven Multiple Hyper-parameter Tuning with Structured Loss FunctionQUOC TUNG LE, Anh Nguyen, Viet Anh NguyenICML 2026 · 2 citations
Builds on18
- NAS-Bench-201: Extending the Scope of Reproducible Neural Architecture SearchXuanyi Dong, Yi YangICLR 2020 · 825 citations
- BANANAS: Bayesian Optimization with Neural Architectures for Neural Architecture SearchColin White, Willie Neiswanger, Yash SavaniAAAI 2021 · 401 citations
- Learning Algebraic Multigrid Using Graph Neural NetworksIlay Luz, Meirav Galun, Haggai Maron, Ronen Basri et al.ICML 2020 · 95 citations
- Geometry-Aware Gradient Algorithms for Neural Architecture SearchLiam Li, Mikhail Khodak, Nina Balcan, Ameet TalwalkarICLR 2021 · 73 citations
- Sample Complexity of Tree Search Configuration: Cutting Planes and BeyondMaria-Florina Balcan, Siddharth Prasad, Tuomas Sandholm, Ellen VitercikNeurIPS 2021 · 54 citations
Related papers
- Stochastic Marginal Likelihood Gradients using Neural Tangent KernelsAlexander Immer, Tycho F. A. van der Ouderaa, Mark van der Wilk, Gunnar Rätsch et al.ICML 2023 · 17 citations
- Why neural networks find simple solutions: The many regularizers of geometric complexityBenoit Dherin, Michael Munn, Mihaela Rosca, David BarrettNeurIPS 2022 · 52 citations
- The Statistical Cost of Robust Kernel Hyperparameter TurningRaphael A. Meyer, Christopher MuscoNeurIPS 2020 · 2 citations
- Global Optimization with Parametric Function ApproximationChong Liu, Yu-Xiang WangICML 2023 · 10 citations
- Supervising the Multi-Fidelity Race of Hyperparameter ConfigurationsMartin Wistuba, Arlind Kadra, Josif GrabockaNeurIPS 2022 · 24 citations
