1. Anomaly segmentation in retinal images with poisson-blending data augmentation.
- Author
-
Wang, Hualin, Zhou, Yuhong, Zhang, Jiong, Lei, Jianqin, Sun, Dongke, Xu, Feng, and Xu, Xiayu
- Subjects
- *
DATA augmentation , *RETINAL imaging , *IMAGE segmentation , *CONVOLUTIONAL neural networks , *DIABETIC retinopathy - Abstract
• We propose a novel Poisson-blending data augmentation to generate large-scale task-specific training data. • We propose a CNN architecture for the simultaneous segmentation of four types of DR lesions. • The method was extensively validated through ablation and comparison studies on two public datasets. • The results indicated that the proposed method outperformed the state-of-the-art methods. Diabetic retinopathy (DR) is one of the most important complications of diabetes. Accurate segmentation of DR lesions is of great importance for the early diagnosis of DR. However, simultaneous segmentation of multi-type DR lesions is technically challenging because of 1) the lack of pixel-level annotations and 2) the large diversity between different types of DR lesions. In this study, first, we propose a novel Poisson-blending data augmentation (PBDA) algorithm to generate synthetic images, which can be easily utilized to expand the existing training data for lesion segmentation. We perform extensive experiments to recognize the important attributes in the PBDA algorithm. We show that position constraints are of great importance and that the synthesis density of one type of lesion has a joint influence on the segmentation of other types of lesions. Second, we propose a convolutional neural network architecture, named DSR-U-Net++ (i.e., DC-SC residual U-Net++), for the simultaneous segmentation of multi-type DR lesions. Ablation studies showed that the mean area under precision recall curve (AUPR) for all four types of lesions increased by >5% with PBDA. The proposed DSR-U-Net++ with PBDA outperformed the state-of-the-art methods by 1.7%-9.9% on the Indian Diabetic Retinopathy Image Dataset (IDRiD) and 67.3% on the e-ophtha dataset with respect to mean AUPR. The developed method would be an efficient tool to generate large-scale task-specific training data for other medical anomaly segmentation tasks. [Display omitted] [ABSTRACT FROM AUTHOR]
- Published
- 2022
- Full Text
- View/download PDF