This work is licensed under a Creative Commons Attribution-NonCommercial-ShareAlike 4.0 International License.
Comparative Study of Classification of Eye Disease Types Using DenseNet and EfficientNetB3
Corresponding Author(s) : Cahaya Jatmoko
Kinetik: Game Technology, Information System, Computer Network, Computing, Electronics, and Control,
Vol. 9, No. 3, August 2024
Abstract
The purpose of this research is to build a classification model that can perform the eye disease identification process so that the diagnosis of eye disease can be known and medical action can be taken as early as possible. This research used a dataset which has a total of 4217 eye image data and had 4 main classes namely cataract, diabetic retinopathy, glaucoma, and normal. With the data distribution of 1038 cataract images, 1098 diabetic retinopathy images, 1007 glaucoma images, and 1074 normal images, which of this data will be divided with a data percentage scheme of 50:10:40, 60:10:30, and 70:10:20, to see the results of which dataset division can produce optimal accuracy. In this study, the classification process will use 2 CNN transfer learning architectures, namely DenseNet, and efficientnetb3, which are both trained using the ImagiNet dataset. The results obtained after completing the testing process on the model built using the DenseNet architecture get optimal accuracy when using data division as much as 60:10:30, which is 78.59% while using the efficientnetb3 architecture optimal accuracy results when using the data division of 70:10:20, which is 95.66%. In research on the classification that had previously been done, it is very rare to find a classification process for eye disease types, therefore, in this study, the classification process will be carried out and provide an overview of the eye disease classification process with the CNN transfer learning method with more optimal accuracy results.
Keywords
Download Citation
Endnote/Zotero/Mendeley (RIS)BibTeX
- M. J. Burton et al., “The Lancet Global Health Commission on Global Eye Health: vision beyond 2020.,” Lancet Glob Health, vol. 9, no. 4, pp. e489–e551, Apr. 2021. https://doi.org/10.1016/S2214-109X(20)30488-5
- K. Varshney and K. Mishra, “An Analysis of Health Benefits of Carrot,” International Journal of Innovative Research in Engineering & Management, pp. 211–214, Feb. 2022. https://doi.org/10.55524/ijirem.2022.9.1.40
- H. Kandel et al., “Quality of life impact of eye diseases: a Save Sight Registries study.,” Clin Exp Ophthalmol, vol. 50, no. 4, pp. 386–397, May 2022. https://doi.org/10.1111/ceo.14050
- R. S. Douglas et al., “Teprotumumab for the Treatment of Active Thyroid Eye Disease,” New England Journal of Medicine, vol. 382, no. 4, pp. 341–352, Jan. 2020. https://doi.org/10.1056/NEJMoa1910434
- X. Chen, J. Xu, X. Chen, and K. Yao, “Cataract: Advances in surgery and whether surgery remains the only treatment in future,” Advances in Ophthalmology Practice and Research, vol. 1, no. 1, p. 100008, Nov. 2021. https://doi.org/10.1016/j.aopr.2021.100008
- Puneet, R. Kumar, and M. Gupta, “Optical coherence tomography image-based eye disease detection using deep convolutional neural network,” Health Inf Sci Syst, vol. 10, no. 1, p. 13, Jun. 2022. https://doi.org/10.1007/s13755-022-00182-y
- A. K. Schuster, C. Erb, E. M. Hoffmann, T. Dietlein, and N. Pfeiffer, “The Diagnosis and Treatment of Glaucoma.,” Dtsch Arztebl Int, vol. 117, no. 13, pp. 225–234, Mar. 2020. https://doi.org/10.3238/arztebl.2020.0225
- M. T. M. Wang, A. Muntz, B. Mamidi, J. S. Wolffsohn, and J. P. Craig, “Modifiable lifestyle risk factors for dry eye disease.,” Cont Lens Anterior Eye, vol. 44, no. 6, p. 101409, Dec. 2021. https://doi.org/10.1016/j.clae.2021.01.004
- H. H. Rashidi, N. K. Tran, E. V. Betts, L. P. Howell, and R. Green, “Artificial Intelligence and Machine Learning in Pathology: The Present Landscape of Supervised Methods,” Acad Pathol, vol. 6, p. 2374289519873088, Jan. 2019. https://doi.org/10.1177/2374289519873088
- N. Mahdi Abdulkareem and A. Mohsin Abdulazeez, “Machine Learning Classification Based on Radom Forest Algorithm: A Review,” International Journal of Science and Business, vol. 5, no. 2, pp. 128–142, 2021. https://doi.org/10.5281/zenodo.4471118
- M. M. Saritas and A. Yasar, “International Journal of Intelligent Systems and Applications in Engineering Performance Analysis of ANN and Naive Bayes Classification Algorithm for Data Classification,” Original Research Paper International Journal of Intelligent Systems and Applications in Engineering IJISAE, vol. 7, no. 2, pp. 88–91, 2019. https://doi.org/10.18201//ijisae.2019252786
- M. Kim et al., “Deep Learning in Medical Imaging,” Neurospine, vol. 16, no. 4, pp. 657–668, Dec. 2019, doi: 10.14245/ns.1938396.198.
- W. Lu, J. Li, J. Wang, and L. Qin, “A CNN-BiLSTM-AM method for stock price prediction,” Neural Comput Appl, vol. 33, no. 10, pp. 4741–4753, May 2021. https://doi.org/10.1007/s00521-020-05532-z
- X.-Q. Zhang, Y. Hu, Z.-J. Xiao, J.-S. Fang, R. Higashita, and J. Liu, “Machine Learning for Cataract Classification/Grading on Ophthalmic Imaging Modalities: A Survey,” Machine Intelligence Research, vol. 19, no. 3, pp. 184–208, Jun. 2022. https://doi.org/10.1007/s11633-022-1329-0
- L. Ren, J. Dong, X. Wang, Z. Meng, L. Zhao, and M. J. Deen, “A Data-Driven Auto-CNN-LSTM Prediction Model for Lithium-Ion Battery Remaining Useful Life,” IEEE Trans Industr Inform, vol. 17, no. 5, pp. 3478–3487, May 2021. https://doi.org/10.1109/TII.2020.3008223
- X. Jiang, Y. Wang, W. Liu, S. Li, and J. Liu, “CapsNet, CNN, FCN: Comparative Performance Evaluation for Image Classification,” Int J Mach Learn Comput, vol. 9, no. 6, pp. 840–848, Dec. 2019. https://doi.org/10.18178/ijmlc.2019.9.6.881
- A. Ramdan, A. Heryana, A. Arisal, R. B. S. Kusumo, and H. F. Pardede, “Transfer Learning and Fine-Tuning for Deep Learning-Based Tea Diseases Detection on Small Datasets,” in 2020 International Conference on Radar, Antenna, Microwave, Electronics, and Telecommunications (ICRAMET), IEEE, Nov. 2020, pp. 206–211. https://doi.org/10.1109/ICRAMET51080.2020.9298575
- M. A. Wakili et al., “Classification of Breast Cancer Histopathological Images Using DenseNet and Transfer Learning.,” Comput Intell Neurosci, vol. 2022, p. 8904768, Oct. 2022. https://doi.org/10.1155/2022/8904768
- G. Marques, D. Agarwal, and I. de la Torre Díez, “Automated medical diagnosis of COVID-19 through EfficientNet convolutional neural network,” Appl Soft Comput, vol. 96, p. 106691, Nov. 2020. https://doi.org/10.1016/j.asoc.2020.106691
- K. Thaiyalnayaki, “Classification of Diabetes Using Deep Learning and SVM Techniques,” Int J Curr Res Rev, vol. 13, no. 01, pp. 146–149, Jan. 2021. https://doi.org/10.31782/IJCRR.2021.13127
- X. Qadamboyevich and H. Abdullayev, “CLASSIFICATION OF EYE DISEASES WITH MOBILENETV3 AND EFFICIENTNETB0 MODELS,” DIGITAL TRANSFORMATION AND ARTIFICIAL INTELLIGENCE, vol. 1, no. 1, pp. 92–96, Apr. 2023, Accessed: Nov. 14, 2023. [Online]. Available: https://dtai.tsue.uz/index.php/dtai/article/view/v1i113
- P. Wang, E. Fan, and P. Wang, “Comparative analysis of image classification algorithms based on traditional machine learning and deep learning,” Pattern Recognit Lett, vol. 141, pp. 61–67, Jan. 2021. https://doi.org/10.1016/j.patrec.2020.07.042
- N. Dua, S. N. Singh, and V. B. Semwal, “Multi-input CNN-GRU based human activity recognition using wearable sensors,” Computing, vol. 103, no. 7, pp. 1461–1478, Jul. 2021. https://doi.org/10.1007/s00607-021-00928-8
- A. Çinar and M. Yildirim, “Detection of tumors on brain MRI images using the hybrid convolutional neural network architecture,” Med Hypotheses, vol. 139, p. 109684, Jun. 202. https://doi.org/10.1016/j.mehy.2020.109684
- G. Shrestha, Deepsikha, M. Das, and N. Dey, “Plant Disease Detection Using CNN,” in 2020 IEEE Applied Signal Processing Conference (ASPCON), IEEE, Oct. 2020, pp. 109–11. https://doi.org/10.1109/ASPCON49795.2020.9276722
- M. Desai and M. Shah, “An anatomization on breast cancer detection and diagnosis employing multi-layer perceptron neural network (MLP) and Convolutional neural network (CNN),” Clinical eHealth, vol. 4, pp. 1–11, Jan. 2021. https://doi.org/10.1016/j.ceh.2020.11.002
- F. Demir, A. M. Ismael, and A. Sengur, “Classification of Lung Sounds With CNN Model Using Parallel Pooling Structure,” IEEE Access, vol. 8, pp. 105376–105383, 2020. https://doi.org/0.1109/ACCESS.2020.3000111
- S. M. Hassan, A. K. Maji, M. Jasiński, Z. Leonowicz, and E. Jasińska, “Identification of Plant-Leaf Diseases Using CNN and Transfer-Learning Approach,” Electronics (Basel), vol. 10, no. 12, p. 1388, Jun. 2021. https://doi.org/10.3390/electronics10121388
- K. Thenmozhi and U. Srinivasulu Reddy, “Crop pest classification based on deep convolutional neural network and transfer learning,” Comput Electron Agric, vol. 164, p. 104906, Sep. 2019. https://doi.org/10.1016/j.compag.2019.104906
- T. Rahman et al., “Transfer Learning with Deep Convolutional Neural Network (CNN) for Pneumonia Detection Using Chest X-ray,” Applied Sciences, vol. 10, no. 9, p. 3233, May 2020. https://doi.org/10.3390/app10093233
- H. Sharma, M. Saraswat, A. Yadav, J. Kim, and J. Bansal, Congress on Intelligent Systems, vol. 1335, no. 1. in Advances in Intelligent Systems and Computing, vol. 1335. Singapore: Springer Singapore, 2021. https://doi.org/10.1007/978-981-33-6984-9
- P. Chlap, H. Min, N. Vandenberg, J. Dowling, L. Holloway, and A. Haworth, “A review of medical image data augmentation techniques for deep learning applications.,” J Med Imaging Radiat Oncol, vol. 65, no. 5, pp. 545–563, Aug. 2021. https://doi.org/10.1111/1754-9485.13261
References
M. J. Burton et al., “The Lancet Global Health Commission on Global Eye Health: vision beyond 2020.,” Lancet Glob Health, vol. 9, no. 4, pp. e489–e551, Apr. 2021. https://doi.org/10.1016/S2214-109X(20)30488-5
K. Varshney and K. Mishra, “An Analysis of Health Benefits of Carrot,” International Journal of Innovative Research in Engineering & Management, pp. 211–214, Feb. 2022. https://doi.org/10.55524/ijirem.2022.9.1.40
H. Kandel et al., “Quality of life impact of eye diseases: a Save Sight Registries study.,” Clin Exp Ophthalmol, vol. 50, no. 4, pp. 386–397, May 2022. https://doi.org/10.1111/ceo.14050
R. S. Douglas et al., “Teprotumumab for the Treatment of Active Thyroid Eye Disease,” New England Journal of Medicine, vol. 382, no. 4, pp. 341–352, Jan. 2020. https://doi.org/10.1056/NEJMoa1910434
X. Chen, J. Xu, X. Chen, and K. Yao, “Cataract: Advances in surgery and whether surgery remains the only treatment in future,” Advances in Ophthalmology Practice and Research, vol. 1, no. 1, p. 100008, Nov. 2021. https://doi.org/10.1016/j.aopr.2021.100008
Puneet, R. Kumar, and M. Gupta, “Optical coherence tomography image-based eye disease detection using deep convolutional neural network,” Health Inf Sci Syst, vol. 10, no. 1, p. 13, Jun. 2022. https://doi.org/10.1007/s13755-022-00182-y
A. K. Schuster, C. Erb, E. M. Hoffmann, T. Dietlein, and N. Pfeiffer, “The Diagnosis and Treatment of Glaucoma.,” Dtsch Arztebl Int, vol. 117, no. 13, pp. 225–234, Mar. 2020. https://doi.org/10.3238/arztebl.2020.0225
M. T. M. Wang, A. Muntz, B. Mamidi, J. S. Wolffsohn, and J. P. Craig, “Modifiable lifestyle risk factors for dry eye disease.,” Cont Lens Anterior Eye, vol. 44, no. 6, p. 101409, Dec. 2021. https://doi.org/10.1016/j.clae.2021.01.004
H. H. Rashidi, N. K. Tran, E. V. Betts, L. P. Howell, and R. Green, “Artificial Intelligence and Machine Learning in Pathology: The Present Landscape of Supervised Methods,” Acad Pathol, vol. 6, p. 2374289519873088, Jan. 2019. https://doi.org/10.1177/2374289519873088
N. Mahdi Abdulkareem and A. Mohsin Abdulazeez, “Machine Learning Classification Based on Radom Forest Algorithm: A Review,” International Journal of Science and Business, vol. 5, no. 2, pp. 128–142, 2021. https://doi.org/10.5281/zenodo.4471118
M. M. Saritas and A. Yasar, “International Journal of Intelligent Systems and Applications in Engineering Performance Analysis of ANN and Naive Bayes Classification Algorithm for Data Classification,” Original Research Paper International Journal of Intelligent Systems and Applications in Engineering IJISAE, vol. 7, no. 2, pp. 88–91, 2019. https://doi.org/10.18201//ijisae.2019252786
M. Kim et al., “Deep Learning in Medical Imaging,” Neurospine, vol. 16, no. 4, pp. 657–668, Dec. 2019, doi: 10.14245/ns.1938396.198.
W. Lu, J. Li, J. Wang, and L. Qin, “A CNN-BiLSTM-AM method for stock price prediction,” Neural Comput Appl, vol. 33, no. 10, pp. 4741–4753, May 2021. https://doi.org/10.1007/s00521-020-05532-z
X.-Q. Zhang, Y. Hu, Z.-J. Xiao, J.-S. Fang, R. Higashita, and J. Liu, “Machine Learning for Cataract Classification/Grading on Ophthalmic Imaging Modalities: A Survey,” Machine Intelligence Research, vol. 19, no. 3, pp. 184–208, Jun. 2022. https://doi.org/10.1007/s11633-022-1329-0
L. Ren, J. Dong, X. Wang, Z. Meng, L. Zhao, and M. J. Deen, “A Data-Driven Auto-CNN-LSTM Prediction Model for Lithium-Ion Battery Remaining Useful Life,” IEEE Trans Industr Inform, vol. 17, no. 5, pp. 3478–3487, May 2021. https://doi.org/10.1109/TII.2020.3008223
X. Jiang, Y. Wang, W. Liu, S. Li, and J. Liu, “CapsNet, CNN, FCN: Comparative Performance Evaluation for Image Classification,” Int J Mach Learn Comput, vol. 9, no. 6, pp. 840–848, Dec. 2019. https://doi.org/10.18178/ijmlc.2019.9.6.881
A. Ramdan, A. Heryana, A. Arisal, R. B. S. Kusumo, and H. F. Pardede, “Transfer Learning and Fine-Tuning for Deep Learning-Based Tea Diseases Detection on Small Datasets,” in 2020 International Conference on Radar, Antenna, Microwave, Electronics, and Telecommunications (ICRAMET), IEEE, Nov. 2020, pp. 206–211. https://doi.org/10.1109/ICRAMET51080.2020.9298575
M. A. Wakili et al., “Classification of Breast Cancer Histopathological Images Using DenseNet and Transfer Learning.,” Comput Intell Neurosci, vol. 2022, p. 8904768, Oct. 2022. https://doi.org/10.1155/2022/8904768
G. Marques, D. Agarwal, and I. de la Torre Díez, “Automated medical diagnosis of COVID-19 through EfficientNet convolutional neural network,” Appl Soft Comput, vol. 96, p. 106691, Nov. 2020. https://doi.org/10.1016/j.asoc.2020.106691
K. Thaiyalnayaki, “Classification of Diabetes Using Deep Learning and SVM Techniques,” Int J Curr Res Rev, vol. 13, no. 01, pp. 146–149, Jan. 2021. https://doi.org/10.31782/IJCRR.2021.13127
X. Qadamboyevich and H. Abdullayev, “CLASSIFICATION OF EYE DISEASES WITH MOBILENETV3 AND EFFICIENTNETB0 MODELS,” DIGITAL TRANSFORMATION AND ARTIFICIAL INTELLIGENCE, vol. 1, no. 1, pp. 92–96, Apr. 2023, Accessed: Nov. 14, 2023. [Online]. Available: https://dtai.tsue.uz/index.php/dtai/article/view/v1i113
P. Wang, E. Fan, and P. Wang, “Comparative analysis of image classification algorithms based on traditional machine learning and deep learning,” Pattern Recognit Lett, vol. 141, pp. 61–67, Jan. 2021. https://doi.org/10.1016/j.patrec.2020.07.042
N. Dua, S. N. Singh, and V. B. Semwal, “Multi-input CNN-GRU based human activity recognition using wearable sensors,” Computing, vol. 103, no. 7, pp. 1461–1478, Jul. 2021. https://doi.org/10.1007/s00607-021-00928-8
A. Çinar and M. Yildirim, “Detection of tumors on brain MRI images using the hybrid convolutional neural network architecture,” Med Hypotheses, vol. 139, p. 109684, Jun. 202. https://doi.org/10.1016/j.mehy.2020.109684
G. Shrestha, Deepsikha, M. Das, and N. Dey, “Plant Disease Detection Using CNN,” in 2020 IEEE Applied Signal Processing Conference (ASPCON), IEEE, Oct. 2020, pp. 109–11. https://doi.org/10.1109/ASPCON49795.2020.9276722
M. Desai and M. Shah, “An anatomization on breast cancer detection and diagnosis employing multi-layer perceptron neural network (MLP) and Convolutional neural network (CNN),” Clinical eHealth, vol. 4, pp. 1–11, Jan. 2021. https://doi.org/10.1016/j.ceh.2020.11.002
F. Demir, A. M. Ismael, and A. Sengur, “Classification of Lung Sounds With CNN Model Using Parallel Pooling Structure,” IEEE Access, vol. 8, pp. 105376–105383, 2020. https://doi.org/0.1109/ACCESS.2020.3000111
S. M. Hassan, A. K. Maji, M. Jasiński, Z. Leonowicz, and E. Jasińska, “Identification of Plant-Leaf Diseases Using CNN and Transfer-Learning Approach,” Electronics (Basel), vol. 10, no. 12, p. 1388, Jun. 2021. https://doi.org/10.3390/electronics10121388
K. Thenmozhi and U. Srinivasulu Reddy, “Crop pest classification based on deep convolutional neural network and transfer learning,” Comput Electron Agric, vol. 164, p. 104906, Sep. 2019. https://doi.org/10.1016/j.compag.2019.104906
T. Rahman et al., “Transfer Learning with Deep Convolutional Neural Network (CNN) for Pneumonia Detection Using Chest X-ray,” Applied Sciences, vol. 10, no. 9, p. 3233, May 2020. https://doi.org/10.3390/app10093233
H. Sharma, M. Saraswat, A. Yadav, J. Kim, and J. Bansal, Congress on Intelligent Systems, vol. 1335, no. 1. in Advances in Intelligent Systems and Computing, vol. 1335. Singapore: Springer Singapore, 2021. https://doi.org/10.1007/978-981-33-6984-9
P. Chlap, H. Min, N. Vandenberg, J. Dowling, L. Holloway, and A. Haworth, “A review of medical image data augmentation techniques for deep learning applications.,” J Med Imaging Radiat Oncol, vol. 65, no. 5, pp. 545–563, Aug. 2021. https://doi.org/10.1111/1754-9485.13261