Pith. sign in

Boundary-weighted logit consistency improves calibration of segmentation networks

1 Pith paper cite this work. Polarity classification is still indexing.

1 Pith paper citing it
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.

fields

cs.CL 1

years

2025 1

verdicts

REJECT 1

representative citing papers

citing papers explorer

Showing 1 of 1 citing paper.