Semi-supervised Blindness Detection with Neural Network Ensemble
DOI:
https://doi.org/10.54097/hset.v12i.1448Keywords:
Machine learning, deep learning, neural network.Abstract
Diabetic retinopathy (DR), a common complication of diabetes mellitus, is a major cause of visual loss among the working-age population. Since DR vision loss is irreversible, early detection of DR is crucial for preventing vision loss in patients. However, manual detection of DR remains time costly and inefficient. In this paper, an ensemble of 6 pre-trained neural networks (including EfficientNets, ResNet, and Inception) are combined. The compatibility of different networks is tested by creating different combinations of networks and evaluating their relative performance. Pseudo-labelling is used to further increase accuracy. With a limited training data set of only 5592 images, the final neural network ensemble achieved an accuracy of 0.864.
Downloads
References
Wejdan L. Alyoubi, Wafaa M. Shalash, and Maysoon F. Abulkhair. Diabetic retinopathy detection through deep learning techniques: A review. Informatics in Medicine Unlocked, 20: 100377, 2020.
Friedman D. S. Lietman T. Congdon, N. G. Important causes of visual impairment in the world today. JAMA, 290: 2057–2060, 2003.
Gregori N. Diabetic retinopathy: Causes, symptoms, treatment. American Academy of Ophthalmology, 2021.
Cheung C. Y. Sim S. S. Tan P. C. Tan G. S. Wong T. Y. Goh, J. K. Retinal imaging techniques for diabetic retinopathy screening. Journal of diabetes science and technology, 10: 282–294, 2016.
R. Taylor and D. Batey. Handbook of retinal screening in diabetes. WileyBlackwell, 2012.
Kele Xu, Dawei Feng, and Haibo Mi. Deep convolutional neural networkbased early automated detection of diabetic retinopathy using fundus image. Molecules, 22 (12), 2017.
Mehdi Torabian Esfahani, Mahsa Ghaderi, and RJLEJPT Kafiyeh. Classification of diabetic and normal fundus images using new deep learning method. Leonardo Electronic Journal of Practices and Technologies, 17 (32): 233–248, 2018.
Harry Pratt, Frans Coenen, Deborah M Broadbent, Simon P Harding, and Yalin Zheng. Convolutional neural networks for diabetic retinopathy. Procedia computer science, 90: 200–205, 2016.
Xiaoliang Wang, Yongjin Lu, Yujuan Wang, and Wei-Bang Chen. Diabetic retinopathy stage classification using convolutional neural networks. In 2018 IEEE International Conference on Information Reuse and Integration (IRI), pages 465–471. IEEE, 2018.
Tao Li, Yingqi Gao, Kai Wang, Song Guo, Hanruo Liu, and Hong Kang. Diagnostic assessment of deep learning algorithms for diabetic retinopathy screening. Information Sciences, 501: 511–522, 2019.
Kaggle Dataset [Online]. Available www.kaggle.com/competitions/aptos2019-blindness-detection
Mingxing Tan and Quoc Le. Efficientnet: Rethinking model scaling for convolutional neural networks. In International conference on machine learning, pages 6105–6114. PMLR, 2019.
Mingxing Tan and Quoc Le. Efficientnetv2: Smaller models and faster training. In International Conference on Machine Learning, pages 10096–10106. PMLR, 2021.
Kaiming He, Xiangyu Zhang, Shaoqing Ren, and Jian Sun. Deep residual learning for image recognition, 2015.
Christian Szegedy, Vincent Vanhoucke, Sergey Ioffe, Jonathon Shlens, and Zbigniew Wojna. Rethinking the inception architecture for computer vision, 2015.
Dong hyun Lee. Pseudo-label: The simple and efficient semi-supervised learning method for deep neural networks, 2013.
Downloads
Published
Issue
Section
License
This work is licensed under a Creative Commons Attribution-NonCommercial 4.0 International License.
