Parameter Optimization with Conscious Allocation (POCA)

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: Inman, Joshua, Khandait, Tanmay, Pedrielli, Giulia, Sankar, Lalitha
Formato: Preprint
Publicado: 2023
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866911743274385408
author Inman, Joshua
Khandait, Tanmay
Pedrielli, Giulia
Sankar, Lalitha
author_facet Inman, Joshua
Khandait, Tanmay
Pedrielli, Giulia
Sankar, Lalitha
contents The performance of modern machine learning algorithms depends upon the selection of a set of hyperparameters. Common examples of hyperparameters are learning rate and the number of layers in a dense neural network. Auto-ML is a branch of optimization that has produced important contributions in this area. Within Auto-ML, hyperband-based approaches, which eliminate poorly-performing configurations after evaluating them at low budgets, are among the most effective. However, the performance of these algorithms strongly depends on how effectively they allocate the computational budget to various hyperparameter configurations. We present the new Parameter Optimization with Conscious Allocation (POCA), a hyperband-based algorithm that adaptively allocates the inputted budget to the hyperparameter configurations it generates following a Bayesian sampling scheme. We compare POCA to its nearest competitor at optimizing the hyperparameters of an artificial toy function and a deep neural network and find that POCA finds strong configurations faster in both settings.
format Preprint
id arxiv_https___arxiv_org_abs_2312_17404
institution arXiv
publishDate 2023
record_format arxiv
spellingShingle Parameter Optimization with Conscious Allocation (POCA)
Inman, Joshua
Khandait, Tanmay
Pedrielli, Giulia
Sankar, Lalitha
Machine Learning
The performance of modern machine learning algorithms depends upon the selection of a set of hyperparameters. Common examples of hyperparameters are learning rate and the number of layers in a dense neural network. Auto-ML is a branch of optimization that has produced important contributions in this area. Within Auto-ML, hyperband-based approaches, which eliminate poorly-performing configurations after evaluating them at low budgets, are among the most effective. However, the performance of these algorithms strongly depends on how effectively they allocate the computational budget to various hyperparameter configurations. We present the new Parameter Optimization with Conscious Allocation (POCA), a hyperband-based algorithm that adaptively allocates the inputted budget to the hyperparameter configurations it generates following a Bayesian sampling scheme. We compare POCA to its nearest competitor at optimizing the hyperparameters of an artificial toy function and a deep neural network and find that POCA finds strong configurations faster in both settings.
title Parameter Optimization with Conscious Allocation (POCA)
topic Machine Learning
url https://arxiv.org/abs/2312.17404