HariyaNobuki / SAENAS-NE

SAENAS-NE

This repo is the official implementation of "Surrogate-assisted evolutionary neural architecture search with network embedding"

Overview

Dataset

CIFAR-10

Experiments

Step 1. Prepare the architecture dataset on DARTS search space

bash scripts/gen_graphs.sh

Step 2. Train the graph2vec model

bash scripts/data_json.sh

Step 3. Search the best architecture on DARTS search space

bash scripts/search.sh

Citation

If you find this project useful in your research, please consider cite:

@article{fan2022surrogate,
  title={Surrogate-assisted evolutionary neural architecture search with network embedding},
  author={Fan, Liang and Wang, Handing},
  journal={Complex \& Intelligent Systems},
  pages={1--19},
  year={2022},
  publisher={Springer}
}