English

GP-NAS-ensemble: a model for NAS Performance Prediction

Machine Learning 2023-01-24 v1 Applications Machine Learning

Abstract

It is of great significance to estimate the performance of a given model architecture without training in the application of Neural Architecture Search (NAS) as it may take a lot of time to evaluate the performance of an architecture. In this paper, a novel NAS framework called GP-NAS-ensemble is proposed to predict the performance of a neural network architecture with a small training dataset. We make several improvements on the GP-NAS model to make it share the advantage of ensemble learning methods. Our method ranks second in the CVPR2022 second lightweight NAS challenge performance prediction track.

Keywords

Cite

@article{arxiv.2301.09231,
  title  = {GP-NAS-ensemble: a model for NAS Performance Prediction},
  author = {Kunlong Chen and Liu Yang and Yitian Chen and Kunjin Chen and Yidan Xu and Lujun Li},
  journal= {arXiv preprint arXiv:2301.09231},
  year   = {2023}
}
R2 v1 2026-06-28T08:17:28.623Z