改进Adam优化器,让深度学习解微分方程和最优控制更快更准。
Averaged Adam accelerates stochastic optimization in the training of deep neural network approximations for partial differential equation and optimal control problems
- 用平均化思想改进Adam,提升训练稳定性与收敛速度。
- 在各类微分方程与最优控制问题中,性能优于标准Adam和SGD。
- 适合做科学计算中的深度学习建模,尤其适用于复杂物理系统模拟。
深度学习方法(通常由深度神经网络DNN在随机梯度下降法SGD训练下实现)现广泛应用于数据驱动学习及科学计算任务,如最优控制(OC)和偏微分方程(PDE)问题。实际应用中常采用比基础SGD更先进的自适应加速优化器,如流行的Adam。受经典Polyak-Ruppert平均思想启发,本文将平均版Adam用于训练DNN求解典型科学计算问题——如PDE与OC问题。实验涵盖物理信息神经网络(PINN)、深度反向随机微分方程(deep BSDE)、基于DNN的热方程、Black-Scholes、Burgers与Allen-Cahn等方程近似,以及最优控制与图像分类(ResNet+ CIFAR-10)。所有案例中,平均版Adam均优于标准Adam与标准SGD,尤其在科学机器学习任务中表现突出。相关代码已开源于GitHub:https://github.com/deeplearningmethods/averaged-adam。
原文摘要 · Abstract (English)
Deep learning methods - usually consisting of a class of deep neural networks (DNNs) trained by a stochastic gradient descent (SGD) optimization method - are nowadays omnipresent in data-driven learning problems as well as in scientific computing tasks such as optimal control (OC) and partial differential equation (PDE) problems. In practically relevant learning tasks, often not the plain-vanilla standard SGD optimization method is employed to train the considered class of DNNs but instead more sophisticated adaptive and accelerated variants of the standard SGD method such as the popular Adam optimizer are used. Inspired by the classical Polyak-Ruppert averaging approach, in this work we apply averaged variants of the Adam optimizer to train DNNs to approximately solve exemplary scientific computing problems in the form of PDEs and OC problems. We test the averaged variants of Adam in a series of learning problems including physics-informed neural network (PINN), deep backward stochastic differential equation (deep BSDE), and deep Kolmogorov approximations for PDEs (such as heat, Black-Scholes, Burgers, and Allen-Cahn PDEs), including DNN approximations for OC problems, and including DNN approximations for image classification problems (ResNet for CIFAR-10). In each of the numerical examples the employed averaged variants of Adam outperform the standard Adam and the standard SGD optimizers, particularly, in the situation of the scientific machine learning problems. The Python source codes for the numerical experiments associated to this work can be found on GitHub at https://github.com/deeplearningmethods/averaged-adam.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。