nas.rst 1.97 KB
Newer Older
1
2
3
#############################################
Retiarii for Neural Architecture Search (NAS)
#############################################
4
5

Automatic neural architecture search is taking an increasingly important role on finding better models.
QuanluZhang's avatar
QuanluZhang committed
6
7
Recent research works have proved the feasibility of automatic NAS, and also found some models that could beat manually tuned models.
Some of representative works are NASNet, ENAS, DARTS, Network Morphism, and Evolution. Moreover, new innovations keep emerging.
8

QuanluZhang's avatar
QuanluZhang committed
9
However, it takes great efforts to implement NAS algorithms, and it is hard to reuse code base of existing algorithms in a new one.
10
11
12
To facilitate NAS innovations (e.g., design and implement new NAS models, compare different NAS models side-by-side),
an easy-to-use and flexible programming interface is crucial.

13
14
15
16
17
18
19
20
21
22
Thus, we design `Retiarii <https://www.usenix.org/system/files/osdi20-zhang_quanlu.pdf>`__. It is a deep learning framework that supports the exploratory training on a neural network model space, rather than on a single neural network model.
Exploratory training with Retiarii allows user to express various search spaces for *Neural Architecture Search* and *Hyper-Parameter Tuning* with high flexibility.

Some frequently used terminologies in this document:

* *Model search space*: it means a set of models from which the best model is explored/searched. Sometimes we use *search space* or *model space* in short.
* *Exploration strategy*: the algorithm that is used to explore a model search space.
* *Model evaluator*: it is used to train a model and evaluate the model's performance.

Follow the instructions below to start your journey with Retiarii.
23
24

..  toctree::
QuanluZhang's avatar
QuanluZhang committed
25
26
    :maxdepth: 2

27
    Overview <NAS/Overview>
28
29
30
    Quick Start <NAS/QuickStart>
    Construct Model Space <NAS/construct_space>
    Multi-trial NAS <NAS/multi_trial_nas>
31
    One-shot NAS <NAS/one_shot_nas>
32
    Hardware-aware NAS <NAS/HardwareAwareNAS>
Yuge Zhang's avatar
Yuge Zhang committed
33
    NAS Benchmarks <NAS/Benchmarks>
34
    NAS API References <NAS/ApiReference>