Review of 3 Published Models for Performance Comparison
Risk score for the prediction of dementia risk in 20 years among middle aged people: a longitudinal, population-based study
This study follows patients for 20 years with an incident dementia rate of 4%. AUC ranged between 0.70-0.80, sensitivity was 0.77, specificity was 0.63, and the negative predictive value (NPV) was 0.98.
Features included age, sex, years of education, systolic blood pressure, BMI, total cholesterol, physical activity level, APOE $\varepsilon$ 4 status, and follow-up time.
| AUC | Sensitivity | Specificity | NPV |
|---|---|---|---|
| 0.70-0.80 | 0.77 | 0.63 | 0.98 |
External Validation of the eRADAR Risk Score for Detecting Undiagnosed Dementia in Two Real-World Healthcare Systems
This study follows patients for 12-months after a 2-year lookback window to diagnose incident dementia in the following 12 months. Features include EHR data (including vital signs, diagnoses, medications, and utilization in the prior 2 years). eRADAR risk scores were generated to aid in the identification of the 50% of dementia patients that otherwise go undiagnosed.
Metrics at eRADAR dementia risk score cut-off at the 90th percentile:
| AUC | Sensitivity | Specificity | PPV |
|---|---|---|---|
| 0.78-0.85 | 0.362-0.543 | 0.898-0.916 | 0.020-0.116 |
Early prediction of Alzheimer’s disease and related dementias using real-world electronic health records
Methodology takeaways
This study demonstrated that data-driven modeling/feature selection outperformed knowledge-driven feature selection. This indicates that ML approaches to ADRD diagnosis can aid clinicians in better predictive diagnostics. The researchers tested both gradient boosting trees and logistic regression models on these varying sets of features.
Knowledge-driven features from clinicians includes medical diagnoses of obesity, diabetes, hyperlipidemia, hypertension, heart disease, stroke, depression, anxiety, concussion, sleep disorders, periodontitis, smoking, and alcohol use; medication exposures, including nonsteroidal anti-inflammatory drugs (NSAIDs), statins, anticholinergics, hormone replacement therapies, antihypertensives, benzodiazepines, and proton pump inhibitors; the most recent vital signs and lab test results, including body mass index (BMI), systolic/diastolic blood pressure, total cholesterol, high-density lipoprotein, glucose, and hemoglobin A1C (HbA1c) in the observation period. The authors encoded diagnoses and medication histories as binary variables in the models. Measurements were categorized based on the reference normal range (e.g., abnormally low, normal, or abnormally high).
For the data-driven features, the authors used all variables captured by the EHRs, including demographic and behavioral variables, such as age, gender, race, ethnicity, marital status, and smoking status. They included all discrete diagnoses, all medications, and all procedure codes recorded in patients’ EHRs as categorical features. To address the sparsity of features, they grouped similar features.
We are especially interested in the 3-year prediction window given it offers a direct comparison to the cohort we have developed.
| Prediction Window (Years) | AUC | Sensitivity | Specificity | PPV | NPV |
|---|---|---|---|---|---|
| 0 | 0.877-0.939 | 0.814-0.862 | 0.781-0.866 | 0.316-0.439 | 0.972-0.981 |
| 1 | 0.854-0.906 | 0.785-0.836 | 0.774-0.827 | 0.337-0.409 | 0.961-0.972 |
| 3 | 0.841-0.884 | 0.772-0.815 | 0.764-0.805 | 0.372-0.444 | 0.947-0.958 |
| 5 | 0.830-0.858 | 0.750-0.800 | 0.742-0.786 | 0.429-0.491 | 0.932-0.942 |
Our Cohort and Models
The cohort we have developed focuses on clinical diagnoses alone, as medications and procedures were not included in the initial scope of this project. They may be worth considering in future research by the Lary Lab. We also limited our feature set to diagnoses closely related to the measurements initially identified by the Lary Lab in addition to a few extra diagnoses that were significant in the chi-square test. It may be worthwhile to expand the scope of diagnoses included in this cohort. For feasibility and proof of concept, we limited the number of diagnoses given the large number of patients we have, time constraints, and available computational power.
Feature selection is one area of exploration we did not spend as much time on and would recommend future groups investigate.
Model Performance Summary
| Metric | Logistic Regression | Random Forest | XGBoost |
|---|---|---|---|
| Accuracy | 0.6310 | 0.6281 | 0.6479 |
| Precision | 0.1311 | 0.1307 | 0.1342 |
| Recall | 0.7007 | 0.7045 | 0.6820 |
| F1 | 0.2209 | 0.2205 | 0.2243 |
| ROC-AUC | 0.7119 | 0.7118 | 0.7132 |
| Sensitivity (TPR) | 0.7007 | 0.7045 | 0.6820 |
| Specificity (TNR) | 0.6254 | 0.6220 | 0.6451 |
| NPV | 0.9628 | 0.9631 | 0.9618 |
| FNR | 0.2993 | 0.2955 | 0.3180 |
Confusion Matrices
| Logistic Regression | Random Forest | XGBoost | |
|---|---|---|---|
| TN | 140,584 | 139,819 | 145,031 |
| FP | 84,222 | 84,987 | 79,775 |
| FN | 5,429 | 5,360 | 5,767 |
| TP | 12,707 | 12,776 | 12,369 |
Best Hyperparameters
| Parameter | Logistic Regression | Random Forest | XGBoost |
|---|---|---|---|
solver |
liblinear | — | — |
penalty |
l1 | — | — |
C |
0.1 | — | — |
n_estimators |
— | 300 | 500 |
max_depth |
— | 10 | 3 |
min_samples_split |
— | 5 | — |
min_samples_leaf |
— | 5 | — |
max_features |
— | sqrt | — |
learning_rate |
— | — | 0.05 |
subsample |
— | — | 0.85 |
colsample_bytree |
— | — | 1.0 |
min_child_weight |
— | — | 5 |
gamma |
— | — | 0.1 |
| CV F1 | 0.6734 | 0.6726 | 0.6677 |
Currently, this model generally underperforms what is available in the literature, mainly due to the data that was available. That being said, we believe that more time spent on feature selection and finetuning the hyperparameters of these models could match or improve upon what is available in the literature.
| Model | Source | AUC | Sensitivity | Specificity | NPV |
|---|---|---|---|---|---|
| Kivipelto et al. | Literature | 0.70–0.80 | 0.77 | 0.63 | 0.98 |
| eRADAR | Literature | 0.78–0.85 | 0.36–0.54 | 0.90–0.92 | — |
| Li et al. (3-yr) | Literature | 0.841–0.884 | 0.772–0.815 | 0.764–0.805 | 0.947–0.958 |
| XGBoost (this work) | Ours | 0.713 | 0.682 | 0.645 | 0.962 |
Our NPV and Sensitivity were our closest metrics, while our AUC is with the bounds of some of these papers, abiet on the lower side.
ROC-AUC Curves

Precision-Recall Curves
