Hyperparameter Search: Adaptive (Asynchronous)¶
adaptive_asha search method employs an
asynchronous version of successive halving (ASHA), which is suitable for
large-scale experiments with hundreds or thousands of trials.
Here are some suggested initial settings for
typically work well.
mode: Set to
max_length: The maximum training length (see Training Units) of any trial that survives to the end of the experiment. This quantity is domain-specific and should roughly reflect the number of minibatches the model must be trained on for it to converge on the data set. For users who would like to determine this number experimentally, train a model with reasonable hyperparameters using the
max_trials: This indicates the total number of hyperparameter settings that will be evaluated in the experiment. Set
max_trialsto at least 500 to take advantage of speedups from early-stopping. You can also set a large
max_trialsand stop the experiment once the desired performance is achieved.
max_concurrent_trials: This field controls the degree of parallelism of the experiment. The experiment will have a maximum of this many trials training simultaneously at any one time. The
adaptive_ashasearcher scales nearly perfectly with additional compute, so you should set this field based on compute environment constraints. If this value is less than the number of brackets produced by the adaptive algorithm, it will be rounded up.
adaptive_asha searcher is a carefully tuned
strategy for spawning multiple ASHA (asynchronous successive halving
algorithm) searchers, themselves hyperparameter search algorithms. ASHA
can be configured to make different tradeoffs between exploration and
exploitation, i.e., how many trials are explored versus how long a
single trial is trained for. Because the right tradeoff between
exploration and exploitation is hard to know in advance, the
adaptive_asha algorithm tries several ASHA searches with different
The configuration settings available to Determined experiments running
adaptive_asha mode mostly affect the ASHA subroutines directly.
mode configuration is the only one affecting the decisions of
adaptive_asha searcher, by changing the number and types of ASHA
The first section here gives a description of the synchronous version of
ASHA called successive halving. The second section discusses the
motivation for the asynchronous promotions used by ASHA. The third
section describes why you would choose adaptive_asha over plain
asynchronous_halving. The final section and conclusion is a set of FAQs
At a high level, SHA prunes (“halves”) a set of trials in successive rounds we call rungs. SHA starts with an initial set of trials. (A trial means one model, with a fixed set of hyperparameter values.) SHA trains all the trials for some length and the trials with the worst validation performance are discarded. In the next rung, the remaining trials are trained for a longer period of time, and then trials with the worst validation performance are pruned once again. This is repeated until the maximum training length is reached.
First, an example of SHA.
Rung 1: SHA creates N initial trials; the hyperparameter values for each trial are randomly sampled from the hyperparameters defined in the experiment configuration file. Each trial is trained for 1 epoch, and then validation metrics are computed.
Rung 2: SHA picks the N/4 top-performing trials according to validation metrics. These are trained for 4 epochs.
Rung 3: SHA picks the N/16 top-performing trials according to validation metrics. These are trained for 16 epochs.
At the end, the trial with best performance has the hyperparameter setting the SHA searcher returns.
In the example above, we generalize “halving” with a field called
divisor, which determines what fraction of trials are kept in successive
rungs, as well as the training length in successive rungs.
max_length is 16 epochs, which is the maximum length a trial is
In general, SHA has a fixed
divisor d. In the first rung, it
generates an initial set of randomly chosen trials and runs until each
trial has trained for the same length. In the next rung, it keeps 1/d of
those trials and closes the rest. Then it runs each remaining trial
until it has trained for d times as long as the previous rung. ASHA
iterates this process until some stopping criterion is reached, such as
completing a specified number of rungs or having only one trial
remaining. The total training length, rungs, and trials within rungs are
fixed within each SHA searcher, but vary across different calls to SHA
by the adaptive algorithm. Note that although the name “SHA” includes
the phrase “halving”, the fraction of trials pruned after every rung is
Why Asynchronous Halving?¶
Successive halving (SHA) promotes hyperparameter configurations synchronously, waiting for each rung to complete before performing any promotions. This allows the algorithm to have complete information about all trials at the time of promotion, but it results in underutilized nodes waiting on completion of validation steps for other configurations. ASHA, asynchronous successive halving, asynchronously promotes trials when it has the minimum information required to make a decision in order to maximize compute efficiency of the searcher. In contrast to SHA which initializes all trials in the bottom rung at the outset, ASHA will continuously add trials to the bottom rung until the desired number of trials is reached.
See the difference in asynchronous vs. sychronous promotions in the two animated GIFs below:
Adaptive over ASHA¶
The adaptive algorithm calls ASHA subroutines with varying parameters.
The exact calls are configured through the choice of
specifies how aggressively to perform early stopping. One way to think
about this behavior is as a spectrum that ranges from “one ASHA run”
(aggressive early stopping; eliminate most trials every rung) to
searcher: random” “multiple ASHA runs, some of which will not early
stop and others will early stop later” (try some without early stopping;
initialized trials may be allowed to run to completion).
On one end,
aggressive applies early stopping in a very eager
manner; this mode essentially corresponds to only making a single call
to ASHA. With the default
divisor of 4, 75% of the remaining trials
will be eliminated in each rung after only being trained for 25% the
length of the next rung. This implies that relatively few trials will be
allowed to finish even a small fraction of the length needed train to
max_length). This aggressive early stopping behavior
allows the searcher to start more trials for a wider exploration of
hyperparameter configurations, at the risk of discarding a configuration
On the other end,
conservative mode is more similar to a
search, in that it performs significantly less pruning. Extra ASHA
subroutines are spawned with fewer rungs and longer training lengths to
account for the high percentage of trials eliminated after only a short
time. However, a
conservative adaptive search will only explore a
small fraction of the configurations explored by an
search, given the same budget.
Once the number and types of calls to ASHA are determined (via
mode), the adaptive algorithm will allocate training length budgets
to the ASHA subroutines, from the overall budget for the adaptive
algorithm (user-specified through
budget). This determines the
number of trials at each rung (N in the above ASHA example).
Q: How do I control how long a trial is trained for before it is potentially discarded?
The training length is guaranteed to be at least
max_length / 256 by
max_length / divisor ^ max_rungs-1 in general. It is
recommended to configure this in records or epochs if the
global_batch_size hyperparameter is not constant, to ensure each
trial trains on the same amount of data.
Q: How do I make sure ``x`` trials are run the full training length (``max_length``)?
The number of initial trials is determined by a combination of
bracket_rungs. Here is a rule of thumb for the default configuration
max_rungs: 5 and
divisor: 4, with
mode: standard and a
The initial number of trials is
To ensure that
xtrials are run
max_trialshigh enough for the brackets with their halving rate (the
divisor) to allow
xtrials to make it to the final
rungs. This can be viewed by the command describe below.
A configuration setting that meets set goals can be found by trial and error. The command
det preview-search <file_name.yaml>
will display information on the number of trials versus training length
for the configuration specified in
Q: The adaptive algorithm sounds great so far. What are its weaknesses?
In our experience, early-stopping works well across a variety of deep learning models. However, there may be some search spaces in which early-stopping underperforms simple random search. This can happen if model complexity varies drastically in a search space leading to different converge rates or if the search space contains hyperparameters that are strongly correlated with training length.