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

浅神经网络学习生成模型的高斯等价性

The Gaussian equivalence of generative models for learning with shallow neural networks

International School of Advanced Studies (SISSA)(国际高等研究学院(SISSA));Duke University(杜克大学);Université PSL(巴黎文理研究大学);Sorbonne Université(索邦大学);Université Paris-Diderot(巴黎第七大学)

Sebastian Goldt, Bruno Loureiro, Galen Reeves, Florent Krzakala, Marc Mézard, Lenka Zdeborová

摘要:

理解数据结构对学习的计算易处理性的影响是神经网络理论的一个关键挑战。许多理论工作没有明确地对训练数据进行建模,或者假设输入是独立于一些简单的概率分布而绘制的。在这里,我们超越了这个简单的范例,通过研究神经网络的性能来训练从预训练的生成模型中提取的数据。这是可能的,因为高斯等价性表明,感兴趣的关键指标,如训练和测试误差,可以通过适当选择的高斯模型完全捕获。我们提供了三条严格的,分析和数值证据证实了这种等价性。首先,我们建立严格的条件,高斯等价的情况下,单层生成模型,以及确定性的速度分布依分布收敛。其次,我们利用这种等价性推导出一组封闭的方程,描述了两个广泛研究的机器学习问题的泛化性能:使用单次随机梯度下降训练的双层神经网络,以及全批预学习特征或核方法。最后,我们进行了实验,展示了我们的理论如何应用于深度,预训练的生成模型。这些结果为基于真实数据的机器学习模型的理论研究开辟了一条可行的道路。

英文摘要:

Understanding the impact of data structure on the computational tractability of learning is a key challenge for the theory of neural networks. Many theoretical works do not explicitly model training data, or assume that inputs are drawn component-wise independently from some simple probability distribution. Here, we go beyond this simple paradigm by studying the performance of neural networks trained on data drawn from pre-trained generative models. This is possible due to a Gaussian equivalence stating that the key metrics of interest, such as the training and test errors, can be fully captured by an appropriately chosen Gaussian model. We provide three strands of rigorous, analytical and numerical evidence corroborating this equivalence. First, we establish rigorous conditions for the Gaussian equivalence to hold in the case of single-layer generative models, as well as deterministic rates for convergence in distribution. Second, we leverage this equivalence to derive a closed set of equations describing the generalisation performance of two widely studied machine learning problems: two-layer neural networks trained using one-pass stochastic gradient descent, and full-batch pre-learned features or kernel methods. Finally, we perform experiments demonstrating how our theory applies to deep, pre-trained generative models. These results open a viable path to the theoretical study of machine learning models with realistic data.

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

↑