arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2404.08005cs.LGeess.IV
收藏

Accel-NASBench:面向加速器感知 NAS 的可持续基准测试

Accel-NASBench: Sustainable Benchmarking for Accelerator-Aware NAS

The Hong Kong University of Science and Technology(香港科技大学)

Afzal Ahmad, Linfeng Du, Zhiyao Xie, Wei Zhang

AI总结:

本文提出可搜索训练代理的基准构建技术,降低真实 NAS 基准成本,并发布结合 ImageNet2012 与 GPU、TPU、FPGA 性能的开源双目标 Accel-NASBench。

摘要:

阻碍神经架构搜索(Neural Architecture Search,NAS)进展的主要挑战之一,是其对高昂计算资源的严重依赖。NAS 基准旨在以零成本模拟 NAS 实验的运行,从而缓解对大量计算的需求。然而,现有 NAS 基准使用合成数据集和模型代理,并对这些数据集和模型的特征作出简化假设,导致评估结果不切实际。我们提出一种技术,能够搜索训练代理,以大幅降低基准构建成本,从而使为大规模数据集构建真实的 NAS 基准成为可能。利用该技术,我们为 ImageNet2012 数据集构建了一个开源双目标 NAS 基准,并结合加速器(包括 GPU、TPU 和 FPGA)的设备端性能。通过在多种 NAS 优化器和硬件平台上进行大量实验,我们表明该基准是准确的,并支持以零成本搜索最先进的硬件感知模型。

英文摘要:

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.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑