HCM: Hardware-Aware Complexity Metric for Neural Network Architectures

• Published in: arXiv.org
• Main Authors: Karbachevsky, Alex, Baskin, Chaim, Zheltonozhskii, Evgenii, Yermolin, Yevgeny, Gabbay, Freddy, Bronstein, Alex M, Mendelson, Avi
• Format: Paper Journal Article
• Language: English
• Published: Ithaca Cornell University Library, arXiv.org 26.04.2020
• Subjects: Accelerators; Accuracy; Artificial neural networks; Complexity; Computer architecture; Computer Science - Hardware Architecture; Computer Science - Learning; Computer vision; Decisions; Design; Design parameters; Electronic devices; Embedded systems; Energy conservation; Hardware; Impact prediction; Natural language processing; Neural networks; Speech recognition
• ISSN: 2331-8422
• DOI: 10.48550/arxiv.2004.08906

Convolutional Neural Networks (CNNs) have become common in many fields including computer vision, speech recognition, and natural language processing. Although CNN hardware accelerators are already included as part of many SoC architectures, the task of achieving high accuracy on resource-restricted devices is still considered challenging, mainly due to the vast number of design parameters that need to be balanced to achieve an efficient solution. Quantization techniques, when applied to the network parameters, lead to a reduction of power and area and may also change the ratio between communication and computation. As a result, some algorithmic solutions may suffer from lack of memory bandwidth or computational resources and fail to achieve the expected performance due to hardware constraints. Thus, the system designer and the micro-architect need to understand at early development stages the impact of their high-level decisions (e.g., the architecture of the CNN and the amount of bits used to represent its parameters) on the final product (e.g., the expected power saving, area, and accuracy). Unfortunately, existing tools fall short of supporting such decisions. This paper introduces a hardware-aware complexity metric that aims to assist the system designer of the neural network architectures, through the entire project lifetime (especially at its early stages) by predicting the impact of architectural and micro-architectural decisions on the final product. We demonstrate how the proposed metric can help evaluate different design alternatives of neural network models on resource-restricted devices such as real-time embedded systems, and to avoid making design mistakes at early stages.