arXiv:2409.05227cs.LGcs.AR2024-09中稿 · IEEE/ACM MICRO 202…被引 25

双向位稀疏技术让深度学习加速更高效,省空间又省电。

BBS: Bi-directional Bit-level Sparsity for Deep Learning Acceleration

  • 双向位稀疏对0和1位都可剪枝,提升计算均衡性
  • 模型尺寸平均减少1.66倍,精度损失小于0.5%
  • 适合量化后的DNN模型,硬件加速器低开销

位级稀疏方法通过跳过无效的零位操作,适用于位串行深度学习加速器。此类稀疏在位级别上既正交又兼容其他网络效率技术,如量化与剪枝。本文提出一种新型算法——双向位稀疏(BBS),其核心思想是对零位和一比特同时进行对称剪枝,显著改善位串行计算的负载均衡,并保证稀疏率超过50%。在此基础上,我们进一步设计两种无需重训练的二值化剪枝方法,可无缝应用于量化DNN。结合新的张量编码方案,BBS既能跳过计算,又能降低双向稀疏位列的内存开销。硬件方面,我们设计了BitVert——一种高效的位串行架构,配备优化的处理单元,利用上述二值化剪枝实现低开销加速。七种代表性DNN模型评估显示:平均模型尺寸缩小1.66倍,精度损失<0.5%;最高提速3.03倍,能耗降低2.44倍。

原文摘要 · Abstract (English)

Bit-level sparsity methods skip ineffectual zero-bit operations and are typically applicable within bit-serial deep learning accelerators. This type of sparsity at the bit-level is especially interesting because it is both orthogonal and compatible with other deep neural network (DNN) efficiency methods such as quantization and pruning. In this work, we improve the practicality and efficiency of bitlevel sparsity through a novel algorithmic bit-pruning, averaging, and compression method, and a co-designed efficient bit-serial hardware accelerator. On the algorithmic side, we introduce bidirectional bit sparsity (BBS). The key insight of BBS is that we can leverage bit sparsity in a symmetrical way to prune either zero-bits or one-bits. This significantly improves the load balance of bit-serial computing and guarantees the level of sparsity to be more than 50%. On top of BBS, we further propose two bit-level binary pruning methods that require no retraining, and can be seamlessly applied to quantized DNNs. Combining binary pruning with a new tensor encoding scheme, BBS can both skip computation and reduce the memory footprint associated with bi-directional sparse bit columns. On the hardware side, we demonstrate the potential of BBS through BitVert, a bitserial architecture with an efficient PE design to accelerate DNNs with low overhead, exploiting our proposed binary pruning. Evaluation on seven representative DNN models shows that our approach achieves: (1) on average 1.66$\times$ reduction in model sizewith negligible accuracy loss of < 0.5%; (2) up to 3.03$\times$ speedupand 2.44$\times$ energy saving compared to prior DNN accelerators.

位稀疏深度学习加速硬件协同设计

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。