Deep Self-Cleansing for Medical Image Segmentation with Noisy Labels
This addresses the issue of inaccurate segmentation due to label noise in medical imaging, which is crucial for disease diagnosis and surgical planning, representing an incremental improvement over existing methods.
The paper tackles the problem of noisy labels in medical image segmentation by proposing a deep self-cleansing framework that filters and cleanses labels during training, achieving prominent segmentation performance on liver tumor and cardiac datasets.
Medical image segmentation is crucial in the field of medical imaging, aiding in disease diagnosis and surgical planning. Most established segmentation methods rely on supervised deep learning, in which clean and precise labels are essential for supervision and significantly impact the performance of models. However, manually delineated labels often contain noise, such as missing labels and inaccurate boundary delineation, which can hinder networks from correctly modeling target characteristics. In this paper, we propose a deep self-cleansing segmentation framework that can preserve clean labels while cleansing noisy ones in the training phase. To achieve this, we devise a gaussian mixture model-based label filtering module that distinguishes noisy labels from clean labels. Additionally, we develop a label cleansing module to generate pseudo low-noise labels for identified noisy samples. The preserved clean labels and pseudo-labels are then used jointly to supervise the network. Validated on a clinical liver tumor dataset and a public cardiac diagnosis dataset, our method can effectively suppress the interference from noisy labels and achieve prominent segmentation performance.