arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2004.10971cs.ET

MemTorch:面向忆阻深度学习系统的开源仿真框架

MemTorch: An Open-source Simulation Framework for Memristive Deep Learning Systems

Corey Lammie, Wei Xiang, Bernabé Linares-Barranco, Mostafa Rahimi Azghadi

AI总结:

本文提出MemTorch开源框架,通过集成PyTorch实现大规模忆阻深度学习系统仿真,重点解决器件非理想特性与外围电路协同建模问题,以提升系统准确度与可靠性。

摘要:

忆阻器件在加速和提升深度学习(DL)系统能效方面展现出巨大潜力。使用这些阻变随机存取存储器(RRAM)器件构建的交叉阵列架构,可用于高效实现各种存内计算操作,例如在深度神经网络(DNN)和卷积神经网络(CNN)中广泛使用的乘加(MAC)和展开卷积。然而,忆阻器件面临老化与非理想特性问题,这限制了忆阻深度学习系统(MDLSs)的准确度、可靠性与鲁棒性,这些问题应在电路级实现前予以考虑。本原创软件出版物(OSP)提出了MemTorch,这是一个用于定制化大规模忆阻DL仿真的开源框架,重点关注器件非理想特性的协同仿真。MemTorch还支持对关键交叉阵列外围电路进行协同建模。MemTorch采用现代软件工程方法,并与著名的PyTorch机器学习(ML)库直接集成。

英文摘要:

Memristive devices have shown great promise to facilitate the acceleration and improve the power efficiency of Deep Learning (DL) systems. Crossbar architectures constructed using these Resistive Random-Access Memory (RRAM) devices can be used to efficiently implement various in-memory computing operations, such as Multiply Accumulate (MAC) and unrolled-convolutions, which are used extensively in Deep Neural Networks (DNNs) and Convolutional Neural Networks (CNNs). However, memristive devices face concerns of aging and non-idealities, which limit the accuracy, reliability, and robustness of Memristive Deep Learning Systems (MDLSs), that should be considered prior to circuit-level realization. This Original Software Publication (OSP) presents MemTorch, an open-source framework for customized large-scale memristive DL simulations, with a refined focus on the co-simulation of device non-idealities. MemTorch also facilitates co-modelling of key crossbar peripheral circuitry. MemTorch adopts a modernized soft-ware engineering methodology and integrates directly with the well-known PyTorch Machine Learning (ML) library

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

↑