Introducing µGUIDE for quantitative imaging via generalized uncertainty-driven inference using deep learning
- Cardiff University Brain Research Imaging Centre (CUBRIC), Cardiff University, United Kingdom
- School of Computer Science and Informatics, Cardiff University, United Kingdom
Figures
Figure 1
µGUIDE framework.
µGUIDE takes as input an observed data vector and relies on the definition of a biophysical or computational model (Ascoli et al., 2007; Callaghan et al., 2020; Jelescu et al., 2020). It outputs a …
Figure 2
µGUIDE summarizes information contained in the estimated posterior distributions.
(A) Examples of degenerate and non-degenerate posterior distributions. Two Gaussian distributions are fitted to the obtained posterior distribution, where the means and standard deviations are …
Figure 3
Comparison between µGUIDE and Markov-Chain-Monte-Carlo (MCMC).
(A) Posterior distributions obtained using either µGUIDE or MCMC on three exemplar simulations with Model 2 (SM − ). Names of the model parameters are indicated in the titles of the panels. (B) …
Figure 4
Fitting accuracy comparison between µGUIDE’s Multi-Layer Perceptron (MLP)-extracted features and manually defined summary statistics.
Maximum A Posterioris (MAPs) extracted from the posterior distributions versus ground truth parameters used for generating the signal for the three models. Orange points correspond to the MAPs …
Figure 5
Exemplar posterior distributions of the microstructure parameters for the Ball&Stick, Standard Model (SM), and extended-SANDI models, obtained using µGUIDE on exemplar noise-free simulations.
As the complexity of the model increases, degeneracies appear (red posterior distributions). µGUIDE allows to highlight those degeneracies present in the model definition.
Figure 6
Parametric maps of the Ball&Stick (top), SM (middle) and extended-SANDI model (bottom), obtained using µGUIDE.
Maximum A Posteriori (MAP), uncertainty and ambiguity measure maps are reported, overlayed with voxels considered degenerate (red dots).
Figure 7
Parametric maps of a participant with epilepsy obtained using µGUIDE with the Standard Model (SM), superimposed with the grey matter (black) and white matter (white) lesions segmentation.
Mean values of the Maximum A Posterior (MAP), uncertainty, and ambiguity measures are reported in the two regions of interest. Lower MAP values are obtained in the lesions for the axonal signal …
Figure 8
Example training set and input signals for µGUIDE.
(A) Examples of input synthetic data vectors and corresponding ground truth model parameters used in the training set of Model 1 (Ball&Stick). (B) Example of input measured signals from a voxel in a …
Appendix 1—figure 1
Correlation matrices between features extracted by the Multi-Layer Perceptron (MLP) in µGUIDE and manually defined summary features for the three models.
Appendix 2—figure 1
SNR uncertainty comparison between signals with different noise levels: no noise, , and using Model 2.
(A) Posterior distributions obtained on one example parameter combination (vertical black dashed line) with the three noise levels. (B) Histogram of the uncertainty obtained for 1000 signals with …
Appendix 3—figure 1
Schematic of Masked Autoencoder for Distribution Estimation (MADE) autoregressive network construction.
Appendix 4—figure 1
Posterior predictive checks.
Comparison between signals generated using random parameter combinations and their reconstructions using samples from .
Tables
Table 1
Number of degenerate cases per parameter on 10,000 noise-free simulations.
Training and estimations of the posterior distributions were performed on CPU. Time for training each model and time for estimating posterior distributions of 10,000 noise-free simulations, define …
| Model (SNR = ∞) | Training time (CPU) | Fitting time (on 10,000 simulations) | Number of degeneracies (on 10,000 simulations) | |||||||
|---|---|---|---|---|---|---|---|---|---|---|
| Model 1: Ball&Stick | 11 min | 96 s | 0 | 0 | 0 | - | - | - | - | - |
| Model 2: Standard Model | 2h02 | 135 s | 4 | 34 | 23 | 3 | 8 | - | - | - |
| Model 3: extended-SANDI model | 2h02 | 1412 s | 205 | 4 | 260 | 57 | - | 1395 | 2571 | 1011 |
Table 2
Number of degenerate cases per parameter on 10,000 noisy simulations (Rician noise with SNR = 50).
Training and estimations of the posterior distributions were performed using a GPU. Time for training each model and time for estimating posterior distributions of 10,000 noisy simulations, define …
| Model (SNR = 50) | Training time (CPU) | Fitting time (on 10,000 simulations) | Number of degeneracies (on 10,000 simulations) | |||||||
|---|---|---|---|---|---|---|---|---|---|---|
| Model 1: Ball&Stick | 26 min | 79 s | 0 | 0 | 0 | - | - | - | - | - |
| Model 2: Standard Model | 42 min | 82 s | 75 | 71 | 117 | 109 | 29 | - | - | - |
| Model 3: Extended-SANDI model | 50 min | 238 s | 47 | 24 | 784 | 6 | - | 828 | 1047 | 56 |
