arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2007.00784cs.LGcs.DCstat.ML

Convolutional Neural Network Training with Distributed K-FAC

Texas Advanced Computing Center(德州高级计算中心);University of Texas at Austin(德克萨斯大学奥斯汀分校);University of Chicago(芝加哥大学);Argonne National Laboratory(阿贡国家实验室)

J. Gregory Pauloski, Zhao Zhang, Lei Huang, Weijia Xu, Ian T. Foster

英文摘要:

Training neural networks with many processors can reduce time-to-solution; however, it is challenging to maintain convergence and efficiency at large scales. The Kronecker-factored Approximate Curvature (K-FAC) was recently proposed as an approximation of the Fisher Information Matrix that can be used in natural gradient optimizers. We investigate here a scalable K-FAC design and its applicability in convolutional neural network (CNN) training at scale. We study optimization techniques such as layer-wise distribution strategies, inverse-free second-order gradient evaluation, and dynamic K-FAC update decoupling to reduce training time while preserving convergence. We use residual neural networks (ResNet) applied to the CIFAR-10 and ImageNet-1k datasets to evaluate the correctness and scalability of our K-FAC gradient preconditioner. With ResNet-50 on the ImageNet-1k dataset, our distributed K-FAC implementation converges to the 75.9% MLPerf baseline in 18-25% less time than does the classic stochastic gradient descent (SGD) optimizer across scales on a GPU cluster.

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

↑