arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2107.02253cs.LGmath.DGmath.PR
收藏

设计泛化:深度学习中泛化的捷径

Generalization by design: Shortcuts to Generalization in Deep Learning

Technical University of Denmark(丹麦技术大学)

Petr Taborsky, Lars Kai Hansen

摘要:

我们从几何学的角度出发,通过Bregman散度损失函数,对监督式深度学习提出了一个统一的观点-这需要频繁的分类和预测任务。受模拟的启发,我们认为,深度模型的香草随机梯度下降训练对"更简单"的函数基本上没有隐含的偏见。相反,我们表明,良好的推广可能会煽动有界谱产品层导致一种新的几何正则化。据透露,在足够深的模型,这样的正则化,使双方,极端的准确性和泛化,达到。我们将流行的正则化技术,如权重衰减、丢弃、批量归一化和提前停止与此观点联系起来。在理论的支持下,我们进一步证明了“设计泛化”在实践中是可能的,并且良好的泛化可以被编码到网络的结构中。我们设计了两个这样的易于使用的结构正则化,插入一个额外的\textit{generalization layer}到一个模型架构,一个跳过连接和另一个辍学。我们在各种前馈和卷积架构的实验中验证了我们的理论结果,包括ResNets和数据集(MNIST,CIFAR10,合成数据)。我们相信这项工作开辟了新的研究途径,以更好地推广架构。

英文摘要:

We take a geometrical viewpoint and present a unifying view on supervised deep learning with the Bregman divergence loss function - this entails frequent classification and prediction tasks. Motivated by simulations we suggest that there is principally no implicit bias of vanilla stochastic gradient descent training of deep models towards "simpler" functions. Instead, we show that good generalization may be instigated by bounded spectral products over layers leading to a novel geometric regularizer. It is revealed that in deep enough models such a regularizer enables both, extreme accuracy and generalization, to be reached. We associate popular regularization techniques like weight decay, drop out, batch normalization, and early stopping with this perspective. Backed up by theory we further demonstrate that "generalization by design" is practically possible and that good generalization may be encoded into the structure of the network. We design two such easy-to-use structural regularizers that insert an additional \textit{generalization layer} into a model architecture, one with a skip connection and another one with drop-out. We verify our theoretical results in experiments on various feedforward and convolutional architectures, including ResNets, and datasets (MNIST, CIFAR10, synthetic data). We believe this work opens up new avenues of research towards better generalizing architectures.

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

↑