Uncertainty quantification for White Matter Hyperintensity segmentation detects silent failures and improves automated Fazekas quantification
White Matter Hyperintensities (WMH) are key neuroradiological markers of small vessel disease present in brain MRI. Assessment of WMH is important in research and clinics. However, WMH are challenging to segment due to their high variability in shape, location, size, poorly defined borders, and simi...
Gespeichert in:
Hauptverfasser: | , , , , , , , , , , |
---|---|
Format: | Artikel |
Sprache: | eng |
Schlagworte: | |
Online-Zugang: | Volltext bestellen |
Tags: |
Tag hinzufügen
Keine Tags, Fügen Sie den ersten Tag hinzu!
|
Zusammenfassung: | White Matter Hyperintensities (WMH) are key neuroradiological markers of
small vessel disease present in brain MRI. Assessment of WMH is important in
research and clinics. However, WMH are challenging to segment due to their high
variability in shape, location, size, poorly defined borders, and similar
intensity profile to other pathologies (e.g stroke lesions) and artefacts (e.g
head motion). In this work, we apply the most effective techniques for
uncertainty quantification (UQ) in segmentation to the WMH segmentation task
across multiple test-time data distributions. We find a combination of
Stochastic Segmentation Networks with Deep Ensembles yields the highest Dice
and lowest Absolute Volume Difference % (AVD) score on in-domain and
out-of-distribution data. We demonstrate the downstream utility of UQ,
proposing a novel method for classification of the clinical Fazekas score using
spatial features extracted for WMH segmentation and UQ maps. We show that
incorporating WMH uncertainty information improves Fazekas classification
performance and calibration, with median class balanced accuracy for
classification models with (UQ and spatial WMH features)/(spatial WMH
features)/(WMH volume only) of 0.71/0.66/0.60 in the Deep WMH and
0.82/0.77/0.73 in the Periventricular WMH regions respectively. We demonstrate
that stochastic UQ techniques with high sample diversity can improve the
detection of poor quality segmentations. Finally, we qualitatively analyse the
semantic information captured by UQ techniques and demonstrate that uncertainty
can highlight areas where there is ambiguity between WMH and stroke lesions,
while identifying clusters of small WMH in deep white matter unsegmented by the
model. |
---|---|
DOI: | 10.48550/arxiv.2411.17571 |