A Comprehensive Machine-Learning Model Applied to Magnetic Resonance Imaging (MRI) to Predict Alzheimer’s Disease (AD) in Older Subjects

Increasing evidence suggests the utility of magnetic resonance imaging (MRI) as an important technique for the diagnosis of Alzheimer’s disease (AD) and for predicting the onset of this neurodegenerative disorder. In this study, we present a sophisticated machine learning (ML) model of great accuracy to diagnose the early stages of AD. A total of 373 MRI tests belonging to 150 subjects (age ≥ 60) were examined and analyzed in parallel with fourteen distinct features related to standard AD diagnosis. Four ML models, such as naive Bayes (NB), artificial neural networks (ANN), K-nearest neighbor (KNN), and support-vector machines (SVM), and the receiver operating characteristic (ROC) curve metric were used to validate the model performance. Each model evaluation was done in three independent experiments. In the first experiment, a manual feature selection was used for model training, and ANN generated the highest accuracy in terms of ROC (0.812). In the second experiment, automatic feature selection was conducted by wrapping methods, and the NB achieved the highest ROC of 0.942. The last experiment consisted of an ensemble or hybrid modeling developed to combine the four models. This approach resulted in an improved accuracy ROC of 0.991. We conclude that the involvement of ensemble modeling, coupled with selective features, can predict with better accuracy the development of AD at an early stage.

[1]  Thomas J. Watson,et al.  An empirical study of the naive Bayes classifier , 2001 .

[2]  Edmond P. F. Lee,et al.  An ab initio study of RbO, CsO and FrO (X2Σ+; A2Π) and their cations (X3Σ−; A3Π) , 2001 .

[3]  M N Rossor,et al.  The prevalence and causes of dementia in people under the age of 65 years , 2003, Journal of neurology, neurosurgery, and psychiatry.

[4]  Isabelle Guyon,et al.  An Introduction to Variable and Feature Selection , 2003, J. Mach. Learn. Res..

[5]  Eric Bauer,et al.  An Empirical Comparison of Voting Classification Algorithms: Bagging, Boosting, and Variants , 1999, Machine Learning.

[6]  Yuehua Wu,et al.  Linear model selection by cross-validation , 2005 .

[7]  Kilian Q. Weinberger,et al.  Distance Metric Learning for Large Margin Nearest Neighbor Classification , 2005, NIPS.

[8]  Charles X. Ling,et al.  Using AUC and accuracy in evaluating learning algorithms , 2005, IEEE Transactions on Knowledge and Data Engineering.

[9]  David E. Goldberg,et al.  Genetic algorithms and Machine Learning , 1988, Machine Learning.

[10]  D. G. Clark,et al.  Early-Onset Dementia: Frequency and Causes Compared to Late-Onset Dementia , 2006, Dementia and Geriatric Cognitive Disorders.

[11]  Pedro Larrañaga,et al.  A review of feature selection techniques in bioinformatics , 2007, Bioinform..

[12]  M. Borga,et al.  Small baseline volume of left hippocampus is associated with subsequent conversion of MCI into dementia: The Göteborg MCI study , 2008, Journal of the Neurological Sciences.

[13]  K. Weigel,et al.  Machine learning classification procedure for selecting SNPs in genomic selection: application to early mortality in broilers. , 2007, Developments in biologicals.

[14]  M. Pencina,et al.  Evaluating the added predictive ability of a new marker: From area under the ROC curve to reclassification and beyond , 2008, Statistics in medicine.

[15]  Leif E. Peterson.  K-nearest neighbor , 2009, Scholarpedia.

[16]  C. Caltagirone,et al.  Alzheimer'S disease (ad) and mild cognitive impairment (mci) patients are characterized by increased bdnf serum levels , 2009, Alzheimer's & Dementia.

[17]  Sylvain Arlot,et al.  A survey of cross-validation procedures for model selection , 2009, 0907.4728.

[18]  Rong Chen,et al.  Machine-learning techniques for building a diagnostic model for very mild dementia , 2010, NeuroImage.

[19]  John G. Csernansky,et al.  Open Access Series of Imaging Studies: Longitudinal MRI Data in Nondemented and Demented Older Adults , 2010, Journal of Cognitive Neuroscience.

[20]  João Maroco,et al.  Data mining methods in the prediction of Dementia: A real-data comparison of the accuracy, sensitivity and specificity of linear discriminant analysis, logistic regression, neural networks, support vector machines, classification trees and random forests , 2011, BMC Research Notes.

[21]  Jing Shi,et al.  Fine tuning support vector machines for short-term wind speed forecasting , 2011 .

[22]  Colin Campbell,et al.  Learning with Support Vector Machines , 2011, Learning with Support Vector Machines.

[23]  Dino Isa,et al.  An enhanced Support Vector Machine classification framework by using Euclidean distance function for text document categorization , 2011, Applied Intelligence.

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

[25]  B. Miller,et al.  Machine learning approaches to diagnosis and laterality effects in semantic dementia discourse , 2014, Cortex.

[26]  Ferat Sahin,et al.  A survey on feature selection methods , 2014, Comput. Electr. Eng..

[27]  Jürgen Schmidhuber,et al.  Deep learning in neural networks: An overview , 2014, Neural Networks.

[28]  Rita Guerreiro,et al.  The age factor in Alzheimer’s disease , 2015, Genome Medicine.

[29]  Nirmalya Roy,et al.  Automated Functional and Behavioral Health Assessment of Older Adults with Dementia , 2016, 2016 IEEE First International Conference on Connected Health: Applications, Systems and Engineering Technologies (CHASE).

[30]  Matthew J. Brookes,et al.  A multi-layer network approach to MEG connectivity analysis , 2016, NeuroImage.

[31]  Bilwaj Gaonkar,et al.  Deep learning in the small sample size setting: cascaded feed forward neural networks for medical image segmentation , 2016, SPIE Medical Imaging.

[32]  Alison M Darcy,et al.  Machine Learning and the Profession of Medicine. , 2016, JAMA.

[33]  Dionysis Goularas,et al.  Distinguishing age-related cognitive decline from dementias: A study based on machine learning algorithms , 2017, Journal of Clinical Neuroscience.

[34]  C. Aditya,et al.  Devising an interpretable calibrated scale to quantitatively assess the dementia stage of subjects with alzheimer's disease: A machine learning approach , 2017 .

[35]  Enrico Pellegrini,et al.  Machine learning of neuroimaging for assisted diagnosis of cognitive impairment and dementia: A systematic review , 2018, Alzheimer's & dementia.

[36]  M. T. Pascarelli,et al.  P68-F Abnormalities of cortical neural synchronization mechanisms in patients with Alzheimer’s diseases dementia: An EEG study , 2019, Clinical Neurophysiology.

[37]  D. Facal,et al.  Machine learning approaches to studying the role of cognitive reserve in conversion from mild cognitive impairment to dementia , 2019, International journal of geriatric psychiatry.

[38]  S. Tayebati,et al.  Comparative Machine-Learning Approach: A Follow-Up Study on Type 2 Diabetes Predictions by Cross-Validation Methods , 2019, Machines.

[39]  Gopi Battineni,et al.  Machine learning in medicine: Performance calculation of dementia prediction by support vector machines (SVM) , 2019, Informatics in Medicine Unlocked.

[40]  Gopi Battineni,et al.  Comparative Machine Learning Approach in Dementia Patient Classification using Principal Component Analysis , 2020, ICAART.

[41]  C. Nalini,et al.  Late-Life Alzheimer's Disease (AD) Detection Using Pruned Decision Trees , 2020 .

[42]  Gopi Battineni,et al.  Applications of Machine Learning Predictive Models in the Chronic Disease Diagnosis , 2020, Journal of personalized medicine.

[43]  Shenghui Zhao,et al.  A new machine learning method for identifying Alzheimer's disease , 2020, Simul. Model. Pract. Theory.