Memristive LSTM Network for Sentiment Analysis

TitleMemristive LSTM Network for Sentiment Analysis
Publication TypeJournal Article
Year of Publication2021
AuthorsS Wen, H Wei, Y Yang, Z Guo, Z Zeng, T Huang, and Y Chen
JournalIeee Transactions on Systems, Man, and Cybernetics: Systems
Volume51
Start Page1794
Issue3
Pagination1794 - 1804
Date Published03/2021
Abstract

This paper presents a complete solution for the hardware design of a memristor-based long short-term memory (MLSTM) network. Throughout the design process, we fully consider the external and internal structures of the long short-term memory (LSTM), both of which are efficiently implemented by memristor crossbars. In the specific design of the internal structure, the parameter sharing mechanism is used between the LSTM cells to minimize the hardware design scale. In particular, we designed a circuit that requires only one memristor crossbar for each unit in the LSTM cell. The activation function, including sigmoid and tanh (hyperbolic tangent function), involved in each unit is approximated by a piecewise function, which is designed with the corresponding hardware. To verify the effectiveness of the system we designed, we test it on IMDB and SemEval datasets. Considering the huge impact of the dimensions of the input data on the scale of the hardware design, we use word2vector instead of one-hot encoding for the input data encoding. With the parameter sharing mechanism, the transformed vectors are input in different periods, so only 65 memristive crossbars are needed in the entire system to complete the sentiment analysis of the input text. The experimental results verify the effectiveness of our proposed MLSTM system.

DOI10.1109/TSMC.2019.2906098
Short TitleIeee Transactions on Systems, Man, and Cybernetics: Systems