Skip to content

OFA²: Train one network, Search once, Deploy in many scenarios

License

Notifications You must be signed in to change notification settings

ito-rafael/once-for-all-2

 
 

Repository files navigation

OFA²: A Multi-Objective Perspective for the Once-for-All Neural Architecture Search [arXiv]

Search Once, Get Many

@misc{ito2023ofa2,
      title={OFA$^2$: A Multi-Objective Perspective for the Once-for-All Neural Architecture Search}, 
      author={Rafael C. Ito and Fernando J. Von Zuben},
      year={2023},
      eprint={2303.13683},
      archivePrefix={arXiv},
      primaryClass={cs.NE}
}

Jupyter Notebooks

  • ofa2.ipynb Open in Colab
    This notebook runs the OFA² multi-objective search for 10,000 generations for three different EMOAs (Evolutionary Multi-objective Optimization Algorithms): NSGA-II, SMS-EMOA and SPEA2. Each algorithm runs 3 times considering 3 different random seeds. The Colab version is a simplified version of this notebook and runs only for 1,000 generations for each algorithm.
  • ofa2-debug.ipynb Open in Colab
    This notebook runs the same OFA² search as the previous notebook, but it expands the evolutionary algorithm section and provide more details related to the code implementation. The framework used to implement the multi-objective optimization is the pymoo.

About

OFA²: Train one network, Search once, Deploy in many scenarios

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Jupyter Notebook 92.2%
  • Python 7.8%