Multi-step Planning for Automated Hyperparameter Optimization with OptFormer

Lucio M. Dery, Abram L. Friesen, Nando de Freitas, MarcAurelio Ranzato, Yutian Chen

Keywords: planning, hyperparameter optimization, OptFormer, autoregressive generation, rollouts

TL;DR: We introduced a planning-based hyperparameter optimization (HPO) method with the OptFormer model. We leverage OptFormer's ability to easily perform efficient rollouts as autoregressive generation to overcome barriers common in non-myopic HPO methods

Abstract: As machine learning permeates more industries and models become more expensive and time consuming to train, the need for efficient automated hyperparameter optimization (HPO) has never been more pressing. Multi-step planning based approaches to hyperparameter optimization promise improved efficiency over myopic alternatives by more effectively balancing out exploration and exploitation. However, the potential of these approaches has not been fully realized due to their technical complexity and computational intensity. In this work, we leverage recent advances in Transformer-based, natural-language-interfaced hyperparameter optimization to circumvent these barriers. We build on top of the recently proposed OptFormer which casts both hyperparameter suggestion and target function approximation as autoregressive generation thus making planning via rollouts simple and efficient. We conduct extensive exploration of different strategies for performing multi-step planning on top of the OptFormer model to highlight its potential for use in constructing non-myopic HPO strategies.