Spatially varying label smoothing: capturing uncertainty from expert
annotations
annotations
File(s)2104.05788v1.pdf (4.58 MB)
Accepted version
Author(s)
Islam, Mobarakol
Glocker, Ben
Type
Conference Paper
Abstract
The task of image segmentation is inherently noisy due to ambiguities
regarding the exact location of boundaries between anatomical structures. We
argue that this information can be extracted from the expert annotations at no
extra cost, and when integrated into state-of-the-art neural networks, it can
lead to improved calibration between soft probabilistic predictions and the
underlying uncertainty. We built upon label smoothing (LS) where a network is
trained on 'blurred' versions of the ground truth labels which has been shown
to be effective for calibrating output predictions. However, LS is not taking
the local structure into account and results in overly smoothed predictions
with low confidence even for non-ambiguous regions. Here, we propose Spatially
Varying Label Smoothing (SVLS), a soft labeling technique that captures the
structural uncertainty in semantic segmentation. SVLS also naturally lends
itself to incorporate inter-rater uncertainty when multiple labelmaps are
available. The proposed approach is extensively validated on four clinical
segmentation tasks with different imaging modalities, number of classes and
single and multi-rater expert annotations. The results demonstrate that SVLS,
despite its simplicity, obtains superior boundary prediction with improved
uncertainty and model calibration.
regarding the exact location of boundaries between anatomical structures. We
argue that this information can be extracted from the expert annotations at no
extra cost, and when integrated into state-of-the-art neural networks, it can
lead to improved calibration between soft probabilistic predictions and the
underlying uncertainty. We built upon label smoothing (LS) where a network is
trained on 'blurred' versions of the ground truth labels which has been shown
to be effective for calibrating output predictions. However, LS is not taking
the local structure into account and results in overly smoothed predictions
with low confidence even for non-ambiguous regions. Here, we propose Spatially
Varying Label Smoothing (SVLS), a soft labeling technique that captures the
structural uncertainty in semantic segmentation. SVLS also naturally lends
itself to incorporate inter-rater uncertainty when multiple labelmaps are
available. The proposed approach is extensively validated on four clinical
segmentation tasks with different imaging modalities, number of classes and
single and multi-rater expert annotations. The results demonstrate that SVLS,
despite its simplicity, obtains superior boundary prediction with improved
uncertainty and model calibration.
Date Issued
2021-06-14
Date Acceptance
2021-02-12
Citation
Lecture Notes in Computer Science, 2021, 12729, pp.677-688
ISBN
978-3-030-78190-3
ISSN
0302-9743
Publisher
Springer Verlag
Start Page
677
End Page
688
Journal / Book Title
Lecture Notes in Computer Science
Volume
12729
Copyright Statement
© 2021 Springer Nature Switzerland AG. The final publication is available at Springer via https://link.springer.com/chapter/10.1007/978-3-030-78191-0_52
Sponsor
Commission of the European Communities
Identifier
http://arxiv.org/abs/2104.05788v1
Grant Number
H2020 - 757173
Source
Information Processing in Medical Imaging (IPMI) 2021
Subjects
cs.CV
cs.CV
Notes
Accepted at IPMI 2021
Publication Status
Published
Start Date
2021-06-28
Finish Date
2021-06-30
Coverage Spatial
Virtual
Date Publish Online
2021-06-14