arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2004.01190stat.MLcond-mat.dis-nncond-mat.stat-mechcs.LGcs.NE
收藏

预测用噪声梯度训练的有限深度神经网络的输出

Predicting the outputs of finite deep neural networks trained with noisy gradients

Racah Institute of Physics, Hebrew University(希伯来大学拉卡物理研究所);Edmond and Lily Safra Center for Brain Sciences, Hebrew University(希伯来大学埃德蒙与莉莉·萨夫拉脑科学中心);Center for Brain Science, Harvard University(哈佛大学脑科学中心);Agilent Research Labs(安捷伦研究实验室)

Gadi Naveh, Oded Ben-David, Haim Sompolinsky, Zohar Ringel

摘要:

最近的一系列工作通过将其近似为高斯过程(GP)来研究广泛的深度神经网络(DNN)。用梯度流训练的DNN被证明可以映射到由神经切线核(NTK)控制的GP,而早期的工作表明,具有i.i.d.其权重的先验映射到所谓的神经网络高斯过程(NNGP)。在这里,我们考虑DNN训练协议,涉及噪声,权重衰减和有限宽度,其结果对应于一定的非高斯随机过程。然后引入了一个分析框架来分析这个非高斯过程,它与GP的偏差由有限宽度控制。我们的贡献是三方面的:(i)在无限宽的限制下,我们在用噪声梯度训练的DNN和NNGP之间建立了对应关系,而不是NTK。(ii)我们为具有任意激活函数和深度的DNN的有限宽度校正(FWC)提供了一种通用的分析形式,并使用它来高精度地预测经验有限网络的输出。分析FWC行为作为$n$(训练集大小)的函数,我们发现,对于非常小的$n$机制,以及令人惊讶的是,对于大的$n$机制(其中GP误差缩放为$O(1/n)$),FWC行为可以忽略不计。(iii)我们从代数上充实了这些FWC如何提高有限卷积神经网络(CNN)相对于图像分类任务上的GP同行的性能。

英文摘要:

A recent line of works studied wide deep neural networks (DNNs) by approximating them as Gaussian Processes (GPs). A DNN trained with gradient flow was shown to map to a GP governed by the Neural Tangent Kernel (NTK), whereas earlier works showed that a DNN with an i.i.d. prior over its weights maps to the so-called Neural Network Gaussian Process (NNGP). Here we consider a DNN training protocol, involving noise, weight decay and finite width, whose outcome corresponds to a certain non-Gaussian stochastic process. An analytical framework is then introduced to analyze this non-Gaussian process, whose deviation from a GP is controlled by the finite width. Our contribution is three-fold: (i) In the infinite width limit, we establish a correspondence between DNNs trained with noisy gradients and the NNGP, not the NTK. (ii) We provide a general analytical form for the finite width correction (FWC) for DNNs with arbitrary activation functions and depth and use it to predict the outputs of empirical finite networks with high accuracy. Analyzing the FWC behavior as a function of $n$, the training set size, we find that it is negligible for both the very small $n$ regime, and, surprisingly, for the large $n$ regime (where the GP error scales as $O(1/n)$). (iii) We flesh out algebraically how these FWCs can improve the performance of finite convolutional neural networks (CNNs) relative to their GP counterparts on image classification tasks.

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

↑