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

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.

[1]  Jian Sun,et al.  Deep Residual Learning for Image Recognition , 2015, 2016 IEEE Conference on Computer Vision and Pattern Recognition (CVPR).

[2]  Fabio A. González,et al.  OCT-NET: A convolutional network for automatic classification of normal and diabetic macular edema using sd-oct volumes , 2018, 2018 IEEE 15th International Symposium on Biomedical Imaging (ISBI 2018).

[3]  K Ang,et al.  A NOTE ON UNIFORM DISTRIBUTION AND EXPERIMENTAL DESIGN , 1981 .

[4]  G. Lang.  Diabetic Macular Edema , 2012, Ophthalmologica.

[5]  Taimur Hassan,et al.  A practical approach to OCT based classification of Diabetic Macular Edema , 2017, 2017 International Conference on Signals and Systems (ICSigSys).

[6]  Mohammad Tariqul Islam,et al.  Classification of Retinal Diseases from OCT scans using Convolutional Neural Networks , 2018, 2018 10th International Conference on Electrical and Computer Engineering (ICECE).

[7]  Digvijay Singh,et al.  A novel deep learning based method for retinal lesion detection , 2017, 2017 International Conference on Advances in Computing, Communications and Informatics (ICACCI).

[8]  Anca L. Ralescu,et al.  Confusion Matrix-based Feature Selection , 2011, MAICS.

[9]  Geoffrey E. Hinton,et al.  ImageNet classification with deep convolutional neural networks , 2012, Commun. ACM.

[10]  Michael H. Goldbaum,et al.  Labeled Optical Coherence Tomography (OCT) and Chest X-Ray Images for Classification , 2018 .

[11]  Oscar J. Perdomo,et al.  A Novel Machine Learning Model Based on Exudate Localization to Detect Diabetic Macular Edema , 2016 .

[12]  Fred J. Hickernell,et al.  A generalized discrepancy and quadrature error bound , 1998, Math. Comput..

[13]  Kostas Marias,et al.  Extending 2-D Convolutional Neural Networks to 3-D for Advancing Deep Learning Cancer Classification With Application to MRI Liver Tumor Differentiation , 2019, IEEE Journal of Biomedical and Health Informatics.