Beyond Training: A Zero-Shot Framework to Neural Architecture and Accelerator Co-Exploration

Wei Fu,Wenqi Lou,Lei Gong,Chao Wang,Xuehai Zhou
DOI: https://doi.org/10.1109/clusterworkshops61563.2024.00032
2024-01-01
Abstract:Recently, the co-exploration of neural architectures and accelerators has become imperative to achieve high accuracy and hardware performance. However, existing methods face a huge training burden during neural network search, which prolongs the search time and increases the task-switching deployment cost. Additionally, their limited hardware search space restricts the potential for optimal results. To address these issues, we propose an efficient zero-shot framework for quickly identifying optimized neural architectures and accelerators based on given requirements. First, we design a training-free proxy to rank the accuracy of the network efficiently, thereby eliminating the need for the network training process. Second, we conduct a finer-grained hardware search space to achieve optimal performance. Preliminary experiments demonstrate that, compared to prior methods, our approach enhances search efficiency by $\mathbf{9.67}\times$ while optimizing the hardware Energy-Delay Product (EDP) by up to $\mathbf{24.6}\times$ , without compromising network accuracy.
What problem does this paper attempt to address?