HowToChooseTuner.md 11.7 KB
Newer Older
QuanluZhang's avatar
QuanluZhang committed
1
# How to use Tuner that NNI supports?
Deshui Yu's avatar
Deshui Yu committed
2

fishyds's avatar
fishyds committed
3
For now, NNI has supported the following tuner algorithms. Note that NNI installation only installs a subset of those algorithms, other algorithms should be installed through `nnictl package install` before you use them. For example, for SMAC the installation command is `nnictl package install --name=SMAC`.
Deshui Yu's avatar
Deshui Yu committed
4

QuanluZhang's avatar
QuanluZhang committed
5
6
7
8
9
10
11
12
 - [TPE](#TPE)
 - [Random Search](#Random)
 - [Anneal](#Anneal)
 - [Naive Evolution](#Evolution)
 - [SMAC](#SMAC) (to install through `nnictl`)
 - [Batch Tuner](#Batch)
 - [Grid Search](#Grid)
 - [Hyperband](#Hyperband)
Lee's avatar
Lee committed
13
 - [Network Morphism](#NetworkMorphism)
Deshui Yu's avatar
Deshui Yu committed
14

QuanluZhang's avatar
QuanluZhang committed
15
 ## Supported tuner algorithms
Deshui Yu's avatar
Deshui Yu committed
16
17


QuanluZhang's avatar
QuanluZhang committed
18
We will introduce some basic knowledge about the tuner algorithms, suggested scenarios for each tuner, and their example usage (for complete usage spec, please refer to [here]()).
Deshui Yu's avatar
Deshui Yu committed
19

QuanluZhang's avatar
QuanluZhang committed
20
<a name="TPE"></a>
QuanluZhang's avatar
QuanluZhang committed
21
**TPE**
Deshui Yu's avatar
Deshui Yu committed
22

QuanluZhang's avatar
QuanluZhang committed
23
24
25
The Tree-structured Parzen Estimator (TPE) is a sequential model-based optimization (SMBO) approach. SMBO methods sequentially construct models to approximate the performance of hyperparameters based on historical measurements, and then subsequently choose new hyperparameters to test based on this model.
The TPE approach models P(x|y) and P(y) where x represents hyperparameters and y the associated evaluate matric. P(x|y) is modeled by transforming the generative process of hyperparameters, replacing the distributions of the configuration prior with non-parametric densities. 
This optimization approach is described in detail in [Algorithms for Hyper-Parameter Optimization][1].
Lee's avatar
Lee committed
26

QuanluZhang's avatar
QuanluZhang committed
27
28
29
_Suggested scenario_: TPE, as a black-box optimization, can be used in various scenarios, and shows good performance in general. Especially when you have limited computation resource and can only try a small number of trials. From a large amount of experiments, we could found that TPE is far better than Random Search.

_Usage_:
Lee's avatar
Lee committed
30
```yaml
QuanluZhang's avatar
QuanluZhang committed
31
32
33
34
35
36
37
38
39
  # config.yaml
  tuner:
    builtinTunerName: TPE
    classArgs:
      # choice: maximize, minimize
      optimize_mode: maximize
```

<a name="Random"></a>
QuanluZhang's avatar
QuanluZhang committed
40
**Random Search**
Deshui Yu's avatar
Deshui Yu committed
41

42
In [Random Search for Hyper-Parameter Optimization][2] show that Random Search might be surprisingly simple and effective. We suggests that we could use Random Search as baseline when we have no knowledge about the prior distribution of hyper-parameters.
QuanluZhang's avatar
QuanluZhang committed
43

xuehui's avatar
xuehui committed
44
_Suggested scenario_: Random search is suggested when each trial does not take too long (e.g., each trial can be completed very soon, or early stopped by assessor quickly), and you have enough computation resource. Or you want to uniformly explore the search space. Random Search could be considered as baseline of search algorithm.
QuanluZhang's avatar
QuanluZhang committed
45
46

_Usage_:
Lee's avatar
Lee committed
47
```yaml
QuanluZhang's avatar
QuanluZhang committed
48
49
50
51
52
53
  # config.yaml
  tuner:
    builtinTunerName: Random
```

<a name="Anneal"></a>
QuanluZhang's avatar
QuanluZhang committed
54
**Anneal**
QuanluZhang's avatar
QuanluZhang committed
55

xuehui's avatar
xuehui committed
56
This simple annealing algorithm begins by sampling from the prior, but tends over time to sample from points closer and closer to the best ones observed. This algorithm is a simple variation on random search that leverages smoothness in the response surface. The annealing rate is not adaptive.
QuanluZhang's avatar
QuanluZhang committed
57

xuehui's avatar
xuehui committed
58
_Suggested scenario_: Anneal is suggested when each trial does not take too long, and you have enough computation resource(almost same with Random Search). Or the variables in search space could be sample from some prior distribution.
QuanluZhang's avatar
QuanluZhang committed
59
60

_Usage_:
Lee's avatar
Lee committed
61
```yaml
xuehui's avatar
xuehui committed
62
63
64
65
66
67
68
  # config.yaml
  tuner:
    builtinTunerName: Anneal
    classArgs:
      # choice: maximize, minimize
      optimize_mode: maximize
```
QuanluZhang's avatar
QuanluZhang committed
69
70

<a name="Evolution"></a>
QuanluZhang's avatar
QuanluZhang committed
71
72
**Naive Evolution**

QuanluZhang's avatar
QuanluZhang committed
73
74
75
Naive Evolution comes from [Large-Scale Evolution of Image Classifiers][3]. It randomly initializes a population based on search space. For each generation, it chooses better ones and do some mutation (e.g., change a hyperparameter, add/remove one layer) on them to get the next generation. Naive Evolution requires many trials to works, but it's very simple and easily to expand new features.

_Suggested scenario_: Its requirement of computation resource is relatively high. Specifically, it requires large inital population to avoid falling into local optimum. If your trial is short or leverages assessor, this tuner is a good choice. And, it is more suggested when your trial code supports weight transfer, that is, the trial could inherit the converged weights from its parent(s). This can greatly speed up the training progress.
Deshui Yu's avatar
Deshui Yu committed
76

QuanluZhang's avatar
QuanluZhang committed
77
_Usage_:
Lee's avatar
Lee committed
78
```yaml
QuanluZhang's avatar
QuanluZhang committed
79
80
81
82
83
84
85
  # config.yaml
  tuner:
    builtinTunerName: Evolution
    classArgs:
      # choice: maximize, minimize
      optimize_mode: maximize
```
Deshui Yu's avatar
Deshui Yu committed
86

QuanluZhang's avatar
QuanluZhang committed
87
<a name="SMAC"></a>
QuanluZhang's avatar
QuanluZhang committed
88
89
**SMAC**

QuanluZhang's avatar
QuanluZhang committed
90
91
[SMAC][4] is based on Sequential Model-Based Optimization (SMBO). It adapts the most prominent previously used model class (Gaussian stochastic process models) and introduces the model class of random forests to SMBO, in order to handle categorical parameters. The SMAC supported by nni is a wrapper on [the SMAC3 github repo][5].

xuehui's avatar
xuehui committed
92
Note that SMAC on nni only supports a subset of the types in [search space spec](./SearchSpaceSpec.md), including `choice`, `randint`, `uniform`, `loguniform`, `quniform(q=1)`.
QuanluZhang's avatar
QuanluZhang committed
93

94
95
96
97
_Installation_: 
* Install swig first. (`sudo apt-get install swig` for Ubuntu users)
* Run `nnictl package install --name=SMAC`

QuanluZhang's avatar
QuanluZhang committed
98
99
100
_Suggested scenario_: Similar to TPE, SMAC is also a black-box tuner which can be tried in various scenarios, and is suggested when computation resource is limited. It is optimized for discrete hyperparameters, thus, suggested when most of your hyperparameters are discrete.

_Usage_:
Lee's avatar
Lee committed
101
```yaml
QuanluZhang's avatar
QuanluZhang committed
102
103
104
105
106
107
108
  # config.yaml
  tuner:
    builtinTunerName: SMAC
    classArgs:
      # choice: maximize, minimize
      optimize_mode: maximize
```
QuanluZhang's avatar
QuanluZhang committed
109

QuanluZhang's avatar
QuanluZhang committed
110
111
<a name="Batch"></a>
**Batch tuner**
QuanluZhang's avatar
QuanluZhang committed
112

xuehui's avatar
xuehui committed
113
Batch tuner allows users to simply provide several configurations (i.e., choices of hyper-parameters) for their trial code. After finishing all the configurations, the experiment is done. Batch tuner only supports the type `choice` in [search space spec](./SearchSpaceSpec.md).
QuanluZhang's avatar
QuanluZhang committed
114

QuanluZhang's avatar
QuanluZhang committed
115
_Suggested sceanrio_: If the configurations you want to try have been decided, you can list them in searchspace file (using `choice`) and run them using batch tuner.
QuanluZhang's avatar
QuanluZhang committed
116

QuanluZhang's avatar
QuanluZhang committed
117
_Usage_:
Lee's avatar
Lee committed
118
```yaml
QuanluZhang's avatar
QuanluZhang committed
119
120
121
122
  # config.yaml
  tuner:
    builtinTunerName: BatchTuner
```
Zejun Lin's avatar
Zejun Lin committed
123

xuehui's avatar
xuehui committed
124
Note that the search space that BatchTuner supported like:
Lee's avatar
Lee committed
125
```json
xuehui's avatar
xuehui committed
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
{
    "combine_params":
    {
        "_type" : "choice",
        "_value" : [{"optimizer": "Adam", "learning_rate": 0.00001},
                    {"optimizer": "Adam", "learning_rate": 0.0001},
                    {"optimizer": "Adam", "learning_rate": 0.001},
                    {"optimizer": "SGD", "learning_rate": 0.01}, 
                    {"optimizer": "SGD", "learning_rate": 0.005}, 
                    {"optimizer": "SGD", "learning_rate": 0.0002}]
    }
}
```
The search space file including the high-level key `combine_params`. The type of params in search space must be `choice` and the `values` including all the combined-params value.

QuanluZhang's avatar
QuanluZhang committed
141
142
<a name="Grid"></a>
**Grid Search**
Zejun Lin's avatar
Zejun Lin committed
143

QuanluZhang's avatar
QuanluZhang committed
144
Grid Search performs an exhaustive searching through a manually specified subset of the hyperparameter space defined in the searchspace file.
xuehui's avatar
xuehui committed
145
Note that the only acceptable types of search space are `choice`, `quniform`, `qloguniform`. **The number `q` in `quniform` and `qloguniform` has special meaning (different from the spec in [search space spec](./SearchSpaceSpec.md)). It means the number of values that will be sampled evenly from the range `low` and `high`.**
Zejun Lin's avatar
Zejun Lin committed
146

QuanluZhang's avatar
QuanluZhang committed
147
_Suggested scenario_: It is suggested when search space is small, it is feasible to exhaustively sweeping the whole search space.
Deshui Yu's avatar
Deshui Yu committed
148

QuanluZhang's avatar
QuanluZhang committed
149
_Usage_:
Lee's avatar
Lee committed
150
```yaml
QuanluZhang's avatar
QuanluZhang committed
151
152
153
154
155
156
  # config.yaml
  tuner:
    builtinTunerName: GridSearch
```

<a name="Hyperband"></a>
QuanluZhang's avatar
QuanluZhang committed
157
158
**Hyperband**

Lee's avatar
Lee committed
159
[Hyperband][6] tries to use limited resource to explore as many configurations as possible, and finds out the promising ones to get the final result. The basic idea is generating many configurations and to run them for small number of STEPs to find out promising one, then further training those promising ones to select several more promising one. More detail can be referred to [here](../src/sdk/pynni/nni/hyperband_advisor/README.md).
QuanluZhang's avatar
QuanluZhang committed
160

QuanluZhang's avatar
QuanluZhang committed
161
162
163
_Suggested scenario_: It is suggested when you have limited computation resource but have relatively large search space. It performs good in the scenario that intermediate result (e.g., accuracy) can reflect good or bad of final result (e.g., accuracy) to some extent.

_Usage_:
Lee's avatar
Lee committed
164
```yaml
QuanluZhang's avatar
QuanluZhang committed
165
166
167
168
169
170
171
172
173
174
175
176
177
  # config.yaml
  advisor:
    builtinAdvisorName: Hyperband
    classArgs:
      # choice: maximize, minimize
      optimize_mode: maximize
      # R: the maximum STEPS (could be the number of mini-batches or epochs) can be
      #    allocated to a trial. Each trial should use STEPS to control how long it runs.
      R: 60
      # eta: proportion of discarded trials
      eta: 3
```

Lee's avatar
Lee committed
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
<a name="NetworkMorphism"></a>
**Network Morphism**

[Network Morphism](7) provides functions to automatically search for architecture of deep learning models. Every child network inherits the knowledge from its parent network and morphs into diverse types of networks, including changes of depth, width and skip-connection. Next, it estimates the value of child network using the history architecture and metric pairs. Then it selects the most promising one to train. More detail can be referred to [here](../src/sdk/pynni/nni/networkmorphism_tuner/README.md). 

_Suggested scenario_: It is suggested that you want to apply deep learning methods to your task (your own dataset) but you have no idea of how to choose or design a network. You modify the [example](../examples/trials/network_morphism/cifar10/cifar10_keras.py) to fit your own dataset and your own data augmentation method. Also you can change the batch size, learning rate or optimizer. It is feasible for different tasks to find a good network architecture. Now this tuner only supports the cv domain.

_Usage_:
```yaml
  # config.yaml
  tuner:
    builtinTunerName: NetworkMorphism
    classArgs:
      #choice: maximize, minimize
      optimize_mode: maximize
      #for now, this tuner only supports cv domain
      task: cv
      #input image width
      input_width: 32
      #input image channel
      input_channel: 3
      #number of classes
      n_output_node: 10
```

QuanluZhang's avatar
QuanluZhang committed
203
204

# How to use Assessor that NNI supports?
Deshui Yu's avatar
Deshui Yu committed
205

QuanluZhang's avatar
QuanluZhang committed
206
For now, NNI has supported the following assessor algorithms.
Deshui Yu's avatar
Deshui Yu committed
207

QuanluZhang's avatar
QuanluZhang committed
208
209
 - [Medianstop](#Medianstop)
 - Curve Extrapolation (ongoing)
Deshui Yu's avatar
Deshui Yu committed
210

QuanluZhang's avatar
QuanluZhang committed
211
## Supported Assessor Algorithms
Deshui Yu's avatar
Deshui Yu committed
212

QuanluZhang's avatar
QuanluZhang committed
213
214
<a name="Medianstop"></a>
**Medianstop**
Deshui Yu's avatar
Deshui Yu committed
215

Lee's avatar
Lee committed
216
Medianstop is a simple early stopping rule mentioned in the [paper][8]. It stops a pending trial X at step S if the trial’s best objective value by step S is strictly worse than the median value of the running averages of all completed trials’ objectives reported up to step S.
Deshui Yu's avatar
Deshui Yu committed
217

QuanluZhang's avatar
QuanluZhang committed
218
_Suggested scenario_: It is applicable in a wide range of performance curves, thus, can be used in various scenarios to speed up the tuning progress.
Deshui Yu's avatar
Deshui Yu committed
219

QuanluZhang's avatar
QuanluZhang committed
220
_Usage_:
Lee's avatar
Lee committed
221
```yaml
QuanluZhang's avatar
QuanluZhang committed
222
223
224
225
226
227
228
229
230
231
  assessor:
    builtinAssessorName: Medianstop
    classArgs:
      #choice: maximize, minimize
      optimize_mode: maximize
      # (optional) A trial is determined to be stopped or not, 
      * only after receiving start_step number of reported intermediate results.
      * The default value of start_step is 0.
      start_step: 5
```
Deshui Yu's avatar
Deshui Yu committed
232

Lee's avatar
Lee committed
233
234
235
236
237
238
239
240
[1]: https://papers.nips.cc/paper/4443-algorithms-for-hyper-parameter-optimization.pdf
[2]: http://www.jmlr.org/papers/volume13/bergstra12a/bergstra12a.pdf
[3]: https://arxiv.org/pdf/1703.01041.pdf
[4]: https://www.cs.ubc.ca/~hutter/papers/10-TR-SMAC.pdf
[5]: https://github.com/automl/SMAC3
[6]: https://arxiv.org/pdf/1603.06560.pdf
[7]: https://arxiv.org/abs/1806.10282
[8]: https://static.googleusercontent.com/media/research.google.com/en//pubs/archive/46180.pdf