DR-LiteNet: A Lightweight Explainable Hybrid CNN Framework for Imbalanced Diabetic Retinopathy Grading via Adaptive SMOTE Fusion
Organizations: NIMISHES Lab, Dhaka, Bangladesh · Brac University, Dhaka, Bangladesh
Abstract
Automated diabetic retinopathy (DR) grading from colour fundus photographs is held back by two compounding challenges: severe class imbalance across five severity grades and the heavy computational cost of large scale architectures in resource constrained clinical settings. We introduce DR-LiteNet, a dual branch hybrid convolutional neural network that tackles both problems at once. The CNN branch extracts 1,280 dimensional deep features through EfficientNetB0, while the handcrafted (HC) branch produces a compact 51 dimensional lesion descriptor capturing vessel density, exudate intensity, texture entropy, and HSV colour histograms. Concatenating both branches yields a 1,331-d hybrid embedding, on which Adaptive SMOTE (ADASYN) is applied during training to synthesise minority class feature vectors, correcting distributional skew without touching the image space. A two phase training protocol first warms up the classification head on balanced features, then fine tunes the full network end to end. Grad CAM saliency maps deliver post-hoc clinical explainability. Tested on APTOS 2019 under five fold stratified cross validation, DR-LiteNet reaches 88.77% +/- 0.61 accuracy, a quadratic weighted kappa (QWK) of 0.884 +/- 0.010, and a macro F1 of 0.792 +/- 0.008 with only 4.69 M parameters beating ResNet50 (23.5 M) by DeltaQWK = 0.022 while staying practical on a single Google Colab T4 GPU. The source code of this project is available at: https://github.com/Nishi-Kanta-Paul/DR-LiteNet.
Figures & tables
| Model | Params (M) | Acc (%) mean std | QWK mean std | F1 mean std | ms img |
| EffNetB0 (baseline) | 4.01 | 84.2 0.9 | 0.832 0.015 | 0.741 0.014 | 31 |
| EffNetB0 + CW | 4.01 | 85.9 0.8 | 0.853 0.012 | 0.758 0.012 | 31 |
| MobileNetV3-S | 0.97 | 82.8 1.0 | 0.821 0.016 | 0.733 0.015 | 18 |
| ResNet50 | 23.5 | 86.4 0.7 | 0.862 0.011 | 0.771 0.010 | 57 |
| LiteNet (no ADASYN) | 4.69 | 85.1 0.9 | 0.846 0.013 | 0.748 0.013 | 38 |
| LiteNet + CW | 4.69 | 86.3 0.7 | 0.862 0.011 | 0.769 0.010 | 38 |
| Configuration | QWK | F1 | Sev. | Pro. |
| (i) Full model (proposed) | 0.884 | 0.792 | 0.712 | 0.847 |
| (ii) HC features | 0.871 | 0.778 | 0.693 | 0.831 |
| (iii) ADASYN | 0.846 | 0.748 | 0.641 | 0.798 |
| (iv) Both (CNN only, no ADASYN) | 0.832 | 0.741 | 0.612 | 0.779 |
| (v) CW loss instead of ADASYN | 0.862 | 0.769 | 0.674 | 0.821 |
| (vi) MobileNetV3-S backbone | 0.851 | 0.763 | 0.681 | 0.814 |
Diabetic Retinopathy Grading
Diabetic retinopathy grading uses retinal images to estimate disease severity and identify cases that need further clinical review.
See all 36 papers