全二值权重+2比特激活,边缘设备上实现高精度图像识别
Efficiera Residual Networks: Hardware-Friendly Fully Binary Weight with 2-bit Activation Model Achieves Practical ImageNet Accuracy
- 全模型采用二值权重与2比特激活,仅末层用浮点计算
- 小模型<1MB在ImageNet上达63.6%准确率,大模型达72.5%
- 可在低成本FPGA上实现300FPS推理,适合资源受限场景
边缘设备部署深度神经网络面临计算成本、硬件资源和能耗的严苛限制。超低比特量化与硬件加速被视为有效解决方案。尽管已有进展,多数高性能超低比特模型仍部分依赖浮点或非超低比特计算(如输入/输出层)。本文提出面向低资源边缘设备的Efficiera残差网络(ERNs),实现全超低比特量化:所有权重(包括首层与输出层)均为二值,激活为2比特。通过引入共享常数缩放因子技术,使残差连接中运算全程使用整数,仅在最后一层卷积前使用浮点。实验证明,ERNs在与ResNet50兼容的架构下达到72.5%的ImageNet Top-1准确率,模型尺寸小于1MB时达63.6%。此外,在低成本FPGA上,最小模型推理速度达300FPS,最大模型达60FPS。
原文摘要 · Abstract (English)
The edge-device environment imposes severe resource limitations, encompassing computation costs, hardware resource usage, and energy consumption for deploying deep neural network models. Ultra-low-bit quantization and hardware accelerators have been explored as promising approaches to address these challenges. Ultra-low-bit quantization significantly reduces the model size and the computational cost. Despite progress so far, many competitive ultra-low-bit models still partially rely on float or non-ultra-low-bit quantized computation such as the input and output layer. We introduce Efficiera Residual Networks (ERNs), a model optimized for low-resource edge devices. ERNs achieve full ultra-low-bit quantization, with all weights, including the initial and output layers, being binary, and activations set at 2 bits. We introduce the shared constant scaling factor technique to enable integer-valued computation in residual connections, allowing our model to operate without float values until the final convolution layer. Demonstrating competitiveness, ERNs achieve an ImageNet top-1 accuracy of 72.5pt with a ResNet50-compatible architecture and 63.6pt with a model size less than 1MB. Moreover, ERNs exhibit impressive inference times, reaching 300FPS with the smallest model and 60FPS with the largest model on a cost-efficient FPGA device.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。