MemTorch: a simulation framework for deep memristive cross-bar architectures

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.

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

View at Publisher Website:


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

Actions (Repository Staff Only)

Item Control Page Item Control Page