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...
Guardado en:
Autores principales: | , , , , , , , |
---|---|
Formato: | article |
Lenguaje: | EN |
Publicado: |
BMC
2021
|
Materias: | |
Acceso en línea: | https://doaj.org/article/2824bc515e364981b4db36e5370d1485 |
Etiquetas: |
Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
|
id |
oai:doaj.org-article:2824bc515e364981b4db36e5370d1485 |
---|---|
record_format |
dspace |
spelling |
oai:doaj.org-article:2824bc515e364981b4db36e5370d14852021-11-14T12:12:56ZArtificial intelligence classification model for macular degeneration images: a robust optimization framework for residual neural networks10.1186/s12859-021-04085-91471-2105https://doaj.org/article/2824bc515e364981b4db36e5370d14852021-11-01T00:00:00Zhttps://doi.org/10.1186/s12859-021-04085-9https://doaj.org/toc/1471-2105Abstract 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.Wen-Hsien HoTian-Hsiang HuangPo-Yuan YangJyh-Horng ChouHong-Siang HuangLi-Chung ChiFu-I ChouJinn-Tsong TsaiBMCarticleResidual Neural NetworkUniform experimental designHyperparameter optimizationMacular degeneration classificationComputer applications to medicine. Medical informaticsR858-859.7Biology (General)QH301-705.5ENBMC Bioinformatics, Vol 22, Iss S5, Pp 1-10 (2021) |
institution |
DOAJ |
collection |
DOAJ |
language |
EN |
topic |
Residual Neural Network Uniform experimental design Hyperparameter optimization Macular degeneration classification Computer applications to medicine. Medical informatics R858-859.7 Biology (General) QH301-705.5 |
spellingShingle |
Residual Neural Network Uniform experimental design Hyperparameter optimization Macular degeneration classification Computer applications to medicine. Medical informatics R858-859.7 Biology (General) QH301-705.5 Wen-Hsien Ho Tian-Hsiang Huang Po-Yuan Yang Jyh-Horng Chou Hong-Siang Huang Li-Chung Chi Fu-I Chou Jinn-Tsong Tsai Artificial intelligence classification model for macular degeneration images: a robust optimization framework for residual neural networks |
description |
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. |
format |
article |
author |
Wen-Hsien Ho Tian-Hsiang Huang Po-Yuan Yang Jyh-Horng Chou Hong-Siang Huang Li-Chung Chi Fu-I Chou Jinn-Tsong Tsai |
author_facet |
Wen-Hsien Ho Tian-Hsiang Huang Po-Yuan Yang Jyh-Horng Chou Hong-Siang Huang Li-Chung Chi Fu-I Chou Jinn-Tsong Tsai |
author_sort |
Wen-Hsien Ho |
title |
Artificial intelligence classification model for macular degeneration images: a robust optimization framework for residual neural networks |
title_short |
Artificial intelligence classification model for macular degeneration images: a robust optimization framework for residual neural networks |
title_full |
Artificial intelligence classification model for macular degeneration images: a robust optimization framework for residual neural networks |
title_fullStr |
Artificial intelligence classification model for macular degeneration images: a robust optimization framework for residual neural networks |
title_full_unstemmed |
Artificial intelligence classification model for macular degeneration images: a robust optimization framework for residual neural networks |
title_sort |
artificial intelligence classification model for macular degeneration images: a robust optimization framework for residual neural networks |
publisher |
BMC |
publishDate |
2021 |
url |
https://doaj.org/article/2824bc515e364981b4db36e5370d1485 |
work_keys_str_mv |
AT wenhsienho artificialintelligenceclassificationmodelformaculardegenerationimagesarobustoptimizationframeworkforresidualneuralnetworks AT tianhsianghuang artificialintelligenceclassificationmodelformaculardegenerationimagesarobustoptimizationframeworkforresidualneuralnetworks AT poyuanyang artificialintelligenceclassificationmodelformaculardegenerationimagesarobustoptimizationframeworkforresidualneuralnetworks AT jyhhorngchou artificialintelligenceclassificationmodelformaculardegenerationimagesarobustoptimizationframeworkforresidualneuralnetworks AT hongsianghuang artificialintelligenceclassificationmodelformaculardegenerationimagesarobustoptimizationframeworkforresidualneuralnetworks AT lichungchi artificialintelligenceclassificationmodelformaculardegenerationimagesarobustoptimizationframeworkforresidualneuralnetworks AT fuichou artificialintelligenceclassificationmodelformaculardegenerationimagesarobustoptimizationframeworkforresidualneuralnetworks AT jinntsongtsai artificialintelligenceclassificationmodelformaculardegenerationimagesarobustoptimizationframeworkforresidualneuralnetworks |
_version_ |
1718429336112988160 |