Adaptive Multiscale Retinal Diagnosis: A Hybrid Trio-Model Approach for Comprehensive Fundus Multi-Disease Detection Leveraging Transfer Learning and Siamese Networks (2405.18449v1)
Abstract: WHO has declared that more than 2.2 billion people worldwide are suffering from visual disorders, such as media haze, glaucoma, and drusen. At least 1 billion of these cases could have been either prevented or successfully treated, yet they remain unaddressed due to poverty, a lack of specialists, inaccurate ocular fundus diagnoses by ophthalmologists, or the presence of a rare disease. To address this, the research has developed the Hybrid Trio-Network Model Algorithm for accurately diagnosing 12 distinct common and rare eye diseases. This algorithm utilized the RFMiD dataset of 3,200 fundus images and the Binary Relevance Method to detect diseases separately, ensuring expandability and avoiding incorrect correlations. Each detector, incorporating finely tuned hyperparameters to optimize performance, consisted of three feature components: A classical transfer learning CNN model, a two-stage CNN model, and a Siamese Network. The diagnosis was made using features extracted through this Trio-Model with Ensembled Machine Learning algorithms. The proposed model achieved an average accuracy of 97% and an AUC score of 0.96. Compared to past benchmark studies, an increase of over 10% in the F1-score was observed for most diseases. Furthermore, using the Siamese Network, the model successfully made predictions in diseases like optic disc pallor, which past studies failed to predict due to low confidence. This diagnostic tool presents a stable, adaptive, cost-effective, efficient, accessible, and fast solution for globalizing early detection of both common and rare diseases.
- World Health Organization, “Blindness and vision impairment,” 2023. [Online]. Available: https://www.who.int/news-room/fact-sheets/detail/blindness-and-visual-impairment. [Accessed: Jan. 1, 2023].
- M. C. V. Stella Mary, E. B. Rajsingh, and G. R. Naik, “Retinal fundus image analysis for diagnosis of glaucoma: A comprehensive survey,” IEEE Access, vol. 4, 2016.
- W. Bouthour, V. Biousse, and N. J. Newman, “Diagnosis of optic disc oedema: Fundus features, ocular imaging findings, and artificial intelligence,” Neuroophthalmology, vol. 47, no. 4, pp. 177-192, 2023.
- A. M. Menke, “Diagnostic error: Types and causes,” The Ophthalmic Risk Management Digest, vol. 26, no. 1, 2016.
- J. W. Bettman, B. H. Demorest, and E. R. Craven, “Risk management issues in the new managed care environment,” Survey of Ophthalmology, vol. 41, no. 3, pp. 268-270, 1996.
- N. Ramachandran, S. C. Hong, M. J. Sime, and G. A. Wilson, “Diabetic retinopathy screening using deep neural network,” Clinical & Experimental Ophthalmology, vol. 46, no. 4, pp. 412-416, 2018.
- A. Melo and H. Paulheim, “Local and global feature selection for multilabel classification with binary relevance: An empirical comparison on flat and hierarchical problems,” Artificial Intelligence Review, vol. 51, no. 1, pp. 33-60, 2019.
- G. Wu, R. Zheng, Y. Tian, and D. Liu, “Joint ranking SVM and binary relevance with robust low-rank learning for multi-label classification,” Neural Networks, vol. 122, pp. 24-39, 2020.
- Y. Ma and Y. Peng, “Lymph node detection method based on multisource transfer learning and convolutional neural network,” International Journal of Imaging Systems and Technology, vol. 30, no. 2, pp. 298-310, 2020.
- Y. Fu and C. Aldrich, “Froth image analysis by use of transfer learning and convolutional neural networks,” Minerals Engineering, vol. 115, pp. 68-78, 2018.
- I. Qureshi, J. Ma, and Q. Abbas, “Diabetic retinopathy detection and stage classification in eye fundus images using active deep learning,” Multimedia Tools and Applications, vol. 80, no. 8, pp. 11691-11721, 2021.
- T. K. Yoo, J. Y. Choi, and H. K. Kim, “Feasibility study to improve deep learning in OCT diagnosis of rare retinal diseases with few-shot classification,” Medical & Biological Engineering & Computing, vol. 59, no. 2, pp. 401-415, 2021.
- S.-C. Lin, Y.-C. I. Chang, and W.-N. Yang, “Meta-learning for imbalanced data and classification ensemble in binary classification,” Neurocomputing, vol. 73, no. 1-3, pp. 484-494, 2009.
- R. K. Sidhu, J. Sachdeva, and D. Katoch, “Segmentation of retinal blood vessels by a novel hybrid technique- principal component analysis (PCA) and contrast limited adaptive histogram equalization (CLAHE),” Microvascular Research, vol. 148, 2023.
- T.-Y. Lin, P. Goyal, R. Girshick, K. He, and P. Dollár, “Focal loss for dense object detection,” IEEE Transactions on Pattern Analysis and Machine Intelligence, vol. 42, no. 2, pp. 318-327, 2020.
- M. Sokolova, N. Japkowicz, and S. Szpakowicz, “Beyond accuracy, F-score and ROC: A family of discriminant measures for performance evaluation,” in AI 2006: Advances in Artificial Intelligence, Lecture Notes in Computer Science, vol. 4304, pp. 1015-1021, 2006.
- M. A. Rodríguez, H. AlMarzouqi, and P. Liatsis, “Multi-label retinal disease classification using transformers,” IEEE Journal of Biomedical and Health Informatics, vol. 27, no. 6, pp. 2739-2750, 2023.
- A. Soleimanipour, M. Azadbakht, and A. Rezaei Asl, “Cultivar identification of pistachio nuts in bulk mode through Efficientnet deep learning model,” Journal of Food Measurement and Characterization, vol. 16, no. 4, pp. 2545-2555, 2022. [Online]. Available: https://doi.org/10.1007/s11694-022-01367-5
- M. Tan and Q. V. Le, “EfficientNet: Improving accuracy and efficiency through AutoML and model scaling,” Google AI Blog, May 28, 2019. [Online]. Available: https://blog.research.google/2019/05/efficientnet-improving-accuracy-and.html?amp=1
- A. H. Williams, “Everything you did and didn’t know about PCA,” It’s Neuronal, Mar. 27, 2016. [Online]. Available: https://alexhwilliams.info/itsneuronalblog/2016/03/27/pca/
- R. Khandelwal, “One-shot learning with Siamese network,” Medium, 2021. [Online]. Available: https://medium.com/swlh/one-shot-learning-with-siamese-network-1c7404c35fda