Repository logo
 

Genetic algorithm based deep learning neural network structure and hyperparameter optimization

Published version
Peer-reviewed

Change log

Abstract

jats:pAlzheimer’s disease is one of the major challenges of population ageing, and diagnosis and prediction of the disease through various biomarkers is the key. While the application of deep learning as imaging technologies has recently expanded across the medical industry, empirical design of these technologies is very difficult. The main reason for this problem is that the performance of the Convolutional Neural Networks (CNN) differ greatly depending on the statistical distribution of the input dataset. Different hyperparameters also greatly affect the convergence of the CNN models. With this amount of information, selecting appropriate parameters for the network structure has became a large research area. Genetic Algorithm (GA), is a very popular technique to automatically select a high-performance network architecture. In this paper, we show the possibility of optimising the network architecture using GA, where its search space includes both network structure configuration and hyperparameters. To verify the performance of our Algorithm, we used an amyloid brain image dataset that is used for Alzheimer’s disease diagnosis. As a result, our algorithm outperforms Genetic CNN by 11.73% on a given classification task.</jats:p>

Description

Keywords

46 Information and Computing Sciences, 4611 Machine Learning, Brain Disorders, Acquired Cognitive Impairment, Alzheimer's Disease, Neurodegenerative, Bioengineering, Dementia, Aging, Alzheimer's Disease including Alzheimer's Disease Related Dementias (AD/ADRD), Neurosciences, Neurological

Journal Title

Applied Sciences (Switzerland)

Conference Name

Journal ISSN

2076-3417
2076-3417

Volume Title

11

Publisher

MDPI AG
Sponsorship
National Research Foundation of Korea (NRF-2018R1A2B2008178)