Media Haze Classification in Retinal Images Using Transfer Learning with Convolutional Neural Networks

被引:0
作者
O'Berry, Jonathan [1 ]
Liu, Xudong [1 ]
机构
[1] Univ N Florida, Sch Comp, Jacksonville, FL 32224 USA
来源
2024 INTERNATIONAL CONFERENCE ON MACHINE LEARNING AND APPLICATIONS, ICMLA | 2024年
关键词
media haze; artificial intelligence; deep learning; transfer learning;
D O I
10.1109/ICMLA61862.2024.00186
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
Media haze (MH) is a serious eye condition that can lead to various issues. This is especially true since MH is often an early warning sign of more severe conditions. If not treated quickly, it can potentially cause discomfort, blurred vision, or even blindness. Currently, MH is often detected by manually examining fundus images. This process allows for human error, where MH may not be detected. This study attempts to increase the detection and diagnosis of MH through our methods of preprocessing fundus images being examined, as well as the use of machine learning models. The models examined include LeNet-5, MLeNet (a modified LeNet), AlexNet, MobileNet, ResNet152v2, and VGG-16, where the former three models were trained on the RFMiD dataset, and the latter three were trained using pre-trained models fine-tuned on RFMiD. We found that all models using the proposed preprocessing demonstrated better performance than using the original images. The most effective model in this study was MobileNet, achieving 97.2% accuracy and 99.2% AUC score on the test set, outperforming the state-of-the-art model we found in the literature.
引用
收藏
页码:1204 / 1207
页数:4
相关论文
共 14 条
[1]  
[Anonymous], Keras Documentation: SGD
[2]  
[Anonymous], 2021, The World Factbook 2021
[3]  
Boyd K., 2022, What is uveitis?
[4]  
Greven M, 2016, INVEST OPHTH VIS SCI, V57
[5]  
Howard A.G., 2017, MOBILENETS EFFICIENT
[6]  
Kittusamy K., 2021, Terrain identification and land price estimation using deep learning, V2387
[7]   Gradient-based learning applied to document recognition [J].
Lecun, Y ;
Bottou, L ;
Bengio, Y ;
Haffner, P .
PROCEEDINGS OF THE IEEE, 1998, 86 (11) :2278-2324
[8]  
N. I. of Health, Cataracts
[9]   Multiple Ocular Disease Diagnosis Using Fundus Images Based on Multi-Label Deep Learning Classification [J].
Ouda, Osama ;
AbdelMaksoud, Eman ;
Abd El-Aziz, A. A. ;
Elmogy, Mohammed .
ELECTRONICS, 2022, 11 (13)
[10]  
Pachade Samiksha, 2020, IEEE DataPort