用查表法实现超低比特深度学习加速,大幅降低功耗与面积
LUT-DLA: Lookup Table as Efficient Extreme Low-Bit Deep Learning Accelerator
- 用向量量化将模型转为查表结构,突破1比特限制
- 相比传统方法,能效提升1.4至7.0倍,面积效率提升1.5至146.1倍
- 适合边缘设备部署,尤其对计算资源受限场景友好
神经网络规模扩张导致计算需求激增。为减小模型尺寸并降低推理成本,近期研究聚焦于低比特量化与专用硬件加速器设计。但受数值表示限制,标量量化难以低于1比特,优势减弱。为此,我们提出LUT-DLA框架,利用向量量化将神经网络模型转换为查找表(LUT),实现极端低比特量化。该框架支持LUTBoost算法,通过多阶段训练将各类DNN模型转化为基于LUT的模型,显著降低计算与硬件开销。同时,通过联合设计空间探索,评估不同模型与硬件参数的影响,优化适配各类应用场景。实验表明,LUT-DLA在保持小幅精度损失的前提下,实现了1.4~7.0倍的功耗效率提升和1.5~146.1倍的面积效率提升。对于CNN模型,使用不同距离度量时精度下降0.1%~3.8%;对于Transformer模型,精度下降1.4%~3.0%。
原文摘要 · Abstract (English)
The emergence of neural network capabilities invariably leads to a significant surge in computational demands due to expanding model sizes and increased computational complexity. To reduce model size and lower inference costs, recent research has focused on simplifying models and designing hardware accelerators using low-bit quantization. However, due to numerical representation limits, scalar quantization cannot reduce bit width lower than 1-bit, diminishing its benefits. To break through these limitations, we introduce LUT-DLA, a Look-Up Table (LUT) Deep Learning Accelerator Framework that utilizes vector quantization to convert neural network models into LUTs, achieving extreme low-bit quantization. The LUT-DLA framework facilitates efficient and cost-effective hardware accelerator designs and supports the LUTBoost algorithm, which helps to transform various DNN models into LUT-based models via multistage training, drastically cutting both computational and hardware overhead. Additionally, through co-design space exploration, LUT-DLA assesses the impact of various model and hardware parameters to fine-tune hardware configurations for different application scenarios, optimizing performance and efficiency. Our comprehensive experiments show that LUT-DLA achieves improvements in power efficiency and area efficiency with gains of $1.4$~$7.0\times$ and $1.5$~$146.1\times$, respectively, while maintaining only a modest accuracy drop. For CNNs, accuracy decreases by $0.1\%$~$3.1\%$ using the $L_2$ distance similarity, $0.1\%$~$3.4\%$ with the $L_1$ distance similarity, and $0.1\%$~$3.8\%$ when employing the Chebyshev distance similarity. For transformer-based models, the accuracy drop ranges from $1.4\%$ to $3.0\%$.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。