arrow
返回

Feed-Forward learning algorithm for resistive memories

delete2022-10-01
delete1
PRE
AI
D
Dev Narayan Yadav
P
Phrangboklang Lyngton Thangkhiew *
K
Kamalika Datta
C
Chakraborty, Sandip
R
Rolf Drechsler
I
Indranil Sengupta
DOI:10.1016/j.sysarc.2022.102730delete
delete原文链接
delete原文求助
delete分享
delete收藏
摘要

摘要

En 中文
Resistive memory systems, due to their inherent ability to perform Vector-Matrix Multiplication (VMM), have drawn the attention of researchers to realize machine learning applications with low overheads. In resistive memory systems, each memory cell (synapse/neuron) stores a weight in the form of resistance/conductance value. Memristor-based resistive memory has been widely explored in this regard because of its small size and low power consumption. The inference quality of a neural network depends on how efficiently and accurately the weights are stored in the synapses. The weights are calculated using various training algorithms, like back-propagation (BP), least mean square (LMS), and random weight change (RWC). The training accuracy of existing algorithms is directly related to the algorithm complexity and the time devoted for training. This paper presents a training algorithm that requires an additional set of memristors and a threshold gate for training and achieves an accuracy similar to existing algorithms without using any complex circuitry. The method can update synapse weights in parallel and requires fewer epochs for training an application. Results on experiments with standard benchmarks reveal that the method can achieve an average speedup of 38x as compared to state-of-the-art methods.
Keyword:
Feed-Forward network
Memristor
Neural network
Resistive memory
Training algorithm

期刊

Journal of Systems Architecture 封面图
Journal of Systems Architecture
IF:
4.1
论文数:
3.0K
被引数:
4.2K

机构

I
indian institute of technology system (iit system)
学者数:
9.5W
论文数: 9.9W
被引数: 93
I
indian institute of technology (iit) - kharagpur
学者数:
6.2K
论文数: 6.5K
被引数: 6