首次理论分析记忆型神经网络的泛化能力,揭示其泛化所需的最小参数量。
Generalizability of Memorization Neural Networks
- 构建了在i.i.d.数据下参数最少的记忆网络,甚至可恒定参数
- 证明泛化需网络宽度至少等于数据维度,现有最优参数网络无法泛化
- 发现某些分布下泛化需指数级参数,提出高效可泛化的算法
神经网络记忆问题旨在研究神经网络对有限数据集的插值表达能力。尽管记忆被认为与过参数化模型中深度学习的强大泛化能力密切相关,但据我们所知,目前尚无关于记忆型神经网络泛化能力的理论研究。本文首次对此进行理论分析。由于独立同分布(i.i.d.)训练数据是学习算法具备泛化能力的必要条件,因此在对数据分布施加弱条件下,发展了i.i.d.数据集上的记忆及其泛化理论。首先,给出了构造i.i.d.数据集上记忆网络的算法,其参数量最小,甚至可为常数。其次,证明了为使记忆网络具有泛化能力,网络宽度必须至少等于数据维度,这意味着现有参数最优的记忆网络不可泛化。第三,给出了通用记忆算法的样本复杂度下界,以及常数参数记忆算法的精确样本复杂度。还证明存在某些数据分布,使得泛化所需记忆网络参数量随数据维度呈指数增长。最后,当训练样本数超过该数据分布的高效记忆样本复杂度时,给出了一个高效且可泛化的记忆算法。
原文摘要 · Abstract (English)
The neural network memorization problem is to study the expressive power of neural networks to interpolate a finite dataset. Although memorization is widely believed to have a close relationship with the strong generalizability of deep learning when using over-parameterized models, to the best of our knowledge, there exists no theoretical study on the generalizability of memorization neural networks. In this paper, we give the first theoretical analysis of this topic. Since using i.i.d. training data is a necessary condition for a learning algorithm to be generalizable, memorization and its generalization theory for i.i.d. datasets are developed under mild conditions on the data distribution. First, algorithms are given to construct memorization networks for an i.i.d. dataset, which have the smallest number of parameters and even a constant number of parameters. Second, we show that, in order for the memorization networks to be generalizable, the width of the network must be at least equal to the dimension of the data, which implies that the existing memorization networks with an optimal number of parameters are not generalizable. Third, a lower bound for the sample complexity of general memorization algorithms and the exact sample complexity for memorization algorithms with constant number of parameters are given. It is also shown that there exist data distributions such that, to be generalizable for them, the memorization network must have an exponential number of parameters in the data dimension. Finally, an efficient and generalizable memorization algorithm is given when the number of training samples is greater than the efficient memorization sample complexity of the data distribution.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。