arXiv · 2001.08450
Low-Complexity LSTM Training and Inference with FloatSD8 Weight Representation
Abstract
The FloatSD technology has been shown to have excellent performance on low-complexity convolutional neural networks (CNNs) training and inference. In this paper, we applied FloatSD to recurrent neural networks (RNNs), specifically long short-term memory (LSTM). In addition to FloatSD weight representation, we quantized the gradients and activations in model training to 8 bits. Moreover, the arithmetic precision for accumulations and the master copy of weights were reduced from 32 bits to 16 bits. We demonstrated that the proposed training scheme can successfully train several LSTM models from scratch, while fully preserving model accuracy. Finally, to verify the proposed method's advantage in implementation, we designed an LSTM neuron circuit and showed that it achieved significantly reduced die area and power consumption.
Explore related subjects
Keep this discovery
Yu-Tung Liu, Tzi-Dar Chiueh. 2020-01-23. Low-Complexity LSTM Training and Inference with FloatSD8 Weight Representation. https://arxiv.org/abs/2001.08450
Cite the original work for its findings. Save a collection to share your selection of sources.