Media Haze Classification in Retinal Images Using Transfer Learning with Convolutional Neural Networks. O'Berry*, J. & Liu, X. In Proceedings of the 23rd IEEE International Conference on Machine Learning and Applications (ICMLA), 2024. IEEE (Acceptance rate: <font color="red">22.5%</font>).
Media Haze Classification in Retinal Images Using Transfer Learning with Convolutional Neural Networks [link]Paper  Media Haze Classification in Retinal Images Using Transfer Learning with Convolutional Neural Networks [pdf]Paper  abstract   bibtex   2 downloads  
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.

Downloads: 2