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:
- Grand Challenge/Workpackage(s): GC1 - WP5
- Responsible Research Lead: Dirk Valkenborg
- Authors: Dirk Valkenborg, Axel-Jan Rousseau
- Programming Language: Python
- Link to the repository
- 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.
