A Decoupled Uncertainty Model for MRI Segmentation Quality Estimation
- URL: http://arxiv.org/abs/2109.02413v1
- Date: Mon, 6 Sep 2021 12:54:44 GMT
- Title: A Decoupled Uncertainty Model for MRI Segmentation Quality Estimation
- Authors: Richard Shaw and Carole H. Sudre and Sebastien Ourselin and M. Jorge
Cardoso and Hugh G. Pemberton
- Abstract summary: We propose a novel CNN architecture to decouple sources of uncertainty related to the task and different k-space artefacts.
We show that our uncertainty predictions provide a better estimate of MRI quality from the point of view of the task.
- Score: 4.104181348044472
- License: http://arxiv.org/licenses/nonexclusive-distrib/1.0/
- Abstract: Quality control (QC) of MR images is essential to ensure that downstream
analyses such as segmentation can be performed successfully. Currently, QC is
predominantly performed visually and subjectively, at significant time and
operator cost. We aim to automate the process using a probabilistic network
that estimates segmentation uncertainty through a heteroscedastic noise model,
providing a measure of task-specific quality. By augmenting training images
with k-space artefacts, we propose a novel CNN architecture to decouple sources
of uncertainty related to the task and different k-space artefacts in a
self-supervised manner. This enables the prediction of separate uncertainties
for different types of data degradation. While the uncertainty predictions
reflect the presence and severity of artefacts, the network provides more
robust and generalisable segmentation predictions given the quality of the
data. We show that models trained with artefact augmentation provide
informative measures of uncertainty on both simulated artefacts and problematic
real-world images identified by human raters, both qualitatively and
quantitatively in the form of error bars on volume measurements. Relating
artefact uncertainty to segmentation Dice scores, we observe that our
uncertainty predictions provide a better estimate of MRI quality from the point
of view of the task (gray matter segmentation) compared to commonly used
metrics of quality including signal-to-noise ratio (SNR) and contrast-to-noise
ratio (CNR), hence providing a real-time quality metric indicative of
segmentation quality.
Related papers
- Adaptive Feature Selection for No-Reference Image Quality Assessment by Mitigating Semantic Noise Sensitivity [55.399230250413986]
We propose a Quality-Aware Feature Matching IQA Metric (QFM-IQM) to remove harmful semantic noise features from the upstream task.
Our approach achieves superior performance to the state-of-the-art NR-IQA methods on eight standard IQA datasets.
arXiv Detail & Related papers (2023-12-11T06:50:27Z) - Comparing the quality of neural network uncertainty estimates for
classification problems [0.0]
Uncertainty quantification (UQ) methods for deep learning (DL) models have received increased attention in the literature.
We use statistical methods of frequentist interval coverage and interval width to evaluate the quality of credible intervals.
We apply these different UQ for DL methods to a hyperspectral image target detection problem and show the inconsistency of the different methods' results.
arXiv Detail & Related papers (2023-08-11T01:55:14Z) - DifFIQA: Face Image Quality Assessment Using Denoising Diffusion
Probabilistic Models [1.217503190366097]
Face image quality assessment (FIQA) techniques aim to mitigate these performance degradations.
We present a powerful new FIQA approach, named DifFIQA, which relies on denoising diffusion probabilistic models (DDPM)
Because the diffusion-based perturbations are computationally expensive, we also distill the knowledge encoded in DifFIQA into a regression-based quality predictor, called DifFIQA(R)
arXiv Detail & Related papers (2023-05-09T21:03:13Z) - Towards Reliable Medical Image Segmentation by utilizing Evidential Calibrated Uncertainty [52.03490691733464]
We introduce DEviS, an easily implementable foundational model that seamlessly integrates into various medical image segmentation networks.
By leveraging subjective logic theory, we explicitly model probability and uncertainty for the problem of medical image segmentation.
DeviS incorporates an uncertainty-aware filtering module, which utilizes the metric of uncertainty-calibrated error to filter reliable data.
arXiv Detail & Related papers (2023-01-01T05:02:46Z) - Uncertainty-Driven Action Quality Assessment [67.20617610820857]
We propose a novel probabilistic model, named Uncertainty-Driven AQA (UD-AQA), to capture the diversity among multiple judge scores.
We generate the estimation of uncertainty for each prediction, which is employed to re-weight AQA regression loss.
Our proposed method achieves competitive results on three benchmarks including the Olympic events MTL-AQA and FineDiving, and the surgical skill JIGSAWS datasets.
arXiv Detail & Related papers (2022-07-29T07:21:15Z) - Textural-Structural Joint Learning for No-Reference Super-Resolution
Image Quality Assessment [59.91741119995321]
We develop a dual stream network to jointly explore the textural and structural information for quality prediction, dubbed TSNet.
By mimicking the human vision system (HVS) that pays more attention to the significant areas of the image, we develop the spatial attention mechanism to make the visual-sensitive areas more distinguishable.
Experimental results show the proposed TSNet predicts the visual quality more accurate than the state-of-the-art IQA methods, and demonstrates better consistency with the human's perspective.
arXiv Detail & Related papers (2022-05-27T09:20:06Z) - Image Quality Assessment using Contrastive Learning [50.265638572116984]
We train a deep Convolutional Neural Network (CNN) using a contrastive pairwise objective to solve the auxiliary problem.
We show through extensive experiments that CONTRIQUE achieves competitive performance when compared to state-of-the-art NR image quality models.
Our results suggest that powerful quality representations with perceptual relevance can be obtained without requiring large labeled subjective image quality datasets.
arXiv Detail & Related papers (2021-10-25T21:01:00Z) - Towards Reducing Aleatoric Uncertainty for Medical Imaging Tasks [5.220940151628734]
Uncertainty in predictions can be attributed to noise or randomness in data (aleatoric) and incorrect model inferences (epistemic)
This work proposes a novel approach that interprets data uncertainty estimated from a self-supervised task as noise inherent to the data.
Our findings demonstrate the effectiveness of the proposed approach in significantly reducing the aleatoric uncertainty in the image segmentation task.
arXiv Detail & Related papers (2021-10-21T09:31:00Z) - Estimating MRI Image Quality via Image Reconstruction Uncertainty [4.483523280360846]
We train CNNs using a heteroscedastic uncertainty model to recover clean images from noisy data.
We argue that quality control for visual assessment cannot be equated to quality control for algorithmic processing.
arXiv Detail & Related papers (2021-06-21T11:22:17Z) - Approaching Neural Network Uncertainty Realism [53.308409014122816]
Quantifying or at least upper-bounding uncertainties is vital for safety-critical systems such as autonomous vehicles.
We evaluate uncertainty realism -- a strict quality criterion -- with a Mahalanobis distance-based statistical test.
We adopt it to the automotive domain and show that it significantly improves uncertainty realism compared to a plain encoder-decoder model.
arXiv Detail & Related papers (2021-01-08T11:56:12Z) - A Heteroscedastic Uncertainty Model for Decoupling Sources of MRI Image
Quality [3.5480752735999417]
Quality control (QC) of medical images is essential to ensure that downstream analyses such as segmentation can be performed successfully.
We aim to automate the process by formulating a probabilistic network that estimates uncertainty through a heteroscedastic noise model.
We show models trained with simulated artefacts provide informative measures of uncertainty on real-world images and we validate our uncertainty predictions on problematic images identified by human-raters.
arXiv Detail & Related papers (2020-01-31T16:04:41Z)
This list is automatically generated from the titles and abstracts of the papers in this site.
This site does not guarantee the quality of this site (including all information) and is not responsible for any consequences.