arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2308.05178eess.IVcs.CV

一种利用迁移学习和集成学习的改进型糖尿病视网膜病变检测模型

An Improved Model for Diabetic Retinopathy Detection by using Transfer Learning and Ensemble Learning

Rajshahi University of Engineering and Technology(拉杰沙希工程技术大学);Jashore University of Science and Technology(杰索尔理工大学)

Md. Simul Hasan Talukder, Ajay Kirshno Sarkar, Sharmin Akter, Md. Nuhi-Alamin

AI总结:

本文利用多种预训练模型的迁移学习及集成学习构建糖尿病视网膜病变检测模型,DenseNet121及DenseNet169与DenseNet201的集成均达到100%准确率。

摘要:

糖尿病视网膜病变(DR)是一种由血液中持续高糖水平引起的眼部疾病,会导致视网膜毛细血管阻塞和出血,造成视网膜组织损伤,通常最终导致失明。早期检测有助于降低DR的风险及其严重程度。对糖尿病视网膜病变进行稳健且准确的预测和检测是一项具有挑战性的任务。本文开发了一种完全准确的用于检测糖尿病视网膜病变的机器学习模型。在迁移学习(TL)方法中,使用了预训练模型,如ResNet50、InceptionV3、Xception、DenseNet121、VGG19、NASNetMobile、MobileNetV2、DenseNet169和DenseNet201,并在其底部添加了池化层、全连接层和适当的dropout层。进行了数据增强和正则化以减少过拟合。DenseNet121的迁移学习模型、DenseNet169和DenseNet201迁移学习架构的平均集成和加权集成分别单独贡献了最高的100%准确率,以及最高的精确率、召回率和F-1分数,分别为100%、100%和100%。

英文摘要:

Diabetic Retinopathy (DR) is an ocular condition caused by a sustained high level of sugar in the blood, which causes the retinal capillaries to block and bleed, causing retinal tissue damage. It usually results in blindness. Early detection can help in lowering the risk of DR and its severity. The robust and accurate prediction and detection of diabetic retinopathy is a challenging task. This paper develops a machine learning model for detecting Diabetic Retinopathy that is entirely accurate. Pre-trained models such as ResNet50, InceptionV3, Xception, DenseNet121, VGG19, NASNetMobile, MobileNetV2, DensNet169, and DenseNet201 with pooling layer, dense layer, and appropriate dropout layer at the bottom of them were carried out in transfer learning (TL) approach. Data augmentation and regularization was performed to reduce overfitting. Transfer Learning model of DenseNet121, Average and weighted ensemble of DenseNet169 and DenseNet201 TL architectures contribute individually the highest accuracy of 100%, the highest precision, recall, F-1 score of 100%, 100%, and 100%, respectively.

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

↑