Neural Architecture Search as Multiobjective Optimization Benchmarks: Problem Formulation and Performance Assessment

Research output: Journal article publicationJournal articleAcademic researchpeer-review

76 Citations (Scopus)

Abstract

The ongoing advancements in network architecture design have led to remarkable achievements in deep learning across various challenging computer vision tasks. Meanwhile, the development of neural architecture search (NAS) has provided promising approaches to automating the design of network architectures for lower prediction error. Recently, the emerging application scenarios of deep learning (e.g., autonomous driving) have raised higher demands for network architectures considering multiple design criteria: number of parameters/weights, number of floating-point operations, inference latency, among others. From an optimization point of view, the NAS tasks involving multiple design criteria are intrinsically multiobjective optimization problems; hence, it is reasonable to adopt evolutionary multiobjective optimization (EMO) algorithms for tackling them. Nonetheless, there is still a clear gap confining the related research along this pathway: on the one hand, there is a lack of a general problem formulation of NAS tasks from an optimization point of view; on the other hand, there are challenges in conducting benchmark assessments of EMO algorithms on NAS tasks. To bridge the gap: 1) we formulate NAS tasks into general multiobjective optimization problems and analyze the complex characteristics from an optimization point of view; 2) we present an end-to-end pipeline, dubbed EvoXBench, to generate benchmark test problems for EMO algorithms to run efficiently—without the requirement of GPUs or Pytorch/Tensorflow; and 3) we instantiate two test suites comprehensively covering two datasets, seven search spaces, and three hardware devices, involving up to eight objectives. Based on the above, we validate the proposed test suites using six representative EMO algorithms and provide some empirical analyses. The code of EvoXBench is available at https://github.com/EMI-Group/EvoXBench.

Original languageEnglish
Article number10004638
Pages (from-to)323-337
Number of pages15
JournalIEEE Transactions on Evolutionary Computation
Volume28
Issue number2
DOIs
Publication statusPublished - 1 Apr 2024

Keywords

  • Deep learning
  • evolutionary multiobjective optimization (EMO)
  • neural architecture search (NAS)

ASJC Scopus subject areas

  • Software
  • Theoretical Computer Science
  • Computational Theory and Mathematics

Fingerprint

Dive into the research topics of 'Neural Architecture Search as Multiobjective Optimization Benchmarks: Problem Formulation and Performance Assessment'. Together they form a unique fingerprint.

Cite this