A Backward SDE Method for Uncertainty Quantification in Deep Learning

Richard Archibald, Feng Bao, Yanzhao Cao +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…