arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2009.00748cs.ARcs.LG
收藏

TensorDash:利用稀疏性加速深度神经网络训练和推理

TensorDash: Exploiting Sparsity to Accelerate Deep Neural Network Training and Inference

University of Toronto(多伦多大学);Cerebras Systems(赛雷布拉斯系统公司);Vector Institute(向量研究所)

Mostafa Mahmoud, Isak Edo, Ali Hadi Zadeh, Omar Mohamed Awad, Gennady Pekhimenko, Jorge Albericio, Andreas Moshovos

摘要:

TensorDash是一种硬件级技术,用于使数据并行MAC单元能够利用其输入操作数流中的稀疏性。当用于构成深度学习的硬件加速器时,TensorDash可以加快训练过程,同时提高能源效率。TensorDash将低成本、稀疏输入操作数互连与面积高效的硬件调度器相结合,该互连包括每个乘法器输入的8输入多路复用器。虽然互连允许每个操作数的非常有限的移动集合,但是当稀疏性存在于神经网络的激活、权重或梯度中时,调度器可以有效地提取稀疏性。在涵盖各种应用的广泛模型集合中,TensorDash通过$1.95{\times}$加速训练过程,同时在考虑片上和片外内存访问时,$1.89\times$更节能,$1.6\times$更节能。虽然TensorDash适用于任何数据类型,但我们使用单精度浮点单元和bfloat16来演示它。

英文摘要:

TensorDash is a hardware level technique for enabling data-parallel MAC units to take advantage of sparsity in their input operand streams. When used to compose a hardware accelerator for deep learning, TensorDash can speedup the training process while also increasing energy efficiency. TensorDash combines a low-cost, sparse input operand interconnect comprising an 8-input multiplexer per multiplier input, with an area-efficient hardware scheduler. While the interconnect allows a very limited set of movements per operand, the scheduler can effectively extract sparsity when it is present in the activations, weights or gradients of neural networks. Over a wide set of models covering various applications, TensorDash accelerates the training process by $1.95{\times}$ while being $1.89\times$ more energy-efficient, $1.6\times$ more energy efficient when taking on-chip and off-chip memory accesses into account. While TensorDash works with any datatype, we demonstrate it with both single-precision floating-point units and bfloat16.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑