arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2102.06365cs.LGcs.ET
收藏

神经网络的动态精确模拟计算

Dynamic Precision Analog Computing for Neural Networks

Luminous Computing(卢米诺斯计算公司)

Sahaj Garg, Joe Lou, Anirudh Jain, Mitchell Nahmias

摘要:

模拟电子和光计算在以低精度执行操作时,在加速深度学习方面表现出了优于数字计算的巨大优势。在这项工作中,我们推导出模拟精度,这是有限的噪声,和数字位精度之间的关系。我们建议扩展模拟计算架构,通过重复操作和平均结果来支持不同级别的精度,从而降低噪声的影响。这样的架构使得能够在精度和其他期望的性能度量(诸如能量效率或吞吐量)之间进行可编程的权衡。为了利用动态精度,我们提出了一种方法,用于学习预训练模型的每一层的精度,而无需重新训练网络权重。我们在模拟架构上对这种方法进行了评估,这些模拟架构受到各种噪声源(如散粒噪声、热噪声和权重噪声)的影响,发现采用动态精度可以将计算机视觉模型(如Resnet 50)的能耗降低89%,将自然语言处理模型(如BERT)的能耗降低24%。在一个示例中,我们将动态精度应用于散粒噪声限制的零差光学神经网络,并在Resnet 50的2.7 aJ/MAC和BERT的1.6 aJ/MAC的光能量消耗下模拟推理,精度下降<2%。

英文摘要:

Analog electronic and optical computing exhibit tremendous advantages over digital computing for accelerating deep learning when operations are executed at low precision. In this work, we derive a relationship between analog precision, which is limited by noise, and digital bit precision. We propose extending analog computing architectures to support varying levels of precision by repeating operations and averaging the result, decreasing the impact of noise. Such architectures enable programmable tradeoffs between precision and other desirable performance metrics such as energy efficiency or throughput. To utilize dynamic precision, we propose a method for learning the precision of each layer of a pre-trained model without retraining network weights. We evaluate this method on analog architectures subject to a variety of noise sources such as shot noise, thermal noise, and weight noise and find that employing dynamic precision reduces energy consumption by up to 89% for computer vision models such as Resnet50 and by 24% for natural language processing models such as BERT. In one example, we apply dynamic precision to a shot-noise limited homodyne optical neural network and simulate inference at an optical energy consumption of 2.7 aJ/MAC for Resnet50 and 1.6 aJ/MAC for BERT with <2% accuracy degradation.

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

↑