PostTrainCalibration
Code for calibration of trained segmentation models using platt scaling, fine tuning and simple convolutional auxiliary networks
Description:
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, and compare them to Monte Carlo dorpout.
More Information:
Originates from the AI-Driven Data Science research challenge (FAIR 2019–2023).
Responsible Research Lead: Dirk Valkenborg
Authors: Dirk Valkenborg, Axel-Jan Rousseau
Programming Language: Python
Categories: Images, Supervised learning, Optimization
Publications:
Rousseau, Axel-Jan, Thijs Becker, Jeroen Bertels, Matthew B. Blaschko, Dirk Valkenborg, "Post Training Uncertainty Calibration of Deep Networks for Medical Image Segmentation", In Proceedings of the 18th IEEE International Symposium on Biomedical Imaging, pp. 1052 -1056, 2021.