arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2004.12041cs.LGstat.ML
收藏

使用流式降维的记忆高效训练

Memory-efficient training with streaming dimensionality reduction

George Washington University(乔治华盛顿大学);National Institute of Standards and Technology(美国国家标准与技术研究院)

Siyuan Huang, Brian D. Hoskins, Matthew W. Daniels, Mark D. Stiles, Gina C. Adam

摘要:

在深度神经网络的训练过程中,大量数据的移动给机器学习工作负载带来了巨大的挑战。为了最大限度地减少这种开销,特别是在梯度信息的移动和计算上,我们引入了流式批量主成分分析作为更新算法。流式批量主成分分析使用随机幂迭代来生成网络梯度的随机k秩近似。我们证明了由流批主成分分析产生的低秩更新可以有效地在各种常见数据集上训练卷积神经网络,其性能与标准迷你批量梯度下降相当。这些结果可以改进用于深度学习的专用集成电路的设计,以及用数据并行训练的机器学习模型的同步速度。

英文摘要:

The movement of large quantities of data during the training of a Deep Neural Network presents immense challenges for machine learning workloads. To minimize this overhead, especially on the movement and calculation of gradient information, we introduce streaming batch principal component analysis as an update algorithm. Streaming batch principal component analysis uses stochastic power iterations to generate a stochastic k-rank approximation of the network gradient. We demonstrate that the low rank updates produced by streaming batch principal component analysis can effectively train convolutional neural networks on a variety of common datasets, with performance comparable to standard mini batch gradient descent. These results can lead to both improvements in the design of application specific integrated circuits for deep learning and in the speed of synchronization of machine learning models trained with data parallelism.

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

↑