arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2102.09235cs.LGstat.ML
收藏

深度残差网络学习Wasserstein空间中的测地线

Deep Residual Networks Learn the Geodesic Curve in the Wasserstein Space

Shanghai Institute for Mathematics and Interdisciplinary Sciences (SIMIS)(上海数学与交叉学科研究院)

Kuo Gai, Shihua Zhang

AI总结:

通过连续性方程建模,发现带L2正则化的深度残差网络倾向于学习Wasserstein空间中的测地线,这解释了其优化和泛化优势。

摘要:

最近的研究揭示了深度神经网络(DNNs)与动态系统之间的数学联系。然而,DNNs,特别是深度残差网络(ResNets),在训练过程中倾向于学习的具体动力学仍未被充分表征。为此,我们使用连续性方程对深度残差网络的前向传播进行建模,其中测度是守恒的,并且测度空间中的无限曲线连接了ResNet的输入分布和输出分布。我们发现带有$L_2$正则化的ResNets试图学习由最优传输映射诱导的Wasserstein空间中的测地线。与普通网络相比,ResNets能更好地逼近测地线,这解释了为什么ResNets能够更好地优化和泛化。数值实验表明,从线形得分来看,ResNet的数据轨迹趋向于直线形状,并且从最优传输得分来看,ResNet学习的映射更接近最优传输映射。总之,我们得出结论:ResNets学习Wasserstein空间中的测地线,并在高维空间中离散地工程化数据变换。

英文摘要:

Recent studies revealed the mathematical connection between deep neural networks (DNNs) and dynamic systems. However, the specific dynamics that DNNs, especially deep residual networks (ResNets), tend to learn during training remain insufficiently characterized. To this end, we model the forward propagation of deep residual networks using continuity equations, in which the measure is conserved and infinite curves in the measure space connect the input distribution to the output one of a ResNet. We find ResNets with $L_2$ regularization attempt to learn the geodesic curve in the Wasserstein space, induced by the optimal transport map. Compared with plain networks, ResNets can better approximate the geodesic curve, which explains why ResNets can be optimized and generalize better. Numerical experiments show that the data tracks of a ResNet tend to be line-shaped in terms of the line-shape score, and the map learned by a ResNet is closer to the optimal transport map in terms of the optimal transport score. In a word, we conclude that ResNets learn the geodesic curve in the Wasserstein space and discretely engineer the data transformation in high-dimensional spaces.

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

↑