mlsanityData to 7 Oct 2026

Explainable Convolutional Neural Networks for Retinal Fundus Classification and Cutting-Edge Segmentation Models for Retinal Blood Vessels from Fundus Images

arXiv 2405.07338 ↗First posted 12 May 2024eess.IV cs.CVcited by 6

Not on the map. Why?

Abstract (arXiv)

Early detection of vision-threatening conditions such as diabetic retinopathy, glaucoma, and age-related macular degeneration depends on retinal fundus image analysis, but manual assessment is slow and expert-dependent. Automated convolutional neural networks classify fundus images accurately yet act as black boxes, and existing retinal vessel segmentation methods lose discriminative power under pathology and seldom exploit attention or transformer backbones. Using the FIVES and DRIVE fundus datasets, we develop a two-pipeline framework that pairs four-class disease classification with attention- and transformer-based vessel segmentation, organised in three stages: (1) FIVES images are augmented by rotation and horizontal and vertical flips and used to fine-tune eight ImageNet-pretrained CNNs: ResNet101, DenseNet169, Xception, InceptionV3, DenseNet121, InceptionResNetV2, ResNet50, and EfficientNetB0. (2) Five gradient-based explanation methods, Grad-CAM, Grad-CAM++, Score-CAM, Faster Score-CAM, and Layer-CAM, are computed on the final convolutional block of each classifier and compared qualitatively across architectures. (3) Ten U-Net variants are benchmarked for vessel segmentation: TransUNet (hybrid CNN--Transformer encoder) and Attention U-Net (gated skip connections), evaluated with ResNet50V2, ResNet101V2, and ResNet152V2 backbones, along with additional Attention U-Net configurations using DenseNet backbones, and the fully transformer-based Swin-UNet. ResNet101 gives the highest classification accuracy: 94.17% (F1 0.942) > 88.33% for EfficientNetB0. For segmentation, the architecture ranking is consistent on both datasets: Attention U-Net > TransUNet > Swin-UNet. The strongest configuration is Attention U-Net with a ResNet101V2 backbone: FIVES IoU 0.722, Dice 0.838; DRIVE IoU 0.648, Dice 0.787, lifting DRIVE IoU 60.80 rightarrow 64.83 over a prior custom U-Net.

Key facts. Cited by 6 papers in the map (data through 7 Oct 2026). Builds on 15 of its 58 references that are in the map. First posted on arXiv 12 May 2024.

Builds on

15 of 58 references.

References · 15

  1. 2015Deep Residual Learning for Image Recognitioncited by 51,809
  2. 2015U-Net: Convolutional Networks for Biomedical Image Segmentationcited by 19,604
  3. 2016Grad-CAM: Visual Explanations from Deep Networks via Gradient-based Localizationcited by 6,690
  4. 2015Rethinking the Inception Architecture for Computer Visioncited by 5,451
  5. 2016Xception: Deep Learning with Depthwise Separable Convolutionscited by 2,320
  6. 2021TransUNet: Transformers Make Strong Encoders for Medical Image Segmentationcited by 1,552
  7. 2016Inception-v4, Inception-ResNet and the Impact of Residual Connections on Learningcited by 1,509
  8. 2018Attention U-Net: Learning Where to Look for the Pancreascited by 1,420
  9. 2019Generalized Intersection over Union: A Metric and A Loss for Bounding Box Regressioncited by 1,361
  10. 2021Swin-Unet: Unet-like Pure Transformer for Medical Image Segmentationcited by 1,218
  11. 2017Grad-CAM++: Improved Visual Explanations for Deep Convolutional Networkscited by 936
  12. 2019Score-CAM: Score-Weighted Visual Explanations for Convolutional Neural Networkscited by 575
3 more
  1. 2018Deep learning to achieve clinically applicable segmentation of head and neck anatomy for radiotherapycited by 76
  2. 2020SA-UNet: Spatial Attention U-Net for Retinal Vessel Segmentationcited by 70
  3. 2019Continuous Dice Coefficient: a Method for Evaluating Probabilistic Segmentationscited by 25

Built on by

Cited by year (2026 to 7 Oct)

Citing papers per year: 2025: 5; 2026: 1.

6 papers.