Lammie, Corey, and Rahimi Azghadi, Mostafa (2020) MemTorch: a simulation framework for deep memristive cross-bar architectures. In: Proceedings of the 2020 IEEE International Symposium on Circuits and Systems. From: ISCAS: 2020 IEEE International Symposium on Circuits and Systems, 10-21 October 2020, Seville, Spain.

PDF (Accepted Author Version) - Accepted Version Restricted to Repository staff only

Abstract

Memristive devices arranged in cross-bar architectures have shown great promise to facilitate the acceleration and improve the power efficiency of Deep Learning (DL) systems for deployment in resource-constrained platforms, such as the Internet-of-Things (IoT) edge devices. These cross-bar architectures can be used to implement various in-memory computing operations, such as Multiply-Accumulate (MAC) and convolution, which are used extensively in Deep Neural Networks (DNNs) and Convolutional Neural Networks (CNNs). Currently, there is a lack of an open source, general, high-level simulation platform that can fully integrate any behavioral or experimental memristive device model into cross-bar architectures. This paper presents such a framework named MemTorch, which integrates directly with the well-known PyTorch Machine Learning (ML) library. To demonstrate an example practical use of MemTorch, we use it to simulate the performance degradation that non-ideal devices introduce to a typical Memristive DNN (MDNN) implementing VGG-16 for CIFAR-10. Our open source 1 MemTorch framework can be used by circuit and system designers to conveniently build customized large-scale simulation platforms, as a preliminary step before circuit-level realization.

Item ID:	65711
Item Type:	Conference Item (Research - E1)
ISBN:	978-1-7281-3320-1
Copyright Information:	(C) IEEE
Additional Information:	This was an online event.
Date Deposited:	04 Feb 2021 01:03
FoR Codes:	46 INFORMATION AND COMPUTING SCIENCES > 4612 Software engineering > 461299 Software engineering not elsewhere classified @ 50% 40 ENGINEERING > 4009 Electronics, sensors and digital hardware > 400908 Microelectronics @ 50%
SEO Codes:	89 INFORMATION AND COMMUNICATION SERVICES > 8902 Computer Software and Services > 890202 Application Tools and System Utilities @ 50% 97 EXPANDING KNOWLEDGE > 970109 Expanding Knowledge in Engineering @ 50%
Downloads:	Total: 1
	More Statistics