Self-relabeling for noise-tolerant retina vessel segmentation through label reliability estimation

• Published in: BMC medical imaging Vol. 22; no. 1; p. 8
• Main Authors: Li, Jiacheng, Li, Ruirui, Han, Ruize, Wang, Song
• Format: Journal Article
• Language: English
• Published: England BioMed Central 12.01.2022; BMC
• Subjects: Accuracy; Annotations; Background noise; Blood vessels; Classification; Datasets; Deep Learning; Errors; Humans; Image processing; Image Processing, Computer-Assisted - methods; Image segmentation; Label map correction; Labeling; Labels; Medical research; Methods; Noise; Noise-tolerant; Pixels; Reliability estimation; Retina; Retina image segmentation; Retinal images; Retinal Vessels - diagnostic imaging; Temporal statistics; Training; Reliability estimation; Noise-tolerant; Label map correction; Temporal statistics; Retina image segmentation
• ISSN: 1471-2342; 1471-2342
• DOI: 10.1186/s12880-021-00732-y

Retinal vessel segmentation benefits significantly from deep learning. Its performance relies on sufficient training images with accurate ground-truth segmentation, which are usually manually annotated in the form of binary pixel-wise label maps. Manually annotated ground-truth label maps, more or less, contain errors for part of the pixels. Due to the thin structure of retina vessels, such errors are more frequent and serious in manual annotations, which negatively affect deep learning performance. In this paper, we develop a new method to automatically and iteratively identify and correct such noisy segmentation labels in the process of network training. We consider historical predicted label maps of network-in-training from different epochs and jointly use them to self-supervise the predicted labels during training and dynamically correct the supervised labels with noises. We conducted experiments on the three datasets of DRIVE, STARE and CHASE-DB1 with synthetic noises, pseudo-labeled noises, and manually labeled noises. For synthetic noise, the proposed method corrects the original noisy label maps to a more accurate label map by 4.0-[Formula: see text] on [Formula: see text] and 10.7-[Formula: see text] on PR on three testing datasets. For the other two types of noise, the method could also improve the label map quality. Experiment results verified that the proposed method could achieve better retinal image segmentation performance than many existing methods by simultaneously correcting the noise in the initial label map.