Propagating uncertainty across cascaded medical imaging tasks for improved deep learning inference
Although deep networks have been shown to perform very well on a variety of medical imaging tasks, inference in the presence of pathology presents several challenges to common models. These challenges impede the integration of deep learning models into real clinical workflows, where the customary pr...
Main Authors: | , , , , , , , , , |
---|---|
Format: | Journal article |
Language: | English |
Published: |
IEEE
2021
|
_version_ | 1797108097303445504 |
---|---|
author | Mehta, R Christinck, T Nair, T Bussy, A Premasiri, S Costantino, M Chakravarthy, MM Arnold, DL Gal, Y Arbel, T |
author_facet | Mehta, R Christinck, T Nair, T Bussy, A Premasiri, S Costantino, M Chakravarthy, MM Arnold, DL Gal, Y Arbel, T |
author_sort | Mehta, R |
collection | OXFORD |
description | Although deep networks have been shown to perform very well on a variety of medical imaging tasks, inference in the presence of pathology presents several challenges to common models. These challenges impede the integration of deep learning models into real clinical workflows, where the customary process of cascading deterministic outputs from a sequence of image-based inference steps (e.g. registration, segmentation) generally leads to an accumulation of errors that impacts the accuracy of downstream inference tasks. In this paper, we propose that by embedding uncertainty estimates across cascaded inference tasks, performance on the downstream inference tasks should be improved. We demonstrate the effectiveness of the proposed approach in three different clinical contexts: (i) We demonstrate that by propagating T2 weighted lesion segmentation results and their associated uncertainties, subsequent T2 lesion detection performance is improved when evaluated on a proprietary large-scale, multi-site, clinical trial dataset acquired from patients with Multiple Sclerosis. (ii) We show an improvement in brain tumour segmentation performance when the uncertainty map associated with a synthesised missing MR volume is provided as an additional input to a follow-up brain tumour segmentation network, when evaluated on the publicly available BraTS-2018 dataset. (iii) We show that by propagating uncertainties from a voxel-level hippocampus segmentation task, the subsequent regression of the Alzheimer's disease clinical score is improved. |
first_indexed | 2024-03-07T07:22:53Z |
format | Journal article |
id | oxford-uuid:6d52bc3d-dc66-44a4-b3b2-d598913b67d1 |
institution | University of Oxford |
language | English |
last_indexed | 2024-03-07T07:22:53Z |
publishDate | 2021 |
publisher | IEEE |
record_format | dspace |
spelling | oxford-uuid:6d52bc3d-dc66-44a4-b3b2-d598913b67d12022-11-04T12:05:13ZPropagating uncertainty across cascaded medical imaging tasks for improved deep learning inference Journal articlehttp://purl.org/coar/resource_type/c_dcae04bcuuid:6d52bc3d-dc66-44a4-b3b2-d598913b67d1EnglishSymplectic ElementsIEEE2021Mehta, RChristinck, TNair, TBussy, APremasiri, SCostantino, MChakravarthy, MMArnold, DLGal, YArbel, TAlthough deep networks have been shown to perform very well on a variety of medical imaging tasks, inference in the presence of pathology presents several challenges to common models. These challenges impede the integration of deep learning models into real clinical workflows, where the customary process of cascading deterministic outputs from a sequence of image-based inference steps (e.g. registration, segmentation) generally leads to an accumulation of errors that impacts the accuracy of downstream inference tasks. In this paper, we propose that by embedding uncertainty estimates across cascaded inference tasks, performance on the downstream inference tasks should be improved. We demonstrate the effectiveness of the proposed approach in three different clinical contexts: (i) We demonstrate that by propagating T2 weighted lesion segmentation results and their associated uncertainties, subsequent T2 lesion detection performance is improved when evaluated on a proprietary large-scale, multi-site, clinical trial dataset acquired from patients with Multiple Sclerosis. (ii) We show an improvement in brain tumour segmentation performance when the uncertainty map associated with a synthesised missing MR volume is provided as an additional input to a follow-up brain tumour segmentation network, when evaluated on the publicly available BraTS-2018 dataset. (iii) We show that by propagating uncertainties from a voxel-level hippocampus segmentation task, the subsequent regression of the Alzheimer's disease clinical score is improved. |
spellingShingle | Mehta, R Christinck, T Nair, T Bussy, A Premasiri, S Costantino, M Chakravarthy, MM Arnold, DL Gal, Y Arbel, T Propagating uncertainty across cascaded medical imaging tasks for improved deep learning inference |
title | Propagating uncertainty across cascaded medical imaging tasks for improved deep learning inference
|
title_full | Propagating uncertainty across cascaded medical imaging tasks for improved deep learning inference
|
title_fullStr | Propagating uncertainty across cascaded medical imaging tasks for improved deep learning inference
|
title_full_unstemmed | Propagating uncertainty across cascaded medical imaging tasks for improved deep learning inference
|
title_short | Propagating uncertainty across cascaded medical imaging tasks for improved deep learning inference
|
title_sort | propagating uncertainty across cascaded medical imaging tasks for improved deep learning inference |
work_keys_str_mv | AT mehtar propagatinguncertaintyacrosscascadedmedicalimagingtasksforimproveddeeplearninginference AT christinckt propagatinguncertaintyacrosscascadedmedicalimagingtasksforimproveddeeplearninginference AT nairt propagatinguncertaintyacrosscascadedmedicalimagingtasksforimproveddeeplearninginference AT bussya propagatinguncertaintyacrosscascadedmedicalimagingtasksforimproveddeeplearninginference AT premasiris propagatinguncertaintyacrosscascadedmedicalimagingtasksforimproveddeeplearninginference AT costantinom propagatinguncertaintyacrosscascadedmedicalimagingtasksforimproveddeeplearninginference AT chakravarthymm propagatinguncertaintyacrosscascadedmedicalimagingtasksforimproveddeeplearninginference AT arnolddl propagatinguncertaintyacrosscascadedmedicalimagingtasksforimproveddeeplearninginference AT galy propagatinguncertaintyacrosscascadedmedicalimagingtasksforimproveddeeplearninginference AT arbelt propagatinguncertaintyacrosscascadedmedicalimagingtasksforimproveddeeplearninginference |