Mathematics Colloquium

Yanzhao Cao, Auburn University
Uncertainty quantification of deep neural networks

In this talk, I will first give a brief mathematical introduction to deep learning. Then I will talk about a recent work on uncertainty quantification  (UQ) of deep learning. Uncertainty quantification of deep neural networks (DNN) is a very important issue in deep learning.  In our UQ for DNN  framework, the DNN architecture is the neural ordinary differential equations (Neural-ODE), which formulates the evolution of potentially huge hidden layers in the DNN as a discretized ordinary differential equation (ODE) system. To characterize the randomness caused by the uncertainty of models and noises of data, we add a multiplicative Brownian motion noise to the ODE as a stochastic diffusion term, which changes the ODE to a stochastic differential equation (SDE), and the deterministic DNN becomes a stochastic neural network (SNN) In the SNN, the drift parameters serve as the prediction of the network, and the stochastic diffusion governs the randomness of network output, which serves to quantify the epistemic uncertainty of deep learning.  I will present results on convergence as well as numerical experiments. 

Event Date: 
May 6, 2021 - 3:30pm
Calendar Category: