Iterative Confidence Relabeling with Deep ConvNets for Organ Segmentation with Partial Labels - Archive ouverte HAL
Journal Articles Computerized Medical Imaging and Graphics Year : 2021

Iterative Confidence Relabeling with Deep ConvNets for Organ Segmentation with Partial Labels

Abstract

Training deep ConvNets requires large labeled datasets. However, collecting pixel-level labels for medical image segmentation is very expensive and requires a high level of expertise. In addition, most existing segmentation masks provided by clinical experts focus on specific anatomical structures. In this paper, we propose a method dedicated to handle such partially labeled medical image datasets. We propose a strategy to identify pixels for which labels are correct, and to train Fully Convolutional Neural Networks with a multi-label loss adapted to this context. In addition, we introduce an iterative confidence self-training approach inspired by curriculum learning to relabel missing pixel labels, which relies on selecting the most confident prediction with a specifically designed confidence network that learns an uncertainty measure which is leveraged in our relabeling process. Our approach, INERRANT for Iterative coNfidencE Relabeling of paRtial ANnoTations, is thoroughly evaluated on two public datasets (TCAI and LITS), and one internal dataset with seven abdominal organ classes. We show that INERRANT robustly deals with partial labels, performing similarly to a model trained on all labels even for large missing label proportions. We also highlight the importance of our iterative learning scheme and the proposed confidence measure for optimal performance. Finally we show a practical use case where a limited number of completely labeled data are enriched by publicly available but partially labeled data.
Fichier principal
Vignette du fichier
CMIG_INERRANT.pdf (7.38 Mo) Télécharger le fichier
Origin Files produced by the author(s)

Dates and versions

hal-03243619 , version 1 (31-05-2021)

Identifiers

Cite

Olivier Petit, Nicolas Thome, Luc Soler. Iterative Confidence Relabeling with Deep ConvNets for Organ Segmentation with Partial Labels. Computerized Medical Imaging and Graphics, 2021, pp.101938. ⟨10.1016/j.compmedimag.2021.101938⟩. ⟨hal-03243619⟩
103 View
246 Download

Altmetric

Share

More