Blog / Bayesian Hyperparameter Optimization with Valohai
Bayesian Hyperparameter Optimization with Valohai

Bayesian Hyperparameter Optimization with Valohai

Magdalena Stenius

Grid search and random search are the most well-known in hyperparameter tuning. They are also both first-class citizens inside the Valohai platform. You define your search space, hit go, and Valohai will start all your machines. It does a search over the designated area of parameters you’ve defined. It is all automatic and doesn’t make you launch or shut down machines by hand. Also, you don't accidentally leave machines running costing you money. But we’ve been missing one central way for hyperparameter tuning, Bayesian optimization. Not anymore!

Bayesian Hyperparameter Optimisation in Valohai

Bayesian optimization is good for problems where the target function f(x) is a black box, and it’s expensive to evaluate it. This means that you don’t want to call it too many times in vain, and you can't use gradient descent based approaches. Bayesian starts by guessing a function g(x) such that it would fulfill the unknown function f(x). It then uses gaussian probability to guess an x value, evaluates f(x) based on it and updates g(x). This is then repeated several times until we are happy with g(x). If that was all too mathematical, all you need to know is that it’s a smarter way to find submarines than just wild shots in the dark . If you want more details about the concepts behind the Bayesian optimization, check out this article by Will Koehrsen!

To conduct a Bayesian search in Valohai, you create a new Task and select Bayesian optimization. Configure the desired amount of executions, set the batch size, and a target metric (e.g., loss). Valohai will then optimize towards the set target value of the said target metric.

Valohai’s UI for starting a Bayesian optimization

After the optimization is completed, you’ll find something like this:

Bayesian optimization results

Valohai’s Bayesian optimization process uses Hyperopt-library’s Tree Parzen Estimator implementation to pick the parameter’s from the previous execution as input for the next.

Under the hood, the optimization works in the following way:

  1. Create a batch of startup executions using random search
  2. Based on these executions, create a simplified function to model the relationship between the hyperparameters and the target metric value (for example "loss")
  3. Based on this simplification of their relationship, find the optimal values for the hyperparameter to make the target metric as close to the target value as possible
  4. Run the next batch of executions and repeat the process from step 2.

Using interactive hyperparameter optimization, you can make hyperparameter tuning faster and more efficient than for example using a random search or an exhaustive grid search.

Get a demo and learn how to use the Valohai platform to do Bayesian hyperparameter optimization for your project.

P.S. This blog post concentrates on Valohai’s capabilities in the Web UI. You can, however, use _any- external optimizer by hooking it up to Valohai’s API or CLI. Ask us more by signing up for a private demo !

Valohai trialSign up for a free trial on Valohai.Join the waitlist and get invited to see Valohai in action.