Uncertainty Quantification in Deep Learning through Stochastic Maximum Principle

Richard Archibald, Feng Bao, Yanzhao Cao, He Zhang

Published 2020-11-28Version 1

We develop a probabilistic machine learning method, which formulates a class of stochastic neural networks by a stochastic optimal control problem. An efficient stochastic gradient descent algorithm is introduced under the stochastic maximum principle framework. Convergence analysis for stochastic gradient descent optimization and numerical experiments for applications of stochastic neural networks are carried out to validate our methodology in both theory and performance.