中文

GP-NAS-ensemble:一种用于 NAS 性能预测的模型

机器学习 2023-01-24 v1 应用统计 机器学习

摘要

在神经架构搜索(NAS)应用中,无需训练即可估计给定模型架构的性能具有重要意义,因为评估一个架构的性能可能耗费大量时间。本文提出一种名为 GP-NAS-ensemble 的新型 NAS 框架,利用小规模训练数据集预测神经网络架构的性能。我们对 GP-NAS 模型作了若干改进,使其具备集成学习方法的优势。我们的方法在 CVPR2022 第二届轻量级 NAS 挑战赛性能预测赛道中排名第二。

关键词

引用

@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}
}