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

Fuente: arXiv
Gespeichert in:
Bibliographische Detailangaben
Hauptverfasser: Lammie, Corey, Xiang, Wei, Linares-Barranco, Bernabé, Azghadi, Mostafa Rahimi
Format: Preprint
Veröffentlicht: 2020
Schlagworte:
Online-Zugang:
Tags: Tag hinzufügen
Keine Tags, Fügen Sie den ersten Tag hinzu!
_version_ 1866909469303111680
author Lammie, Corey
Xiang, Wei
Linares-Barranco, Bernabé
Azghadi, Mostafa Rahimi
author_facet Lammie, Corey
Xiang, Wei
Linares-Barranco, Bernabé
Azghadi, Mostafa Rahimi
contents 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
format Preprint
id arxiv_https___arxiv_org_abs_2004_10971
institution arXiv
publishDate 2020
record_format arxiv
spellingShingle MemTorch: An Open-source Simulation Framework for Memristive Deep Learning Systems
Lammie, Corey
Xiang, Wei
Linares-Barranco, Bernabé
Azghadi, Mostafa Rahimi
Emerging Technologies
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
title MemTorch: An Open-source Simulation Framework for Memristive Deep Learning Systems
topic Emerging Technologies
url https://arxiv.org/abs/2004.10971