Get our free extension to see links to code for papers anywhere online!

Chrome logo Add to Chrome

Firefox logo Add to Firefox

Uncertainty Quantification in Deep Learning through Stochastic Maximum Principle

Nov 28, 2020
Richard Archibald, Feng Bao, Yanzhao Cao, He Zhang

Share this with someone who'll enjoy it:

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.

   Access Paper Source

Share this with someone who'll enjoy it: