Abstract:Long short-term memory (LSTM) is a type of powerful deep neural network that has been widely used in many sequence analysis and modeling applications. However, the large model size problem of LSTM networks make their practical deployment still very challenging, especially for the video recognition tasks that require high-dimensional input data. Aiming to overcome this limitation and fully unlock the potentials of LSTM models, in this paper we propose to perform algorithm and hardware co-design towards high-performance energy-efficient LSTM networks. At algorithm level, we propose to develop fully decomposed hierarchical Tucker (FDHT) structure-based LSTM, namely FDHT-LSTM, which enjoys ultra-low model complexity while still achieving high accuracy. In order to fully reap such attractive algorithmic benefit, we further develop the corresponding customized hardware architecture to support the efficient execution of the proposed FDHT-LSTM model. With the delicate design of memory access scheme, the complicated matrix transformation can be efficiently supported by the underlying hardware without any access conflict in an on-the-fly way. Our evaluation results show that both the proposed ultra-compact FDHT-LSTM models and the corresponding hardware accelerator achieve very high performance. Compared with the state-of-the-art compressed LSTM models, FDHT-LSTM enjoys both order-of-magnitude reduction in model size and significant accuracy improvement across different video recognition datasets. Meanwhile, compared with the state-of-the-art tensor decomposed model-oriented hardware TIE, our proposed FDHT-LSTM architecture achieves better performance in throughput, area efficiency and energy efficiency, respectively on LSTM-Youtube workload. For LSTM-UCF workload, our proposed design also outperforms TIE with higher throughput, higher energy efficiency and comparable area efficiency.

<sup>2</sup>b-sigmoid and <sup>2</sup>b-tanh: Low Hardware Complexity Activation Functions for LSTM

2 β-softmax: A Hardware-Friendly Activation Function with Low Complexity and High Performance

An Efficient FP16-Based Hardware Architecture with Enhanced CORDIC to Implement Sigmoid and Tanh Functions

An Efficient Hardware Architecture with Adjustable Precision and Extensible Range to Implement Sigmoid and Tanh Functions

Base-2 Softmax Function: Suitability for Training and Efficient Hardware Implementation

Enhancing Sequential Model Performance with Squared Sigmoid TanH (SST) Activation Under Data Constraints

A Compact and Configurable Long Short-Term Memory Neural Network Hardware Architecture.

A CORDIC-Based Architecture with Adjustable Precision and Flexible Scalability to Implement Sigmoid and Tanh Functions.

Design Space Exploration of Neural Network Activation Function Circuits

Enhancement of Neural Networks with an Alternative Activation Function Tanhlu

A Highly Configurable 7.62gop/s Hardware Implementation for LSTM

Hardware-Efficient SoftMax Architecture With Bit-Wise Exponentiation and Reciprocal Calculation

Hardware-oriented Algorithms for Softmax and Layer Normalization of Large Language Models

Efficient Weight Reuse for Large LSTMs.

Hardware-Guided Symbiotic Training for Compact, Accurate, yet Execution-Efficient LSTM

Computing with magnetic tunnel junction based sigmoidal activation functions

Algorithm and Hardware Co-Design of Energy-Efficient LSTM Networks for Video Recognition with Hierarchical Tucker Tensor Decomposition

High-Precision Method and Architecture for Base-2 Softmax Function in DNN Training.

Towards Binary-Valued Gates for Robust LSTM Training

Activation function optimization method: Learnable series linear units (LSLUs)

Smish: A Novel Activation Function for Deep Learning Methods