arXiv · 2011.14145
A Backward SDE Method for Uncertainty Quantification in Deep Learning
Abstract
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. Numerical experiments for applications of stochastic neural networks are carried out to validate the effectiveness of our methodology.
Explore related subjects
Keep this discovery
Richard Archibald, Feng Bao, Yanzhao Cao, He Zhang. 2020-11-28. A Backward SDE Method for Uncertainty Quantification in Deep Learning. https://arxiv.org/abs/2011.14145
Cite the original work for its findings. Save a collection to share your selection of sources.