论文标题
通过转移学习对糖尿病性视网膜病的预测分析
Predictive Analysis of Diabetic Retinopathy with Transfer Learning
论文作者
论文摘要
随着糖尿病的患病率,糖尿病性视网膜病变(DR)正在成为世界上主要的健康问题。由于DR引起的长期医疗并发症对患者和社会产生了重大影响,因为该疾病在其最有生产力的岁月中主要影响个人。早期检测和治疗可以帮助减少对患者的损害程度。卷积神经网络在医学领域进行预测分析的兴起为可DR检测的强大解决方案铺平了道路。本文借助转移学习研究了几种高效且可扩展的CNN体系结构进行糖尿病性视网膜病变分类的性能。该研究的重点是VGG16,RESNET50 V2和EFIDEDNET B0模型。使用几个性能指标(包括真实正率,假阳性率,准确性等)对分类性能进行分析。此外,还绘制了几个性能图,以可视化建筑性能,包括混淆矩阵,ROC曲线等。结果表明,使用VGG 16使用Imagenet重量进行传输,使用VGG 16模型表明,具有最佳准确性的95%的最佳分类性能。紧随其后的是Resnet50 V2体系结构,最佳准确度为93%。本文表明,通过在卷积神经网络上的转移学习来实现来自视网膜图像的DR的预测分析。
With the prevalence of Diabetes, the Diabetes Mellitus Retinopathy (DR) is becoming a major health problem across the world. The long-term medical complications arising due to DR have a significant impact on the patient as well as the society, as the disease mostly affects individuals in their most productive years. Early detection and treatment can help reduce the extent of damage to the patients. The rise of Convolutional Neural Networks for predictive analysis in the medical field paves the way for a robust solution to DR detection. This paper studies the performance of several highly efficient and scalable CNN architectures for Diabetic Retinopathy Classification with the help of Transfer Learning. The research focuses on VGG16, Resnet50 V2 and EfficientNet B0 models. The classification performance is analyzed using several performance metrics including True Positive Rate, False Positive Rate, Accuracy, etc. Also, several performance graphs are plotted for visualizing the architecture performance including Confusion Matrix, ROC Curve, etc. The results indicate that Transfer Learning with ImageNet weights using VGG 16 model demonstrates the best classification performance with the best Accuracy of 95%. It is closely followed by ResNet50 V2 architecture with the best Accuracy of 93%. This paper shows that predictive analysis of DR from retinal images is achieved with Transfer Learning on Convolutional Neural Networks.