Saved in:
Bibliographic Details
Main Authors: Ahmad, Afzal, Du, Linfeng, Xie, Zhiyao, Zhang, Wei
Format: Preprint
Published: 2024
Subjects:
Online Access:https://arxiv.org/abs/2404.08005
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866909225718906880
author Ahmad, Afzal
Du, Linfeng
Xie, Zhiyao
Zhang, Wei
author_facet Ahmad, Afzal
Du, Linfeng
Xie, Zhiyao
Zhang, Wei
contents One of the primary challenges impeding the progress of Neural Architecture Search (NAS) is its extensive reliance on exorbitant computational resources. NAS benchmarks aim to simulate runs of NAS experiments at zero cost, remediating the need for extensive compute. However, existing NAS benchmarks use synthetic datasets and model proxies that make simplified assumptions about the characteristics of these datasets and models, leading to unrealistic evaluations. We present a technique that allows searching for training proxies that reduce the cost of benchmark construction by significant margins, making it possible to construct realistic NAS benchmarks for large-scale datasets. Using this technique, we construct an open-source bi-objective NAS benchmark for the ImageNet2012 dataset combined with the on-device performance of accelerators, including GPUs, TPUs, and FPGAs. Through extensive experimentation with various NAS optimizers and hardware platforms, we show that the benchmark is accurate and allows searching for state-of-the-art hardware-aware models at zero cost.
format Preprint
id arxiv_https___arxiv_org_abs_2404_08005
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Accel-NASBench: Sustainable Benchmarking for Accelerator-Aware NAS
Ahmad, Afzal
Du, Linfeng
Xie, Zhiyao
Zhang, Wei
Machine Learning
Image and Video Processing
One of the primary challenges impeding the progress of Neural Architecture Search (NAS) is its extensive reliance on exorbitant computational resources. NAS benchmarks aim to simulate runs of NAS experiments at zero cost, remediating the need for extensive compute. However, existing NAS benchmarks use synthetic datasets and model proxies that make simplified assumptions about the characteristics of these datasets and models, leading to unrealistic evaluations. We present a technique that allows searching for training proxies that reduce the cost of benchmark construction by significant margins, making it possible to construct realistic NAS benchmarks for large-scale datasets. Using this technique, we construct an open-source bi-objective NAS benchmark for the ImageNet2012 dataset combined with the on-device performance of accelerators, including GPUs, TPUs, and FPGAs. Through extensive experimentation with various NAS optimizers and hardware platforms, we show that the benchmark is accurate and allows searching for state-of-the-art hardware-aware models at zero cost.
title Accel-NASBench: Sustainable Benchmarking for Accelerator-Aware NAS
topic Machine Learning
Image and Video Processing
url https://arxiv.org/abs/2404.08005