arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2109.06294math.OCcs.LGmath.PR
收藏

深度神经网络分布式鲁棒学习的正则化风险

On the regularized risk of distributionally robust learning over deep neural networks

University College London(伦敦大学学院);University of Wisconsin-Madison(威斯康星大学麦迪逊分校)

Camilo Garcia Trillos, Nicolas Garcia Trillos

摘要:

在本文中,我们探讨了分布式鲁棒学习与不同形式的正则化之间的关系,以增强深度神经网络的鲁棒性。特别是,从一个具体的最小-最大分布鲁棒问题,并使用最优输运理论的工具,我们得到一阶和二阶近似的分布鲁棒问题的适当的正则化风险最小化问题。在深度ResNet模型的背景下,我们将由此产生的正则化问题的结构确定为平均场最优控制问题,其中状态变量的数量和维度在原始非鲁棒问题的维度的无量纲因子内。使用庞特里亚金最大值原理与这些问题,我们激励一个家庭的可扩展算法的强大的神经网络的训练。我们的分析恢复了文献中已知的一些结果和算法(在整个论文中解释的设置),并提供了许多其他理论和算法的见解,据我们所知是新颖的。在我们的分析中,我们使用了我们认为对未来分析更一般的对抗性学习问题有用的工具。

英文摘要:

In this paper we explore the relation between distributionally robust learning and different forms of regularization to enforce robustness of deep neural networks. In particular, starting from a concrete min-max distributionally robust problem, and using tools from optimal transport theory, we derive first order and second order approximations to the distributionally robust problem in terms of appropriate regularized risk minimization problems. In the context of deep ResNet models, we identify the structure of the resulting regularization problems as mean-field optimal control problems where the number and dimension of state variables is within a dimension-free factor of the dimension of the original unrobust problem. Using the Pontryagin maximum principles associated to these problems we motivate a family of scalable algorithms for the training of robust neural networks. Our analysis recovers some results and algorithms known in the literature (in settings explained throughout the paper) and provides many other theoretical and algorithmic insights that to our knowledge are novel. In our analysis we employ tools that we deem useful for a future analysis of more general adversarial learning problems.

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

↑