融合多源眼底数据与可解释AI,提升糖尿病视网膜病变自动诊断准确率
VR-FuseNet: A Fusion of Heterogeneous Fundus Data and Explainable Deep Network for Diabetic Retinopathy Classification
- 设计新型混合模型VR-FuseNet,融合VGG19与ResNet50V2提取多层级特征
- 在五大数据集上训练,达到91.824%准确率,优于单一模型
- 结合XAI技术生成可视化解释,帮助医生理解诊断依据
糖尿病视网膜病变是糖尿病患者常见的严重眼病,若不及时干预可能导致失明。本文提出一种全自动检测方法,构建基于五个公开数据集的混合数据集,并通过SMOTE进行类别平衡、CLAHE增强图像质量,提升数据鲁棒性。提出的VR-FuseNet模型融合VGG19的细粒度空间特征提取能力与ResNet50V2的深层层次特征表达能力,在分类任务中取得91.824%的准确率,显著优于单个网络结构。为提升临床可用性,模型整合多种可解释AI(XAI)技术,生成微动脉瘤、出血点和渗出物等关键病变区域的可视化解释,使医生能验证模型决策依据。
原文摘要 · Abstract (English)
Diabetic retinopathy is a severe eye condition caused by diabetes where the retinal blood vessels get damaged and can lead to vision loss and blindness if not treated. Early and accurate detection is key to intervention and stopping the disease progressing. For addressing this disease properly, this paper presents a comprehensive approach for automated diabetic retinopathy detection by proposing a new hybrid deep learning model called VR-FuseNet. Diabetic retinopathy is a major eye disease and leading cause of blindness especially among diabetic patients so accurate and efficient automated detection methods are required. To address the limitations of existing methods including dataset imbalance, diversity and generalization issues this paper presents a hybrid dataset created from five publicly available diabetic retinopathy datasets. Essential preprocessing techniques such as SMOTE for class balancing and CLAHE for image enhancement are applied systematically to the dataset to improve the robustness and generalizability of the dataset. The proposed VR-FuseNet model combines the strengths of two state-of-the-art convolutional neural networks, VGG19 which captures fine-grained spatial features and ResNet50V2 which is known for its deep hierarchical feature extraction. This fusion improves the diagnostic performance and achieves an accuracy of 91.824%. The model outperforms individual architectures on all performance metrics demonstrating the effectiveness of hybrid feature extraction in Diabetic Retinopathy classification tasks. To make the proposed model more clinically useful and interpretable this paper incorporates multiple XAI techniques. These techniques generate visual explanations that clearly indicate the retinal features affecting the model's prediction such as microaneurysms, hemorrhages and exudates so that clinicians can interpret and validate.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。