Abstract:Machine learning (ML) could have advantages over traditional statistical models in identifying risk factors. Using ML algorithms, our objective was to identify the most important variables associated with mortality after dementia diagnosis in the Swedish Registry for Cognitive/Dementia Disorders (SveDem). From SveDem, a longitudinal cohort of 28,023 dementia-diagnosed patients was selected for this study. Sixty variables were considered as potential predictors of mortality risk, such as age at dementia diagnosis, dementia type, sex, body mass index (BMI), mini-mental state examination (MMSE) score, time from referral to initiation of work-up, time from initiation of work-up to diagnosis, dementia medications, comorbidities, and some specific medications for chronic comorbidities (e.g., cardiovascular disease). We applied sparsity-inducing penalties for three ML algorithms and identified twenty important variables for the binary classification task in mortality risk prediction and fifteen variables to predict time to death. Area-under-ROC curve (AUC) measure was used to evaluate the classification algorithms. Then, an unsupervised clustering algorithm was applied on the set of twenty-selected variables to find two main clusters which accurately matched surviving and dead patient clusters. A support-vector-machines with an appropriate sparsity penalty provided the classification of mortality risk with accuracy = 0.7077, AUROC = 0.7375, sensitivity = 0.6436, and specificity = 0.740. Across three ML algorithms, the majority of the identified twenty variables were compatible with literature and with our previous studies on SveDem. We also found new variables which were not previously reported in literature as associated with mortality in dementia. Performance of basic dementia diagnostic work-up, time from referral to initiation of work-up, and time from initiation of work-up to diagnosis were found to be elements of the diagnostic process identified by the ML algorithms. The median follow-up time was 1053 (IQR = 516–1771) days in surviving and 1125 (IQR = 605–1770) days in dead patients. For prediction of time to death, the CoxBoost model identified 15 variables and classified them in order of importance. These highly important variables were age at diagnosis, MMSE score, sex, BMI, and Charlson Comorbidity Index with selection scores of 23%, 15%, 14%, 12% and 10%, respectively. This study demonstrates the potential of sparsity-inducing ML algorithms in improving our understanding of mortality risk factors in dementia patients and their application in clinical settings. Moreover, ML methods can be used as a complement to traditional statistical methods.

Predicting Risk of Dementia with Survival Machine Learning and Statistical Methods: Results on the English Longitudinal Study of Ageing Cohort

Predicting Alzheimers Disease Diagnosis Risk over Time with Survival Machine Learning on the ADNI Cohort

A comparison of machine learning methods for survival analysis of high-dimensional clinical data for dementia prediction

Comparative study of machine learning methods for modeling associations between risk factors and future dementia cases

Dementia prediction in the general population using clinically accessible variables: a proof-of-concept study using machine learning. The AGES-Reykjavik study

Predictors of Dementia in the Oldest Old: A Novel Machine Learning Approach

Dementia risk prediction in individuals with mild cognitive impairment: a comparison of Cox regression and machine learning models

Machine learning algorithms for identifying predictive variables of mortality risk following dementia diagnosis: a longitudinal cohort study

Predicting Alzheimer’s Disease from Cognitive Footprints in Mid and Late Life: How Much Can Register Data and Machine Learning Help?

Development of a Novel Dementia Risk Prediction Model in the General Population: A Large, Longitudinal, Population-Based Machine-Learning Study

Predicting Alzheimer's Disease with Interpretable Machine Learning.

Predicting Alzheimer's disease from cognitive footprints in mid and late life: How much can register data and machine learning help?

A Machine Learning Approach for Predicting Deterioration in Alzheimer's Disease

Using Machine Learning and Electronic Health Record (EHR) Data for the Early Prediction of Alzheimer's Disease and Related Dementias.

Data-driven Approaches to Executive Function Performance and Structure in Aging: Integrating Person-Centered Analyses and Machine Learning Risk Prediction.

Performance of Machine Learning Algorithms for Predicting Progression to Dementia in Memory Clinic Patients

Predicting mortality in the very old: a machine learning analysis on claims data

Predicting Dementia Risk for Elderly Community Dwellers in Primary Care Services Using Subgroup-specific Prediction Models

Early Prediction of Alzheimers Disease Leveraging Symptom Occurrences from Longitudinal Electronic Health Records of US Military Veterans

Comparison of Machine Learning Techniques for Mortality Prediction in a Prospective Cohort of Older Adults

Early Prediction of Alzheimer's Disease and Related Dementias Using Real-World Electronic Health Records.