通过排序权重分段降低存算一体芯片的模数转换能耗
Sorted Weight Sectioning for Energy-Efficient Unstructured Sparse DNNs on Compute-in-Memory Crossbars
- 将小权重和零权重集中分配到低位列,减少模数转换器数量与精度
- 在无结构稀疏BERT模型上实现89.5%的模数转换能耗降低
- 适用于需要低功耗部署的大规模稀疏神经网络推理场景
我们提出一种名为有序权重分段(SWS)的权重分配算法,将深度神经网络(DNN)权重按大小排序后映射至比特切片的存算一体(CIM)交叉阵列,以降低模拟-数字转换器(ADC)的能耗。交叉阵列操作中最耗能的环节是数据转换。利用权重分布呈钟形且多数接近零的特性,SWS仅需低阶列来处理小幅度权重部分,从而显著减少所需ADC的数量与分辨率,实现指数级降低能耗,且对模型精度影响甚微。无结构稀疏化进一步压缩权重分布,在小幅精度损失下提升稀疏性。但其在硬件中追踪零值面临挑战:未排序交叉阵列无法随意切换零行至其他层权重而无需索引匹配。SWS通过离线重映射将零值置于前置分段,充分释放稀疏潜力,最大化能效。实验表明,该方法在无结构稀疏BERT模型上可降低ADC能耗89.5%。本文为无结构稀疏DNN工作负载在存算一体交叉阵列上的高效能实现提供了新算法。
原文摘要 · Abstract (English)
We introduce $\textit{sorted weight sectioning}$ (SWS): a weight allocation algorithm that places sorted deep neural network (DNN) weight sections on bit-sliced compute-in-memory (CIM) crossbars to reduce analog-to-digital converter (ADC) energy consumption. Data conversions are the most energy-intensive process in crossbar operation. SWS effectively reduces this cost leveraging (1) small weights and (2) zero weights (weight sparsity). DNN weights follow bell-shaped distributions, with most weights near zero. Using SWS, we only need low-order crossbar columns for sections with low-magnitude weights. This reduces the quantity and resolution of ADCs used, exponentially decreasing ADC energy costs without significantly degrading DNN accuracy. Unstructured sparsification further sharpens the weight distribution with small accuracy loss. However, it presents challenges in hardware tracking of zeros: we cannot switch zero rows to other layer weights in unsorted crossbars without index matching. SWS efficiently addresses unstructured sparse models using offline remapping of zeros into earlier sections, which reveals full sparsity potential and maximizes energy efficiency. Our method reduces ADC energy use by 89.5% on unstructured sparse BERT models. Overall, this paper introduces a novel algorithm to promote energy-efficient CIM crossbars for unstructured sparse DNN workloads.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。