Pick-and-Learn: Automatic Quality Evaluation for Noisy-Labeled Image Segmentation
This addresses a crucial issue for medical imaging where expert annotations are costly and noisy, offering an incremental improvement over existing methods.
The paper tackles the problem of noisy labels in medical image segmentation by introducing an automatic label quality evaluation strategy that trains on clean labels without additional annotations, achieving high accuracy and good generalization across different noise levels.
Deep learning methods have achieved promising performance in many areas, but they are still struggling with noisy-labeled images during the training process. Considering that the annotation quality indispensably relies on great expertise, the problem is even more crucial in the medical image domain. How to eliminate the disturbance from noisy labels for segmentation tasks without further annotations is still a significant challenge. In this paper, we introduce our label quality evaluation strategy for deep neural networks automatically assessing the quality of each label, which is not explicitly provided, and training on clean-annotated ones. We propose a solution for network automatically evaluating the relative quality of the labels in the training set and using good ones to tune the network parameters. We also design an overfitting control module to let the network maximally learn from the precise annotations during the training process. Experiments on the public biomedical image segmentation dataset have proved the method outperforms baseline methods and retains both high accuracy and good generalization at different noise levels.