Counteracting Data Bias and Class Imbalance—Towards a Useful and Reliable Retinal Disease Recognition System

Multiple studies presented satisfactory performances for the treatment of various ocular diseases. To date, there has been no study that describes a multiclass model, medically accurate, and trained on large diverse dataset. No study has addressed a class imbalance problem in one giant dataset origi...

Full description

Bibliographic Details
Main Authors: Adam R. Chłopowiec, Konrad Karanowski, Tomasz Skrzypczak, Mateusz Grzesiuk, Adrian B. Chłopowiec, Martin Tabakov
Format: Article
Language:English
Published: MDPI AG 2023-05-01
Series:Diagnostics
Subjects:
Online Access:https://www.mdpi.com/2075-4418/13/11/1904
_version_ 1797597728583188480
author Adam R. Chłopowiec
Konrad Karanowski
Tomasz Skrzypczak
Mateusz Grzesiuk
Adrian B. Chłopowiec
Martin Tabakov
author_facet Adam R. Chłopowiec
Konrad Karanowski
Tomasz Skrzypczak
Mateusz Grzesiuk
Adrian B. Chłopowiec
Martin Tabakov
author_sort Adam R. Chłopowiec
collection DOAJ
description Multiple studies presented satisfactory performances for the treatment of various ocular diseases. To date, there has been no study that describes a multiclass model, medically accurate, and trained on large diverse dataset. No study has addressed a class imbalance problem in one giant dataset originating from multiple large diverse eye fundus image collections. To ensure a real-life clinical environment and mitigate the problem of biased medical image data, 22 publicly available datasets were merged. To secure medical validity only Diabetic Retinopathy (DR), Age-Related Macular Degeneration (AMD) and Glaucoma (GL) were included. The state-of-the-art models ConvNext, RegNet and ResNet were utilized. In the resulting dataset, there were 86,415 normal, 3787 GL, 632 AMD and 34,379 DR fundus images. ConvNextTiny achieved the best results in terms of recognizing most of the examined eye diseases with the most metrics. The overall accuracy was 80.46 ± 1.48. Specific accuracy values were: 80.01 ± 1.10 for normal eye fundus, 97.20 ± 0.66 for GL, 98.14 ± 0.31 for AMD, 80.66 ± 1.27 for DR. A suitable screening model for the most prevalent retinal diseases in ageing societies was designed. The model was developed on a diverse, combined large dataset which made the obtained results less biased and more generalizable.
first_indexed 2024-03-11T03:09:35Z
format Article
id doaj.art-709b76efe60f471db832ff68b50137ab
institution Directory Open Access Journal
issn 2075-4418
language English
last_indexed 2024-03-11T03:09:35Z
publishDate 2023-05-01
publisher MDPI AG
record_format Article
series Diagnostics
spelling doaj.art-709b76efe60f471db832ff68b50137ab2023-11-18T07:42:41ZengMDPI AGDiagnostics2075-44182023-05-011311190410.3390/diagnostics13111904Counteracting Data Bias and Class Imbalance—Towards a Useful and Reliable Retinal Disease Recognition SystemAdam R. Chłopowiec0Konrad Karanowski1Tomasz Skrzypczak2Mateusz Grzesiuk3Adrian B. Chłopowiec4Martin Tabakov5Department of Artificial Intelligence, Wroclaw University of Science and Technology, Wybrzeże Wyspianskiego 27, 50-370 Wroclaw, PolandDepartment of Artificial Intelligence, Wroclaw University of Science and Technology, Wybrzeże Wyspianskiego 27, 50-370 Wroclaw, PolandFaculty of Medicine, Wroclaw Medical University, Wybrzeże Ludwika Pasteura 1, 50-367 Wroclaw, PolandDepartment of Artificial Intelligence, Wroclaw University of Science and Technology, Wybrzeże Wyspianskiego 27, 50-370 Wroclaw, PolandDepartment of Artificial Intelligence, Wroclaw University of Science and Technology, Wybrzeże Wyspianskiego 27, 50-370 Wroclaw, PolandDepartment of Artificial Intelligence, Wroclaw University of Science and Technology, Wybrzeże Wyspianskiego 27, 50-370 Wroclaw, PolandMultiple studies presented satisfactory performances for the treatment of various ocular diseases. To date, there has been no study that describes a multiclass model, medically accurate, and trained on large diverse dataset. No study has addressed a class imbalance problem in one giant dataset originating from multiple large diverse eye fundus image collections. To ensure a real-life clinical environment and mitigate the problem of biased medical image data, 22 publicly available datasets were merged. To secure medical validity only Diabetic Retinopathy (DR), Age-Related Macular Degeneration (AMD) and Glaucoma (GL) were included. The state-of-the-art models ConvNext, RegNet and ResNet were utilized. In the resulting dataset, there were 86,415 normal, 3787 GL, 632 AMD and 34,379 DR fundus images. ConvNextTiny achieved the best results in terms of recognizing most of the examined eye diseases with the most metrics. The overall accuracy was 80.46 ± 1.48. Specific accuracy values were: 80.01 ± 1.10 for normal eye fundus, 97.20 ± 0.66 for GL, 98.14 ± 0.31 for AMD, 80.66 ± 1.27 for DR. A suitable screening model for the most prevalent retinal diseases in ageing societies was designed. The model was developed on a diverse, combined large dataset which made the obtained results less biased and more generalizable.https://www.mdpi.com/2075-4418/13/11/1904deep learningmedical image classificationconvolutional neural networks
spellingShingle Adam R. Chłopowiec
Konrad Karanowski
Tomasz Skrzypczak
Mateusz Grzesiuk
Adrian B. Chłopowiec
Martin Tabakov
Counteracting Data Bias and Class Imbalance—Towards a Useful and Reliable Retinal Disease Recognition System
Diagnostics
deep learning
medical image classification
convolutional neural networks
title Counteracting Data Bias and Class Imbalance—Towards a Useful and Reliable Retinal Disease Recognition System
title_full Counteracting Data Bias and Class Imbalance—Towards a Useful and Reliable Retinal Disease Recognition System
title_fullStr Counteracting Data Bias and Class Imbalance—Towards a Useful and Reliable Retinal Disease Recognition System
title_full_unstemmed Counteracting Data Bias and Class Imbalance—Towards a Useful and Reliable Retinal Disease Recognition System
title_short Counteracting Data Bias and Class Imbalance—Towards a Useful and Reliable Retinal Disease Recognition System
title_sort counteracting data bias and class imbalance towards a useful and reliable retinal disease recognition system
topic deep learning
medical image classification
convolutional neural networks
url https://www.mdpi.com/2075-4418/13/11/1904
work_keys_str_mv AT adamrchłopowiec counteractingdatabiasandclassimbalancetowardsausefulandreliableretinaldiseaserecognitionsystem
AT konradkaranowski counteractingdatabiasandclassimbalancetowardsausefulandreliableretinaldiseaserecognitionsystem
AT tomaszskrzypczak counteractingdatabiasandclassimbalancetowardsausefulandreliableretinaldiseaserecognitionsystem
AT mateuszgrzesiuk counteractingdatabiasandclassimbalancetowardsausefulandreliableretinaldiseaserecognitionsystem
AT adrianbchłopowiec counteractingdatabiasandclassimbalancetowardsausefulandreliableretinaldiseaserecognitionsystem
AT martintabakov counteractingdatabiasandclassimbalancetowardsausefulandreliableretinaldiseaserecognitionsystem