May, 2024

基于可解释卷积神经网络的视网膜底部分类及基于先进视网膜血管图像分割模型

TL;DR我们的研究聚焦于通过检查眼底图像中的视网膜血管进行疾病的早期诊断。我们利用八个预训练的 CNN 模型进行基于深度学习的分类,同时应用了可解释的 AI 技术,如 Grad-CAM、Grad-CAM++、Score-CAM、Faster Score-CAM 和 Layer CAM。通过对多个模型进行评估,我们发现 ResNet101 具有最高的准确率(94.17%),而 EfficientNetB0 的准确率最低(88.33%)。对于图像分割,Swin-Unet 表现出 86.19% 的平均像素准确率,而 Attention U-Net 基于 DenseNet201 的准确率最低(75.87%)。