Code for Efficient Forward Neural Architecture Search, Neurips 2019!

Note this repo is under active development and the code base is expected to rapidly change. We are currently rewriting Petridish in Pytorch with evaluation on many more datasets and pretrained models. It will appear here shortly.

Petridishnn has adopted the Microsoft Open Source Code of Conduct. For more information on this code of conduct, see the Code of Conduct FAQ or contact [email protected] with any additional questions or comments. Read Microsoft’s statement on Privacy & Cookies

We have developed and tested Petridish on Ubuntu 16.04 LTS (64-bit), Anaconda python distribution and Tensorflow.

1. Install Anaconda python distribution for Ubuntu
2. Create a python 3.6 environment conda create python=3.6 -n py36
3. Follow instructions to install a recent Tensorflow (TF) version. 1.12 is tested.
4. Clone the repo: git clone petridishnn
5. Install dependency packages python -m pip install -r <path_to_petridishnn>/requirements.txt
6. Petridish needs some environment variables: GLOBAL_LOG_DIR: directory where logs will be written to by jobs running locally. GLOBAL_MODEL_DIR: directory where models will be written to by jobs running locally. GLOBAL_DATA_DIR: directory from where local jobs will read data. Set them to appropriate values in your bashrc. E.g. export GLOBAL_MODEL_DIR="/home/dedey/data"

Petridish code assumes datasets are in certain format (e.g. we transform ImageNet raw data to lmdb format). While one can always download the raw data of standard datasets and use the relevant scripts in petridishnn/petridish/data to convert them Debadeepta Dey [email protected] maintains an Azure blob with all the data in the converted format. (For Microsoft employees only) Please email him for access.

Before doing full scale search on Azure it is common to check everything is running on local machine. An example job script is at petridishnn/scripts/test_distributed.sh. Make sure you have all the environment variables used in this script. Run this from root folder of petridishn as bash scripts/test_distributed.sh. This will output somethings to stdout but will output models and logs to the corresponding folders. If this succeeds you have a working installation. Yay!

We provide a number of scripts to analyze and post-process the search results in the directory petridish/analysis. We also provide a script to generate training scripts to train the found models. We list them in the order of usage as follows. Please refer to the header of each linked file for usage.

1. Inspect the search log
2. Generate scripts to train found models
3. Check performance of model training

• Debadeepta Dey ([email protected])
• Hanzhang Hu ([email protected])
• John Langford ([email protected])
• Rich Caruana ([email protected])
• Eric Horvitz ([email protected])

Please read the contributing policy

If you would like to use this work for your research, please cite the following:

@article{hu2019forwardnas,
  title={Efficient Forward Architecture Search},
  author={Hanzhang Hu and John Langford and Rich Caruana and Saurajit Mukherjee and Eric Horvitz and Debadeepta Dey},
  journal={Neural Information Processing Systems},
  year={2019}
}