Bayesian PINNs for uncertainty-aware inverse problems (BPINN-IP)
Ali Mohammad-Djafari
TL;DR
BPINN-IP tackles linear inverse problems by merging physics-informed neural networks with hierarchical Bayesian inference to obtain point estimates and posterior uncertainty for reconstructions. The framework handles both supervised and unsupervised data and uses Bayesian updates for the latent variables, with uncertainty propagated through the forward model; NN weights are also treated in a Bayesian manner to yield predictive uncertainty. Applied to infrared image processing, BPINN-IP demonstrates deconvolution and super-resolution capabilities with accompanying uncertainty maps, using both synthetic and real data and showing robustness with limited labels. This approach offers a practical, uncertainty-aware tool for ill-posed inverse problems across imaging and related domains.
Abstract
The main contribution of this paper is to develop a hierarchical Bayesian formulation of PINNs for linear inverse problems, which is called BPINN-IP. The proposed methodology extends PINN to account for prior knowledge on the nature of the expected NN output, as well as its weights. Also, as we can have access to the posterior probability distributions, naturally uncertainties can be quantified. Also, variational inference and Monte Carlo dropout are employed to provide predictive means and variances for reconstructed images. Un example of applications to deconvolution and super-resolution is considered, details of the different steps of implementations are given, and some preliminary results are presented.
