A Heteroscedastic Uncertainty Model for Decoupling Sources of MRI Image Quality

Richard Shaw10000-0002-3141-5562, Carole H. Sudre2, Sebastien Ourselin2, M. Jorge Cardoso2, Hugh G. Pemberton1
1: University College London, UK, 2: King’s College London, UK
Publication date: 2021/09/07
PDF · Video · arXiv


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.


uncertainty · student teacher networks · bayesian deep learning · augmentation · quality control · artefacts · mri · deep learning · machine learning

Bibtex @article{melba:2021:010:shaw, title = "A Heteroscedastic Uncertainty Model for Decoupling Sources of MRI Image Quality", author = "Shaw, Richard and Sudre, Carole H. and Ourselin, Sebastien and Cardoso, M. Jorge and Pemberton, Hugh G.", journal = "Machine Learning for Biomedical Imaging", volume = "1", issue = "MIDL 2020 special issue", year = "2021", pages = "1--23", issn = "2766-905X", doi = "https://doi.org/10.59275/j.melba.2021-8678", url = "https://melba-journal.org/2021:010" }
RISTY - JOUR AU - Shaw, Richard AU - Sudre, Carole H. AU - Ourselin, Sebastien AU - Cardoso, M. Jorge AU - Pemberton, Hugh G. PY - 2021 TI - A Heteroscedastic Uncertainty Model for Decoupling Sources of MRI Image Quality T2 - Machine Learning for Biomedical Imaging VL - 1 IS - MIDL 2020 special issue SP - 1 EP - 23 SN - 2766-905X DO - https://doi.org/10.59275/j.melba.2021-8678 UR - https://melba-journal.org/2021:010 ER -

2021:010 cover