基于平行图像的糖尿病视网膜病变智能诊断
DOI:
作者:
作者单位:

1.河南工业大学电气工程学院 郑州 450001; 2.河南工业大学大数据与人工智能学院 郑州 450001

作者简介:

通讯作者:

中图分类号:

TP391

基金项目:

国家自然科学基金(61473114,62106068)


Intelligent diagnosis of diabetic retinopathy based on parallel images
Author:
Affiliation:

1.College of Electrical Engineering, Henan University of Technology, Zhengzhou 450001, China; 2.School of Artificial Intelligence and Big Data, Henan University of Technology, Zhengzhou 450001, China

Fund Project:

  • 摘要
  • |
  • 图/表
  • |
  • 访问统计
  • |
  • 参考文献
  • |
  • 相似文献
  • |
  • 引证文献
  • |
  • 资源附件
  • |
  • 文章评论
    摘要:

    针对深度学习诊断糖尿病视网膜病变(diabetic retinopathy, DR)面临数据集小、类别不均衡及诊断效果不佳等问题,本文提出基于平行图像和Swin Transformer的DR分级模型。首先基于StyleGAN2-ada构建平行图像生成模型,解决训练图像过少和类别失衡问题。经FID、KID和目视评估,构建的平行图像符合后续工作要求。然后,基于注意力与窗口滑动机制构建DR诊断模型改善诊断效果。最后,使用平行图像训练诊断模型。经验证,本文提出的诊断模型准确率为93.5%、特异性最高为99%、F1分数最高为0.96。与原始图像相比,使用平行图像训练模型后其准确率提升20%、精确率最高提升70%。与其他三种深度学习模型对比,本文所提方法各项指标均达到最优。以上结果表明,本文构建的模型可在小样本数据集下实现较好的诊断效果。

    Abstract:

    Aiming at the problems of small datasets, unbalanced categories and poor diagnostic results in deep learning diagnosis of diabetic retinopathy (DR), this paper proposes a DR grading model based on parallel images and Swin Transformer. First, build a parallel image generation model based on StyleGAN2-ada to solve the problem of too few training images and class imbalance. After FID, KID and visual evaluation, the constructed parallel images meet the requirements of subsequent work. Then, a DR diagnosis model is constructed based on the attention and window shifting mechanism to improve the diagnosis effect. Finally, a diagnostic model is trained using the parallel images. After verification, the accuracy of the diagnostic model proposed in this paper is 93.5%, the highest specificity is 99%, and the highest F1-score is 0.96. Compared with the original images, the accuracy of the model is improved by 20% and the accuracy is improved by up to 70% after training the model with parallel images. Compared with the other three deep learning models, all the indicators of the method proposed in this paper are optimal. The above results show that the model constructed in this paper can achieve better diagnostic results under a small sample data set.

    参考文献
    相似文献
    引证文献
引用本文

赵亮,付园坤,陈涵欣,魏政杰,云晴,金军委.基于平行图像的糖尿病视网膜病变智能诊断[J].电子测量技术,2022,45(14):131-139

复制
分享
文章指标
  • 点击次数:
  • 下载次数:
  • HTML阅读次数:
  • 引用次数:
历史
  • 收稿日期:
  • 最后修改日期:
  • 录用日期:
  • 在线发布日期: 2024-04-08
  • 出版日期: