针对视网膜血管分割中数据不平衡问题,提出双层平衡的深度学习方法。
A Deep Convolutional Neural Network-Based Novel Class Balancing for Imbalance Data Segmentation
- 采用双层类别平衡策略,分别处理血管/非血管与粗/细血管间的分布不均
- 在标准数据集上达到98.23%的ROC曲线下面积,灵敏度达81.57%
- 适用于医学图像分割,尤其适合小目标、不均衡数据场景
视网膜眼底图像能揭示人眼内部结构及重要特征(如血管、视盘、黄斑、中心凹)。然而,由于数据分布不平衡和血管粗细差异,精确分割视网膜血管仍具挑战。本文提出BLCB-CNN,一种基于深度学习的新型双层类别平衡分割框架。该方法通过两级平衡机制:一级平衡血管与非血管像素分布,二级平衡粗血管与细血管分布。输入图像经全局对比度归一化(GCN)、限制性自适应直方图均衡化(CLAHE)和伽马校正预处理,提升对比度与亮度均匀性。最终在标准数据集上实现98.23%的ROC曲线下面积、96.22%准确率、81.57%敏感度和97.65%特异性。外部交叉验证在STARE数据集上进一步证实了模型的泛化能力。
原文摘要 · Abstract (English)
Retinal fundus images provide valuable insights into the human eye's interior structure and crucial features, such as blood vessels, optic disk, macula, and fovea. However, accurate segmentation of retinal blood vessels can be challenging due to imbalanced data distribution and varying vessel thickness. In this paper, we propose BLCB-CNN, a novel pipeline based on deep learning and bi-level class balancing scheme to achieve vessel segmentation in retinal fundus images. The BLCB-CNN scheme uses a Convolutional Neural Network (CNN) architecture and an empirical approach to balance the distribution of pixels across vessel and non-vessel classes and within thin and thick vessels. Level-I is used for vessel/non-vessel balancing and Level-II is used for thick/thin vessel balancing. Additionally, pre-processing of the input retinal fundus image is performed by Global Contrast Normalization (GCN), Contrast Limited Adaptive Histogram Equalization (CLAHE), and gamma corrections to increase intensity uniformity as well as to enhance the contrast between vessels and background pixels. The resulting balanced dataset is used for classification-based segmentation of the retinal vascular tree. We evaluate the proposed scheme on standard retinal fundus images and achieve superior performance measures, including an area under the ROC curve of 98.23%, Accuracy of 96.22%, Sensitivity of 81.57%, and Specificity of 97.65%. We also demonstrate the method's efficacy through external cross-validation on STARE images, confirming its generalization ability.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。