中文

小型ReLU网络是强大的记忆器:记忆容量的紧致分析

机器学习 2019-10-30 v3 机器学习

摘要

我们研究有限样本表达能力,即ReLU网络的记忆能力。近期结果需要NN个隐藏节点来记忆/插值任意NN个数据点。相反,通过利用深度,我们表明具有Ω(N)\Omega(\sqrt{N})个隐藏节点的3层ReLU网络可以完美记忆大多数含NN个点的数据集。我们还证明了宽度Θ(N)\Theta(\sqrt{N})对于记忆NN个数据点是必要且充分的,给出了记忆容量的紧界。该充分性结果可推广至更深网络;我们表明隐藏层具有WW个参数的LL层网络若W=Ω(N)W = \Omega(N)则可记忆NN个数据点。结合近期关于VC维的上界O(WLlogW)O(WL\log W),我们的构造对任意固定LL近乎紧致。随后,我们在一般位置假设下分析残差网络的记忆容量;我们证明的结果大幅降低了已知的NN个隐藏节点的要求。最后,我们研究随机梯度下降(SGD)的动力学,并表明当在靠近经验风险的记忆化全局最小值处初始化时,SGD会快速找到一个具有更小经验风险的邻近点。

关键词

引用

@article{arxiv.1810.07770,
  title  = {Small ReLU networks are powerful memorizers: a tight analysis of memorization capacity},
  author = {Chulhee Yun and Suvrit Sra and Ali Jadbabaie},
  journal= {arXiv preprint arXiv:1810.07770},
  year   = {2019}
}

备注

28 pages, 2 figures. NeurIPS 2019 Camera-ready version