论文标题
使用深度学习的视网膜眼镜图像中的视盘定位和青光眼分类的两阶段框架
Two-stage framework for optic disc localization and glaucoma classification in retinal fundus images using deep learning
论文作者
论文摘要
随着强大图像处理和机器学习技术的发展,CAD在包括眼科在内的所有医学领域都变得越来越普遍。由于视盘是视网膜底面检测的最重要部分,因此本文提出了一个两阶段的框架,该框架首先检测和定位视盘,然后将其分为健康或青光眼。第一阶段是基于RCNN的,负责从视网膜底面图像中定位和提取视盘,而第二阶段则使用深CNN将提取的盘分类为健康或青光眼。除了提出的解决方案外,我们还开发了一种基于规则的半自动地面真相生成方法,该方法为训练基于RCNN的模型提供了必要的注释,用于自动盘定位。在七个可公开可用的数据集和Origa数据集上评估了所提出的方法,该数据集是最大的公开数据集用于青光眼分类。自动本地化的结果标志着六个数据集上的最新最先进的结果,其中四个数据集的精度达到了100%。对于青光眼分类,我们实现了等于0.874的AUC,比先前获得的原始分类的最新结果相对相对提高了2.7%。一旦经过仔细注释的数据培训,基于深度学习的视盘检测和定位方法不仅是稳健,准确且完全自动化的,而且还消除了对数据集依赖性启发式算法的需求。我们对Origa的青光眼分类的经验评估表明,仅报告AUC,对于具有类不平衡的数据集且没有预定的火车和测试拆分,就不会描绘出分类器的性能的真实情况,并要求提供额外的性能指标来证明结果。
With the advancement of powerful image processing and machine learning techniques, CAD has become ever more prevalent in all fields of medicine including ophthalmology. Since optic disc is the most important part of retinal fundus image for glaucoma detection, this paper proposes a two-stage framework that first detects and localizes optic disc and then classifies it into healthy or glaucomatous. The first stage is based on RCNN and is responsible for localizing and extracting optic disc from a retinal fundus image while the second stage uses Deep CNN to classify the extracted disc into healthy or glaucomatous. In addition to the proposed solution, we also developed a rule-based semi-automatic ground truth generation method that provides necessary annotations for training RCNN based model for automated disc localization. The proposed method is evaluated on seven publicly available datasets for disc localization and on ORIGA dataset, which is the largest publicly available dataset for glaucoma classification. The results of automatic localization mark new state-of-the-art on six datasets with accuracy reaching 100% on four of them. For glaucoma classification we achieved AUC equal to 0.874 which is 2.7% relative improvement over the state-of-the-art results previously obtained for classification on ORIGA. Once trained on carefully annotated data, Deep Learning based methods for optic disc detection and localization are not only robust, accurate and fully automated but also eliminates the need for dataset-dependent heuristic algorithms. Our empirical evaluation of glaucoma classification on ORIGA reveals that reporting only AUC, for datasets with class imbalance and without pre-defined train and test splits, does not portray true picture of the classifier's performance and calls for additional performance metrics to substantiate the results.