Fully Bayesian VIB-DeepSSM.

Autor: Adams J; Scientific Computing and Imaging Institute, University of Utah, UT, USA.; Kahlert School of Computing, University of Utah, UT, USA., Elhabian SY; Scientific Computing and Imaging Institute, University of Utah, UT, USA.; Kahlert School of Computing, University of Utah, UT, USA.
Jazyk: angličtina
Zdroj: Medical image computing and computer-assisted intervention : MICCAI ... International Conference on Medical Image Computing and Computer-Assisted Intervention [Med Image Comput Comput Assist Interv] 2023 Oct; Vol. 14222, pp. 346-356. Date of Electronic Publication: 2023 Oct 01.
DOI: 10.1007/978-3-031-43898-1_34
Abstrakt: Statistical shape modeling (SSM) enables population-based quantitative analysis of anatomical shapes, informing clinical diagnosis. Deep learning approaches predict correspondence-based SSM directly from unsegmented 3D images but require calibrated uncertainty quantification, motivating Bayesian formulations. Variational information bottleneck DeepSSM (VIB-DeepSSM) is an effective, principled framework for predicting probabilistic shapes of anatomy from images with aleatoric uncertainty quantification. However, VIB is only half-Bayesian and lacks epistemic uncertainty inference. We derive a fully Bayesian VIB formulation and demonstrate the efficacy of two scalable implementation approaches: concrete dropout and batch ensemble. Additionally, we introduce a novel combination of the two that further enhances uncertainty calibration via multimodal marginalization. Experiments on synthetic shapes and left atrium data demonstrate that the fully Bayesian VIB network predicts SSM from images with improved uncertainty reasoning without sacrificing accuracy.
Databáze: MEDLINE