Hybrid attention-based deep learning for multi-label ophthalmic disease detection on fundus images.

BACKGROUND

Ophthalmic diseases significantly impact vision and quality of life. Early diagnosis using fundus images is critical for timely treatment. Traditional deep learning models often lack accuracy, interpretability, and efficiency for multi-label classification tasks in ophthalmology.

METHODS

We propose HAM-DNet, a hybrid deep learning model combining EfficientNetV2 and Vision Transformers (ViT) for multi-label ophthalmic disease detection. The model includes SE (Squeeze-and-Excitation) blocks for attention-based feature refinement and a U-Net-based lesion localization module for improved interpretability. The model was trained and tested on multiple fundus image datasets (ODIR-5 K, Messidor, G1020, and Joint Shantou International Eye Centre).

RESULTS

HAM-DNet achieved superior performance with an accuracy of 95.3%, precision of 96.2%, recall of 97.1%, AUC of 98.42, and F1-score of 96.75, while maintaining low computational cost (9.7 GFLOPS). It outperformed existing models including Shallow CNN and EfficientNet, particularly in handling multi-label classifications and reducing false positives and negatives.

CONCLUSIONS

HAM-DNet offers a robust, accurate, and interpretable solution for automated detection of multiple ophthalmic diseases. Its lightweight architecture makes it suitable for clinical deployment, especially in telemedicine and resource-constrained environments.