arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2005.14284cs.CVcs.LGeess.IV
收藏

使用深度学习在视网膜眼底图像中进行视盘定位和青光眼分类的两阶段框架

Two-stage framework for optic disc localization and glaucoma classification in retinal fundus images using deep learning

Technische Universität Kaiserslautern(凯泽斯劳滕工业大学)

Muhammad Naseer Bajwa, Muhammad Imran Malik, Shoaib Ahmed Siddiqui, Andreas Dengel, Faisal Shafait, Wolfgang Neumeier, Sheraz Ahmed

摘要:

随着强大的图像处理和机器学习技术的进步,CAD在包括眼科在内的所有医学领域变得越来越普遍。由于视盘是青光眼检测视网膜眼底图像中最重要的部分,本文提出了一个两阶段的框架,首先检测和定位视盘,然后将其分为健康或青光眼。第一阶段基于RCNN,负责从视网膜眼底图像中定位和提取视盘,而第二阶段使用Deep CNN将提取的视盘分类为健康或昏迷。除了所提出的解决方案之外,我们还开发了一种基于规则的半自动地面实况生成方法,该方法为训练基于RCNN的自动光盘定位模型提供了必要的注释。所提出的方法进行了评估,在七个公开可用的数据集的光盘定位和ORIGA数据集,这是最大的公开可用的青光眼分类数据集。自动定位的结果在六个数据集上标志着新的最新技术水平,其中四个数据集的准确率达到100%。对于青光眼分类,我们实现了等于0.874的AUC,这比先前获得的ORIGA分类的最新结果相对改善了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.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑