Long short-term memory networks in memristor crossbar arrays
Can Li,Zhongrui Wang,Mingyi Rao,Daniel Belkin,Wenhao Song,Hao Jiang,Peng Yan,Yunning Li,Peng Lin,Miao Hu,Ning Ge,John Paul Strachan,Mark Barnell,Qing Wu,R. Stanley Williams,J. Joshua Yang,Qiangfei Xia
DOI: https://doi.org/10.1038/s42256-018-0001-4
IF: 23.8
2019-01-01
Nature Machine Intelligence
Abstract:Recent breakthroughs in recurrent deep neural networks with long short-term memory (LSTM) units have led to major advances in artificial intelligence. However, state-of-the-art LSTM models with significantly increased complexity and a large number of parameters have a bottleneck in computing power resulting from both limited memory capacity and limited data communication bandwidth. Here we demonstrate experimentally that the synaptic weights shared in different time steps in an LSTM can be implemented with a memristor crossbar array, which has a small circuit footprint, can store a large number of parameters and offers in-memory computing capability that contributes to circumventing the ‘von Neumann bottleneck’. We illustrate the capability of our crossbar system as a core component in solving real-world problems in regression and classification, which shows that memristor LSTM is a promising low-power and low-latency hardware platform for edge inference.
computer science, artificial intelligence, interdisciplinary applications