arXiv · 2307.08163
Boundary-weighted logit consistency improves calibration of segmentation networks
Abstract
Neural network prediction probabilities and accuracy are often only weakly-correlated. Inherent label ambiguity in training data for image segmentation aggravates such miscalibration. We show that logit consistency across stochastic transformations acts as a spatially varying regularizer that prevents overconfident predictions at pixels with ambiguous labels. Our boundary-weighted extension of this regularizer provides state-of-the-art calibration for prostate and heart MRI segmentation.
Explore related subjects
Keep this discovery
Neerav Karani, Neel Dey, Polina Golland. 2023-07-16. Boundary-weighted logit consistency improves calibration of segmentation networks. https://arxiv.org/abs/2307.08163
Cite the original work for its findings. Save a collection to share your selection of sources.