arrow
Return

Mixed-Precision Continual Learning Based on Computational Resistance Random Access Memory

delete2022-05-17
delete9
delete
OA
AI
Y
Yi Li
W
Woyu Zhang
X
Xiaoxin Xu
何以凡 (Yifan He)
D
Danian Dong
N
Nanjia Jiang
F
Fei Wang
Z
Zeyu Guo
S
Shaocong Wang
C
Chunmeng Dou
Y
Yongpan Liu
Z
Zhongrui Wang *
D
Dashan Shang *
DOI:10.1002/aisy.202200026delete
deleteOriginal
deleteOriginal request for help
deleteShare
deleteSave
Abstract

Abstract

En 中文
Artificial neural networks have acquired remarkable achievements in the field of artificial intelligence. However, it suffers from catastrophic forgetting when dealing with continual learning problems, i.e., the loss of previously learned knowledge upon learning new information. Although several continual learning algorithms have been proposed, it remains a challenge to implement these algorithms efficiently on conventional digital systems due to the physical separation between memory and processing units. Herein, a software-hardware codesigned in-memory computing paradigm is proposed, where a mixed-precision continual learning (MPCL) model is deployed on a hybrid analogue-digital hardware system equipped with resistance random access memory chip. Software-wise, the MPCL effectively alleviates catastrophic forgetting and circumvents the requirement for high-precision weights. Hardware-wise, the hybrid analogue-digital system takes advantage of the colocation of memory and processing units, greatly improving energy efficiency. By combining the MPCL with an in situ fine-tuning method, high classification accuracies of 94.9% and 95.3% (software baseline 97.0% and 97.7%) on the 5-split-MNIST and 5-split-FashionMNIST are achieved, respectively. The proposed system reduces approximate to 200 times energy consumption of the multiply-and-accumulation operations during the inference phase compared to the conventional digital systems. This work paves the way for future autonomous systems at the edge.
Keywords:
continual learning
in-memory computing
mixed precision
resistance random access memory

Journal

Advanced Intelligent Systems cover
Advanced Intelligent Systems
IF:
6.1
Papers:
2.0K
Citations:
8.4K

Organization

U
university of chinese academy of sciences, cas
Scholars:
4.1W
Papers: 3.8W
Citations: 75
C
chinese academy of sciences
Scholars:
56.2W
Papers: 44.8W
Citations: 704