MLAILGJan 19, 2024

A Unified Gaussian Process for Branching and Nested Hyperparameter Optimization

arXiv:2402.04885v1
AI Analysis

This addresses a practical limitation in hyperparameter tuning for machine learning practitioners, offering an incremental improvement over existing methods by handling dependent parameters.

The paper tackles the problem of hyperparameter optimization for neural networks when parameters have conditional dependencies, such as branching and nested structures, by proposing a unified Bayesian optimization framework based on a new Gaussian process kernel. The result shows higher prediction accuracy and better optimization efficiency in synthetic simulations and real data applications.

Choosing appropriate hyperparameters plays a crucial role in the success of neural networks as hyper-parameters directly control the behavior and performance of the training algorithms. To obtain efficient tuning, Bayesian optimization methods based on Gaussian process (GP) models are widely used. Despite numerous applications of Bayesian optimization in deep learning, the existing methodologies are developed based on a convenient but restrictive assumption that the tuning parameters are independent of each other. However, tuning parameters with conditional dependence are common in practice. In this paper, we focus on two types of them: branching and nested parameters. Nested parameters refer to those tuning parameters that exist only within a particular setting of another tuning parameter, and a parameter within which other parameters are nested is called a branching parameter. To capture the conditional dependence between branching and nested parameters, a unified Bayesian optimization framework is proposed. The sufficient conditions are rigorously derived to guarantee the validity of the kernel function, and the asymptotic convergence of the proposed optimization framework is proven under the continuum-armed-bandit setting. Based on the new GP model, which accounts for the dependent structure among input variables through a new kernel function, higher prediction accuracy and better optimization efficiency are observed in a series of synthetic simulations and real data applications of neural networks. Sensitivity analysis is also performed to provide insights into how changes in hyperparameter values affect prediction accuracy.

Foundations

The foundational work for this paper's niche, ranked by how specifically the neighbourhood builds on it — not by global fame.

Your Notes