Automatic quality control framework for more reliable integration of machine learning-based image segmentation into medical workflows
Machine learning algorithms underpin modern diagnostic-aiding software, which has proved valuable in clinical practice, particularly in radiology. However, inaccuracies, mainly due to the limited availability of clinical samples for training these algorithms, hamper their wider applicability, accept...
Saved in:
Main Authors | , , , , , , , , , , , |
---|---|
Format | Journal Article |
Language | English |
Published |
06.12.2021
|
Subjects | |
Online Access | Get full text |
Cover
Loading…
Summary: | Machine learning algorithms underpin modern diagnostic-aiding software, which
has proved valuable in clinical practice, particularly in radiology. However,
inaccuracies, mainly due to the limited availability of clinical samples for
training these algorithms, hamper their wider applicability, acceptance, and
recognition amongst clinicians. We present an analysis of state-of-the-art
automatic quality control (QC) approaches that can be implemented within these
algorithms to estimate the certainty of their outputs. We validated the most
promising approaches on a brain image segmentation task identifying white
matter hyperintensities (WMH) in magnetic resonance imaging data. WMH are a
correlate of small vessel disease common in mid-to-late adulthood and are
particularly challenging to segment due to their varied size, and
distributional patterns. Our results show that the aggregation of uncertainty
and Dice prediction were most effective in failure detection for this task.
Both methods independently improved mean Dice from 0.82 to 0.84. Our work
reveals how QC methods can help to detect failed segmentation cases and
therefore make automatic segmentation more reliable and suitable for clinical
practice. |
---|---|
DOI: | 10.48550/arxiv.2112.03277 |