中文

具有瓶颈的无限宽神经网络中的隐式加速与特征学习

机器学习 2021-07-05 v2

摘要

我们分析了具有有限大小瓶颈的无限宽神经网络的学习动态。与神经正切核极限不同,在 otherwise 无限宽的网络中,瓶颈允许在其瓶颈表示中进行数据依赖的特征学习。我们通过实验表明,与纯无限网络相比,无限网络中的单个瓶颈显著加速了训练,并改善了整体性能。我们通过类比无限宽深度线性模型来讨论这一加速现象,其中瓶颈的加速效应可以从理论上理解。

关键词

引用

@article{arxiv.2107.00364,
  title  = {Implicit Acceleration and Feature Learning in Infinitely Wide Neural Networks with Bottlenecks},
  author = {Etai Littwin and Omid Saremi and Shuangfei Zhai and Vimal Thilak and Hanlin Goh and Joshua M. Susskind and Greg Yang},
  journal= {arXiv preprint arXiv:2107.00364},
  year   = {2021}
}