Diffusion-augmented contrastive learning for diabetic retinopathy severity classification
- 1Department of Electrical Engineering and Computer Science, Florida Atlantic University, Boca Raton, FL, United States
Abstract
Automated diabetic retinopathy (DR) severity classification faces persistent challenges from limited annotated datasets, severe class imbalance across the five-level International Clinical Diabetic Retinopathy (ICDR) scale, and domain shift between imaging environments. This paper presents an integrated deep learning framework that extends our earlier Fused Ensemble Convolutional Inception Network (93.2% accuracy) through three complementary innovations. First, a class-conditional latent denoising diffusion probabilistic model (DDPM) generates anatomically realistic synthetic fundus images for underrepresented DR severity levels, addressing data scarcity and class imbalance simultaneously. Second, a SimCLR contrastive self-supervised pre-training protocol, applied to the combined real-and-synthetic corpus, learns discriminative retinal representations prior to supervised fine-tuning. Third, a five-model ensemble (EfficientNet-B3, ResNet-50, VGG16, VGG19, and DenseNet-121) fused via a learned Pattern-Net achieves improved generalization through architectural diversity. Evaluated on the APTOS 2019 dataset under 5-fold stratified cross-validation, the integrated framework achieves 95.1% accuracy, 0.928 quadratic weighted kappa (QWK), 0.91 F1-score, and 0.97 AUC. Class-wise improvements over the EfficientNet-B3 baseline reach +23.8 percentage points for Severe DR (61.8 → 85.6%) and +17.7 for Mild DR (71.2 → 88.9%). Independent zero-shot validation on EyePACS yields 78.4% accuracy, 0.809 QWK, 0.738 macro F1, and 0.887 macro AUC, quantifying the expected performance reduction under domain shift. Fold-level variability and paired out-of-fold statistical analyses are additionally reported, and Grad-CAM visualization is included for qualitative model interpretability. These results demonstrate the synergistic benefit of generative augmentation with self-supervised representation learning for imbalanced medical image classification, with implications for scalable automated DR screening in resource-limited settings.
Related
- Contrastive learning and advanced bundle metrics outperform fractional anisotropy for Alzheimer’s disease diffusion MRI prediction: findings from the Beyond FA challenge
- Deep Learning for Multi‐Disease OCT Classification in Clinical Retina Practice Using Attention Mechanisms and Contrastive Learning
- A multi-class fundus disease classification framework based on self-supervised contrastive learning and dual-branch network
- CBAM-enhanced convolutional networks for diabetic retinopathy classification in the ODIR dataset
- Improving cognitive impairment classification in Alzheimer’s disease using tabular diffusion augmentation and late fusion deep learning
BibTeX
@article{abidalkareem2026diffusionaugmented,
title = {{Diffusion-augmented contrastive learning for diabetic retinopathy severity classification}},
author = {Ali Abidalkareem and Ali K. Ibrahim and Hanqi Zhuang},
journal = {Frontiers in Artificial Intelligence},
year = {2026},
month = oct,
doi = {10.3389/frai.2026.1932259},
url = {https://doi.org/10.3389/frai.2026.1932259},
note = {CC BY 4.0},
}
RIS
TY - JOUR
TI - Diffusion-augmented contrastive learning for diabetic retinopathy severity classification
AU - Ali Abidalkareem
AU - Ali K. Ibrahim
AU - Hanqi Zhuang
PY - 2026
DA - 2026/10/12
JO - Frontiers in Artificial Intelligence
DO - 10.3389/frai.2026.1932259
UR - https://doi.org/10.3389/frai.2026.1932259
AB - Automated diabetic retinopathy (DR) severity classification faces persistent challenges from limited annotated datasets, severe class imbalance across the five-level International Clinical Diabetic Retinopathy (ICDR) scale, and domain shift between imaging environments. This paper presents an integrated deep learning framework that extends our earlier Fused Ensemble Convolutional Inception Network (93.2% accuracy) through three complementary innovations. First, a class-conditional latent denoising diffusion probabilistic model (DDPM) generates anatomically realistic synthetic fundus images for underrepresented DR severity levels, addressing data scarcity and class imbalance simultaneously. Second, a SimCLR contrastive self-supervised pre-training protocol, applied to the combined real-and-synthetic corpus, learns discriminative retinal representations prior to supervised fine-tuning. Third, a five-model ensemble (EfficientNet-B3, ResNet-50, VGG16, VGG19, and DenseNet-121) fused via a learned Pattern-Net achieves improved generalization through architectural diversity. Evaluated on the APTOS 2019 dataset under 5-fold stratified cross-validation, the integrated framework achieves 95.1% accuracy, 0.928 quadratic weighted kappa (QWK), 0.91 F1-score, and 0.97 AUC. Class-wise improvements over the EfficientNet-B3 baseline reach +23.8 percentage points for Severe DR (61.8 → 85.6%) and +17.7 for Mild DR (71.2 → 88.9%). Independent zero-shot validation on EyePACS yields 78.4% accuracy, 0.809 QWK, 0.738 macro F1, and 0.887 macro AUC, quantifying the expected performance reduction under domain shift. Fold-level variability and paired out-of-fold statistical analyses are additionally reported, and Grad-CAM visualization is included for qualitative model interpretability. These results demonstrate the synergistic benefit of generative augmentation with self-supervised representation learning for imbalanced medical image classification, with implications for scalable automated DR screening in resource-limited settings.
SN - 2624-8212
ER -