Accel-NASBench: Sustainable Benchmarking for Accelerator-Aware NAS

التفاصيل البيبلوغرافية
العنوان: Accel-NASBench: Sustainable Benchmarking for Accelerator-Aware NAS
المؤلفون: Ahmad, Afzal, Du, Linfeng, Xie, Zhiyao, Zhang, Wei
سنة النشر: 2024
المجموعة: Computer Science
مصطلحات موضوعية: Computer Science - Machine Learning, Electrical Engineering and Systems Science - 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.
Comment: Accepted at Design Automation Conference DAC'24
نوع الوثيقة: Working Paper
DOI: 10.1145/3649329.3657391
URL الوصول: http://arxiv.org/abs/2404.08005
رقم الأكسشن: edsarx.2404.08005
قاعدة البيانات: arXiv