Title
Mixed-precision training of deep neural networks using computational memory.
Abstract
Deep neural networks have revolutionized the field of machine learning by providing unprecedented human-like performance in solving many real-world problems such as image and speech recognition. Training of large DNNs, however, is a computationally intensive task, and this necessitates the development of novel computing architectures targeting this application. A computational memory unit where resistive memory devices are organized in crossbar arrays can be used to locally store the synaptic weights in their conductance states. The expensive multiply accumulate operations can be performed in place using Kirchhoffu0027s circuit laws in a non-von Neumann manner. However, a key challenge remains the inability to alter the conductance states of the devices in a reliable manner during the weight update process. We propose a mixed-precision architecture that combines a computational memory unit storing the synaptic weights with a digital processing unit and an additional memory unit accumulating weight updates in high precision. The new architecture delivers classification accuracies comparable to those of floating-point implementations without being constrained by challenges associated with the non-ideal weight update characteristics of emerging resistive memories. A two layer neural network in which the computational memory unit is realized using non-linear stochastic models of phase-change memory devices achieves a test accuracy of 97.40% on the MNIST handwritten digit classification problem.
Year
Venue
Field
2017
arXiv: Emerging Technologies
MNIST database,Computer science,Resistive touchscreen,Electronic engineering,Implementation,Stochastic modelling,Artificial neural network,Computer engineering,Crossbar switch,Deep neural networks,Resistive random-access memory
DocType
Volume
Citations 
Journal
abs/1712.01192
0
PageRank 
References 
Authors
0.34
10
6
Name
Order
Citations
PageRank
S. R. Nandakumar1103.67
Manuel Le Gallo2479.73
Irem Boybat3345.41
Bipin Rajendran44712.17
Sebastian, A.526744.35
Evangelos Eleftheriou61590118.20