Skip to content

Latest commit

 

History

History
52 lines (40 loc) · 2.32 KB

File metadata and controls

52 lines (40 loc) · 2.32 KB

BootstrapNAS Jupyter Notebooks


BootstrapNAS Architecture

BootstrapNAS (1) takes a pre-trained model as input. (2) It uses this model to generate a weight-sharing super-network. (3) BootstrapNAS then applies a training strategy, and once the super-network has been trained, (4) it searches for efficient subnetworks that satisfy the user's requirements. (5) The configuration of the discovered sub-network(s) is returned to the user.

Quickstart

Please follow the instructions here.

If you already have a super-network trained with BootstrapNAS, please follow the instructions to search for sub-networks here.

More information about BootstrapNAS is available in our papers:

Automated Super-Network Generation for Scalable Neural Architecture Search.

  @inproceedings{
    munoz2022automated,
    title={Automated Super-Network Generation for Scalable Neural Architecture Search},
    author={Muñoz, J. Pablo and Lyalyushkin, Nikolay and Lacewell, Chaunte and Senina, Anastasia and Cummings, Daniel and Sarah, Anthony  and Kozlov, Alexander and Jain, Nilesh},
    booktitle={First Conference on Automated Machine Learning (Main Track)},
    year={2022},
    url={https://openreview.net/forum?id=HK-zmbTB8gq}
  }

Enabling NAS with Automated Super-Network Generation

@article{
  bootstrapNAS,
  author    = {Muñoz, J. Pablo  and Lyalyushkin, Nikolay  and Akhauri, Yash and Senina, Anastasia and Kozlov, Alexander  and Jain, Nilesh},
  title     = {Enabling NAS with Automated Super-Network Generation},
  journal   = {CoRR},
  volume    = {abs/2112.10878},
  year      = {2021},
  url       = {https://arxiv.org/abs/2112.10878},
  eprinttype = {arXiv},
  eprint    = {2112.10878},
  timestamp = {Tue, 04 Jan 2022 15:59:27 +0100},
  biburl    = {https://dblp.org/rec/journals/corr/abs-2112-10878.bib},
  bibsource = {dblp computer science bibliography, https://dblp.org}
}

Contributing to BootstrapNAS

Please follow the contribution guidelines in NNCF.