Influence of uncertainty estimation techniques on false-positive reduction in liver lesion detection

Ishaan Bhat10000-0002-4398-018X, Josien P.W. Pluim2, Max A. Viergever1, Hugo J. Kuijf10000-0001-6997-9059
1: University Medical Center Utrecht, 2: Eindhoven University of Technology
December 2022 issue
Publication date: 2022/12/15
PDF · arXiv

Abstract

Deep learning techniques show success in detecting objects in medical images, but still suffer from false-positive predictions that may hinder accurate diagnosis. The estimated uncertainty of the neural network output has been used to flag incorrect predictions. We study the role played by features computed from neural network uncertainty estimates and shape-based features computed from binary predictions in reducing false positives in liver lesion detection by developing a classification-based post-processing step for different uncertainty estimation methods. We demonstrate an improvement in the lesion detection performance of the neural network (with respect to F1-score) for all uncertainty estimation methods on two datasets, comprising abdominal MR and CT images respectively. We show that features computed from neural network uncertainty estimates tend not to contribute much toward reducing false positives. Our results show that factors like class imbalance (true over false positive ratio) and shape-based features extracted from uncertainty maps play an important role in distinguishing false positive from true positive predictions.

Keywords

Deep learning · Uncertainty estimation · False-positive reduction

Bibtex @article{melba:2022:030:bhat, title = "Influence of uncertainty estimation techniques on false-positive reduction in liver lesion detection", author = "Bhat, Ishaan and Pluim, Josien P.W. and Viergever, Max A. and Kuijf, Hugo J.", journal = "Machine Learning for Biomedical Imaging", volume = "1", issue = "December 2022 issue", year = "2022", pages = "1--33", issn = "2766-905X", url = "https://melba-journal.org/2022:030" }
RISTY - JOUR AU - Bhat, Ishaan AU - Pluim, Josien P.W. AU - Viergever, Max A. AU - Kuijf, Hugo J. PY - 2022 TI - Influence of uncertainty estimation techniques on false-positive reduction in liver lesion detection T2 - Machine Learning for Biomedical Imaging VL - 1 IS - December 2022 issue SP - 1 EP - 33 SN - 2766-905X UR - https://melba-journal.org/2022:030 ER -

2022:030 cover