Abstract:Statistical shape modeling (SSM) directly from 3D medical images is an underutilized tool for detecting pathology, diagnosing disease, and conducting population-level morphology analysis. Deep learning frameworks have increased the feasibility of adopting SSM in medical practice by reducing the expert-driven manual and computational overhead in traditional SSM workflows. However, translating such frameworks to clinical practice requires calibrated uncertainty measures as neural networks can produce over-confident predictions that cannot be trusted in sensitive clinical decision-making. Existing techniques for predicting shape with aleatoric (data-dependent) uncertainty utilize a principal component analysis (PCA) based shape representation computed in isolation of the model training. This constraint restricts the learning task to solely estimating pre-defined shape descriptors from 3D images and imposes a linear relationship between this shape representation and the output (i.e., shape) space. In this paper, we propose a principled framework based on the variational information bottleneck theory to relax these assumptions while predicting probabilistic shapes of anatomy directly from images without supervised encoding of shape descriptors. Here, the latent representation is learned in the context of the learning task, resulting in a more scalable, flexible model that better captures data non-linearity. Additionally, this model is self-regularized and generalizes better given limited training data. Our experiments demonstrate that the proposed method provides an accuracy improvement and better calibrated aleatoric uncertainty estimates than state-of-the-art methods.

State Estimation with Model Reduction and Shape Variability. Application to biomedical problems

Model reduction of coupled systems based on non-intrusive approximations of the boundary response maps

Left Ventricular Motion Estimation under Stochastic Uncertainties

A Simultaneous Framework for Recovering Three Dimensional Shape and Nonrigid Motion from Cardiac Image Sequences

Model order reduction by convex displacement interpolation

Data-driven reduced-order modelling for blood flow simulations with geometry-informed snapshots

Subject Specific Shape Modeling with Incremental Mixture Models

Parametric/Stochastic Model Reduction: Low-Rank Representation, Non-Intrusive Bi-Fidelity Approximation, and Convergence Analysis

From Images to Probabilistic Anatomical Shapes: A Deep Variational Bottleneck Approach

Efficient Sparse Shape Composition with Its Applications in Biomedical Image Analysis: an Overview

Model Reduction for Large Scale Systems

Dictionary-based model reduction for state estimation

Multiscale model reduction method for Bayesian inverse problems of subsurface flow

A model reduction approach for inverse problems with operator valued data

Data-driven structure-preserving model reduction for stochastic Hamiltonian systems

Low-dimensional Data-based Surrogate Model of a Continuum-mechanical Musculoskeletal System Based on Non-intrusive Model Order Reduction

Mathematical Analysis of a Model-Constrained Inverse Problem For the Reconstruction of Early States of Prostate Cancer Growth

Super-Resolution Surface Reconstruction from Few Low-Resolution Slices

Numerical modelling of a peripheral arterial stenosis using dimensionally reduced models and kernel methods

An Inter-Landmark Approach to 4-D Shape Extraction and Interpretation: Application to Myocardial Motion Assessment in Mri