Please use this identifier to cite or link to this item:
http://hdl.handle.net/1942/34686
Title: | Post Training Uncertainty Calibration Of Deep Networks For Medical Image Segmentation | Authors: | ROUSSEAU, Axel-Jan BECKER, Thijs Bertels, Jeroen Blaschko, Matthew B. VALKENBORG, Dirk |
Issue Date: | 2021 | Publisher: | IEEE | Source: | 2021 IEEE 18th International Symposium on Biomedical Imaging (ISBI), p. 1052 -1056 | Series/Report: | IEEE International Symposium on Biomedical Imaging | Abstract: | Neural networks for automated image segmentation are typically trained to achieve maximum accuracy, while less attention has been given to the calibration of their confidence scores. However, well-calibrated confidence scores provide valuable information towards the user. We investigate several post hoc calibration methods that are straightforward to implement, some of which are novel. They are compared to Monte Carlo (MC) dropout and are applied to neural networks trained with cross-entropy (CE) and soft Dice (SD) losses on BraTS 2018 and ISLES 2018. Surprisingly, models trained on SD loss are not necessarily less calibrated than those trained on CE loss. In all cases, at least one post hoc method improves the calibration. There is limited consistency across the results, so we can't conclude on one method being superior. In all cases, post hoc calibration is competitive with MC dropout. Although average calibration improves compared to the base model, subject-level variance of the calibration remains similar. | Keywords: | uncertainty estimation;confidence calibration;deep learning;Medical image segmentation | Document URI: | http://hdl.handle.net/1942/34686 | ISBN: | 978-1-6654-1246-9 | DOI: | 10.1109/ISBI48211.2021.9434131 | ISI #: | 000786144100220 | Category: | C1 | Type: | Proceedings Paper | Validations: | ecoom 2023 |
Appears in Collections: | Research publications |
Files in This Item:
File | Description | Size | Format | |
---|---|---|---|---|
POST_TRAINING_UNCERTAINTY_CALIBRATION_OF_DEEP_NETWORKS_FOR_MEDICAL_IMAGE_SEGMENTATION.pdf Restricted Access | Published version | 1.65 MB | Adobe PDF | View/Open Request a copy |
4545_arXiv.pdf | Non Peer-reviewed author version | 1.13 MB | Adobe PDF | View/Open |
Items in DSpace are protected by copyright, with all rights reserved, unless otherwise indicated.