Neural Architecture Search using Covariance Matrix Adaptation Evolution Strategy

Nilotpal Sinha,Kuan-Wen Chen
DOI: https://doi.org/10.1162/evco_a_00331
IF: 4.7662
2023-04-13
Evolutionary Computation
Abstract:Abstract Evolution-based neural architecture search methods have shown promising results but they require high computational resources since these methods involve training each candidate architecture from scratch and then evaluating its fitness which results in long search time. Covariance Matrix Adaptation Evolution Strategy (CMA-ES) has shown promising results in tuning hyperparameters of neural networks but has not been used for neural architecture search. In this work, we propose a framework called CMANAS which applies the faster convergence property of CMA-ES to the deep neural architecture search problem. Instead of training each individual architecture seperately, we used the accuracy of a trained one shot model (OSM) on the validation data as a prediction of the fitness of the architecture resulting in reduced search time. We also used an architecture-fitness table (AF table) for keeping record of the already evaluated architecture, thus further reducing the search time. The architectures are modelled using a normal distribution, which is updated using CMA-ES based on the fitness of the sampled population. Experimentally, CMANAS achieves better results than previous evolution-based methods while reducing the search time significantly. The effectiveness of CMANAS is shown on 2 different search spaces for datasets: CIFAR-10, CIFAR-100, ImageNet and ImageNet16-120. All the results show that CMANAS is a viable alternative to previous evolution-based methods and extends the application of CMA-ES to the deep neural architecture search field.
computer science, artificial intelligence, theory & methods
What problem does this paper attempt to address?