Download PDFOpen PDF in browser

Classification of Stages Diabetic Retinopathy Using MobileNetV2 Model

11 pagesPublished: January 16, 2022

Abstract

Diabetic retinopathy (DR) is a complication of diabetes mellitus that causes retinal damage that can lead to vision loss if not detected and treated promptly. The common diagnosis stages of the disease take time, effort, and cost and can be misdiagnosed. In the recent period with the explosion of artificial intelligence, deep learning has become the most popular tool with high performance in many fields, especially in the analysis and classification of medical images. The Convolutional Neural Network (CNN) is more widely used as a deep learning method in medical imaging analysis with highly effective. In this paper, the five-stage image of modern DR (healthy, mild, moderate, severe, and proliferative) can be detected and classified using the deep learning technique. After cross-validation training and testing on the corresponding 5,590-image dataset, a pre-MobileNetV2 training model is proposed in classifying stages of diabetic retinopathy. The average accuracy of the model achieved was 93.89% with the precision of 94.00%, recall 92.00% and f1-score 90.00%. The corresponding thermal image is also given to help experts for evaluating the influence of the retina in each different stage.

Keyphrases: balancing dataset, deep learning, diabetic retinopathy, mobilenetv2

In: Tich Thien Truong, Trung Nghia Tran, Thanh Nha Nguyen and Quoc Khai Le (editors). Proceedings of International Symposium on Applied Science 2021, vol 4, pages 147-157.

BibTeX entry
@inproceedings{ISAS2021:Classification_Stages_Diabetic_Retinopathy,
  author    = {Hoang Nhut Huynh and Minh Thanh Do and Gia Thinh Huynh and Anh Tu Tran and Trung Nghia Tran},
  title     = {Classification of Stages Diabetic Retinopathy Using MobileNetV2 Model},
  booktitle = {Proceedings of International Symposium on Applied Science 2021},
  editor    = {Tich Thien Truong and Trung Nghia Tran and Thanh Nha Nguyen and Quoc Khai Le},
  series    = {Kalpa Publications in Engineering},
  volume    = {4},
  publisher = {EasyChair},
  bibsource = {EasyChair, https://easychair.org},
  issn      = {2515-1770},
  url       = {/publications/paper/k4C9},
  doi       = {10.29007/h46n},
  pages     = {147-157},
  year      = {2022}}
Download PDFOpen PDF in browser