arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2205.09653stat.MLcond-mat.dis-nncs.LG
收藏

广义神经网络核演化的自洽动力场论

Self-Consistent Dynamical Field Theory of Kernel Evolution in Wide Neural Networks

Harvard University(哈佛大学);Center for Brain Science(脑科学中心);John Paulson School of Engineering and Applied Sciences(约翰·保尔森工程与应用科学学院)

Blake Bordelon, Cengiz Pehlevan

摘要:

我们通过自洽动力场理论分析了梯度流训练的无限宽神经网络的特征学习。我们构建了一组确定性的动态序参数,这些参数是在成对的时间点上每层中隐藏单元激活和梯度的内积内核,通过训练提供了对网络活动的简化描述。这些核序参数共同定义了隐藏层激活分布、神经正切核的演化以及因此的输出预测。我们证明了场论推导恢复了Yang和Hu(2021)用张量程序获得的无限宽度特征学习网络的递归随机过程。对于深度线性网络,这些核满足一组代数矩阵方程。对于非线性网络,我们提供了一个交替采样的过程来自洽地解决核序参数。我们提供的自洽的解决方案,包括静态NTK近似,梯度独立性假设,和领导阶扰动理论的各种近似方案的比较,表明这些近似可以打破一般自洽的解决方案仍然提供了准确的描述制度。最后,我们提供了在更现实的设置中的实验,这些实验表明,在CIFAR分类任务中,在固定特征学习强度下,CNN的损失和内核动态在不同的宽度上都得到了保留。

英文摘要:

We analyze feature learning in infinite-width neural networks trained with gradient flow through a self-consistent dynamical field theory. We construct a collection of deterministic dynamical order parameters which are inner-product kernels for hidden unit activations and gradients in each layer at pairs of time points, providing a reduced description of network activity through training. These kernel order parameters collectively define the hidden layer activation distribution, the evolution of the neural tangent kernel, and consequently output predictions. We show that the field theory derivation recovers the recursive stochastic process of infinite-width feature learning networks obtained from Yang and Hu (2021) with Tensor Programs . For deep linear networks, these kernels satisfy a set of algebraic matrix equations. For nonlinear networks, we provide an alternating sampling procedure to self-consistently solve for the kernel order parameters. We provide comparisons of the self-consistent solution to various approximation schemes including the static NTK approximation, gradient independence assumption, and leading order perturbation theory, showing that each of these approximations can break down in regimes where general self-consistent solutions still provide an accurate description. Lastly, we provide experiments in more realistic settings which demonstrate that the loss and kernel dynamics of CNNs at fixed feature learning strength is preserved across different widths on a CIFAR classification task.

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

↑