Random forest model for feature-based Alzheimer’s disease conversion prediction from early mild cognitive impairment subjects
Open Access
- 29 April 2021
- journal article
- research article
- Published by Public Library of Science (PLoS) in PLOS ONE
- Vol. 16 (4) , e0244773
Abstract
Alzheimer’s Disease (AD) conversion prediction from the mild cognitive impairment (MCI) stage has been a difficult challenge. This study focuses on providing an individualized MCI to AD conversion prediction using a balanced random forest model that leverages clinical data. In order to do this, 383 Early Mild Cognitive Impairment (EMCI) patients were gathered from the Alzheimer’s Disease Neuroimaging Initiative (ADNI). Of these patients, 49 would eventually convert to AD (EMCI_C), whereas the remaining 334 did not convert (EMCI_NC). All of these patients were split randomly into training and testing data sets with 95 patients reserved for testing. Nine clinical features were selected, comprised of a mix of demographic, brain volume, and cognitive testing variables. Oversampling was then performed in order to balance the initially imbalanced classes prior to training the model with 1000 estimators. Our results showed that a random forest model was effective (93.6% accuracy) at predicting the conversion of EMCI patients to AD based on these clinical features. Additionally, we focus on explainability by assessing the importance of each clinical feature. Our model could impact the clinical environment as a tool to predict the conversion to AD from a prodromal stage or to identify ideal candidates for clinical trials.Keywords
All Related Versions
This publication has 14 references indexed in Scilit:
- A Novel Ensemble-Based Machine Learning Algorithm to Predict the Conversion From Mild Cognitive Impairment to Alzheimer's Disease Using Socio-Demographic Characteristics, Clinical Information, and Neuropsychological MeasuresFrontiers in Neurology, 2019
- Predicting the course of Alzheimer’s progressionBrain Informatics, 2019
- Random forest prediction of Alzheimer’s disease using pairwise selection from time series dataPLOS ONE, 2019
- Random Forest Algorithm for the Classification of Neuroimaging Data in Alzheimer's Disease: A Systematic ReviewFrontiers in Aging Neuroscience, 2017
- Relevant feature set estimation with a knock-out strategy and random forestsNeuroImage, 2015
- The age factor in Alzheimer’s diseaseGenome Medicine, 2015
- Comparison of DNA methylation profiles in human fetal and adult red blood cell progenitorsGenome Medicine, 2015
- Hierarchical Interactions Model for Predicting Mild Cognitive Impairment (MCI) to Alzheimer's Disease (AD) ConversionPLOS ONE, 2014
- Random Forest ensembles for detection and prediction of Alzheimer's disease with a good between-cohort robustnessNeuroImage: Clinical, 2014
- The Alzheimer's Disease Neuroimaging Initiative: A review of papers published since its inceptionAlzheimer's & Dementia, 2013