Uncertainty propagation through trained multi-layer perceptrons: Exact analytical results
Andrew Thompson, Miles McCrory
TL;DR
This paper derives exact analytical expressions for propagating uncertainty through a trained single-hidden-layer MLP with ReLU activations when the input is multivariate Gaussian. By obtaining closed-form moments of the rectified Gaussian hidden activations—specifically $\mathbb{E}[X_i]$, $\mathbb{E}[X_i^2]$, and $\mathbb{E}[X_i X_j]$—the authors compute the output moments $E[Y]$ and $Var[Y]$ exactly as $E[Y] = \beta^T \gamma + d$ and $Var[Y] = \beta^T \Gamma \beta$. These results rely on standard univariate and bivariate Gaussian integrals and avoid infinite series, in contrast to prior work. The analytical expressions are validated against Monte Carlo simulations on an EIS-based state-of-health (SOH) prediction problem, demonstrating convergence consistent with $1/\sqrt{n}$. The work offers a transparent, accurate, and reproducible method for uncertainty quantification in regression models and discusses extensions to other activations and deeper networks as future directions.
Abstract
We give analytical results for propagation of uncertainty through trained multi-layer perceptrons (MLPs) with a single hidden layer and ReLU activation functions. More precisely, we give expressions for the mean and variance of the output when the input is multivariate Gaussian. In contrast to previous results, we obtain exact expressions without resort to a series expansion.
