Artificial intelligence classification model for macular degeneration images: a robust optimization framework for residual neural networks

Abstract

Background: The prevalence of chronic disease is growing in aging societies, and artificial-intelligence-assisted interpretation of macular degeneration images is a topic that merits research. This study proposes a residual neural network (ResNet) model constructed using uniform design. The ResNet model is an artificial intelligence model that classifies macular degeneration images and can assist medical professionals in related tests and classification tasks, enhance confidence in making diagnoses, and reassure patients. However, the various hyperparameters in a ResNet lead to the problem of hyperparameter optimization in the model. This study employed uniform design-a systematic, scientific experimental design-to optimize the hyperparameters of the ResNet and establish a ResNet with optimal robustness.

Results: An open dataset of macular degeneration images ( https://data.mendeley.com/datasets/rscbjbr9sj/3 ) was divided into training, validation, and test datasets. According to accuracy, false negative rate, and signal-to-noise ratio, this study used uniform design to determine the optimal combination of ResNet hyperparameters. The ResNet model was tested and the results compared with results obtained in a previous study using the same dataset. The ResNet model achieved higher optimal accuracy (0.9907), higher mean accuracy (0.9848), and a lower mean false negative rate (0.015) than did the model previously reported. The optimal ResNet hyperparameter combination identified using the uniform design method exhibited excellent performance.

Conclusion: The high stability of the ResNet model established using uniform design is attributable to the study's strict focus on achieving both high accuracy and low standard deviation. This study optimized the hyperparameters of the ResNet model by using uniform design because the design features uniform distribution of experimental points and facilitates effective determination of the representative parameter combination, reducing the time required for parameter design and fulfilling the requirements of a systematic parameter design process.

Keywords: Hyperparameter optimization; Macular degeneration classification; Residual Neural Network; Uniform experimental design.