arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 1806.00166cs.LGcs.ETstat.ML
收藏

用阻性交叉点器件训练LSTM网络

Training LSTM Networks with Resistive Cross-Point Devices

IBM Research AI(IBM研究院人工智能)

Tayfun Gokmen, Malte Rasch, Wilfried Haensch

摘要:

在我们之前的工作中,我们已经证明了电阻交叉点器件,即所谓的电阻处理单元(RPU)器件,在训练深度全连接网络以及卷积神经网络时可以提供显著的功率和速度优势。在这项工作中,我们进一步扩展了RPU的概念,用于训练递归神经网络(RNN),即LSTM。我们证明了递归层的映射与全连接层的映射非常相似,因此RPU概念也可能为RNN提供大的加速因子。此外,我们研究了各种设备缺陷和系统参数对训练性能的影响。更新的对称性对于RNN来说变得更加重要;与用浮点数训练的理想情况相比,百分之几的不对称性已经导致测试误差的增加。此外,设备阵列的输入信号分辨率需要为至少7位以用于成功训练。然而,我们表明,随机舍入方案可以降低输入信号的分辨率回到5位。此外,我们发现RPU设备的变化和硬件噪声足以减轻过拟合,因此不太需要使用dropout。我们注意到,这里训练的模型大约是在MNIST数据集上训练的全连接网络的1500倍,这是每个时期执行的乘法和求和运算的总数。因此,在这里,我们试图研究的有效性RPU方法的大规模网络。

英文摘要:

In our previous work we have shown that resistive cross point devices, so called Resistive Processing Unit (RPU) devices, can provide significant power and speed benefits when training deep fully connected networks as well as convolutional neural networks. In this work, we further extend the RPU concept for training recurrent neural networks (RNNs) namely LSTMs. We show that the mapping of recurrent layers is very similar to the mapping of fully connected layers and therefore the RPU concept can potentially provide large acceleration factors for RNNs as well. In addition, we study the effect of various device imperfections and system parameters on training performance. Symmetry of updates becomes even more crucial for RNNs; already a few percent asymmetry results in an increase in the test error compared to the ideal case trained with floating point numbers. Furthermore, the input signal resolution to device arrays needs to be at least 7 bits for successful training. However, we show that a stochastic rounding scheme can reduce the input signal resolution back to 5 bits. Further, we find that RPU device variations and hardware noise are enough to mitigate overfitting, so that there is less need for using dropout. We note that the models trained here are roughly 1500 times larger than the fully connected network trained on MNIST dataset in terms of the total number of multiplication and summation operations performed per epoch. Thus, here we attempt to study the validity of the RPU approach for large scale networks.

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

↑