arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2001.08450cs.LGstat.ML

Low-Complexity LSTM Training and Inference with FloatSD8 Weight Representation

National Taiwan University(台湾大学)

Yu-Tung Liu, Tzi-Dar Chiueh

英文摘要:

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.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑