Propagation and Attribution of Uncertainty in Medical Imaging Pipelines

التفاصيل البيبلوغرافية
العنوان: Propagation and Attribution of Uncertainty in Medical Imaging Pipelines
المؤلفون: Feiner, Leonhard F., Menten, Martin J., Hammernik, Kerstin, Hager, Paul, Huang, Wenqi, Rueckert, Daniel, Braren, Rickmer F., Kaissis, Georgios
سنة النشر: 2023
المجموعة: Computer Science
مصطلحات موضوعية: Computer Science - Computer Vision and Pattern Recognition, Computer Science - Machine Learning
الوصف: Uncertainty estimation, which provides a means of building explainable neural networks for medical imaging applications, have mostly been studied for single deep learning models that focus on a specific task. In this paper, we propose a method to propagate uncertainty through cascades of deep learning models in medical imaging pipelines. This allows us to aggregate the uncertainty in later stages of the pipeline and to obtain a joint uncertainty measure for the predictions of later models. Additionally, we can separately report contributions of the aleatoric, data-based, uncertainty of every component in the pipeline. We demonstrate the utility of our method on a realistic imaging pipeline that reconstructs undersampled brain and knee magnetic resonance (MR) images and subsequently predicts quantitative information from the images, such as the brain volume, or knee side or patient's sex. We quantitatively show that the propagated uncertainty is correlated with input uncertainty and compare the proportions of contributions of pipeline stages to the joint uncertainty measure.
نوع الوثيقة: Working Paper
DOI: 10.1007/978-3-031-44336-7_1
URL الوصول: http://arxiv.org/abs/2309.16831
رقم الأكسشن: edsarx.2309.16831
قاعدة البيانات: arXiv