Iterative Confidence Relabeling with Deep ConvNets for Organ Segmentation with Partial Labels - Archive ouverte HAL
Article Dans Une Revue Computerized Medical Imaging and Graphics Année : 2021

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

Résumé

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
Origine Fichiers produits par l'(les) auteur(s)

Dates et versions

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

Identifiants

Citer

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 Consultations
246 Téléchargements

Altmetric

Partager

More