Paper Info
Title
A Convolutional Neural Network Model Using Weighted Loss Function to Detect Diabetic Retinopathy
Abstract
AbstractNowadays, artificial intelligence (AI) provides tremendous prospects for driving future healthcare while empowering patients and service providers. The extensive use of digital healthcare produces a massive amount of multimedia healthcare data continuously (e.g., MRI, X-Ray, ultrasound images, etc.). Hence, it needs special data analytics techniques to provide a smart diagnosis to the patients. Recent advancements in artificial intelligence and machine learning techniques, particularly Deep learning (DL) methods, have demonstrated tremendous medical diagnosis progress and achievements. Diabetic Retinopathy (DR), cataract, macular degeneration, and glaucoma are the most common eye problems due to diabetes. Numerous models have been proposed using deep learning models to diagnose diabetic retinopathy, but no model is perfect for detecting DR diseases. This article presents a deep learning model to analyze diabetic retinopathy images to classify DR patients’ severity levels. The model applies a custom-weighted loss function in the model’s training and achieves 92.49% accuracy and a 0.945 Cohen Kappa score on test data. The model’s weighted average precision was 93%, recall 92%, and f1 score 93%. The model is compared with several state-of-the-art pre-trained models. We observe that the proposed model performs better in accuracy results and Cohen Kappa score.
Year
DOI
Venue
2022
10.1145/3470976
ACM Transactions on Multimedia Computing, Communications, and Applications
Keywords
DocType
Volume
Deep learning, diabetic retinopathy, computer vision, convolutional neural network
Journal
18
Issue
ISSN
Citations 
1s
1551-6857
0
PageRank 
References 
Authors
0.34
0
3
Name
Order
Citations
PageRank
Mehedi Masud17726.95
Mohammed F. Alhamid200.34
Yin Zhang37618.44