Development and validation of prediction models for stroke and myocardial infarction in type 2 diabetes based on health insurance claims: does machine learning outperform traditional regression approaches?
Authors' conclusion
Discrimination performance of claims-based models reached a ceiling at around 0.09 AUPRC and 0.7 AUROC. While for AUROC this performance was comparable to existing epidemiological models incorporating clinical information, comparison of other, potentially more relevant metrics, such as AUPRC, sensitivity and Positive Predictive Value was hampered by lack of reporting in the literature. The fact that machine learning including deep learning methods did not outperform more traditional approaches may suggest that feature richness and complexity were exploited before the choice of algorithm could become critical to maximize performance. Future research might focus on the impact of different feature derivation approaches on performance ceilings. In the absence of other more powerful screening alternatives, applying transparent regression-based models in routine claims, though certainly imperfect, remains a promising scalable low-cost approach for population-based cardiovascular risk prediction and stratification.
Study context
- Population
- Patients with T2D in German health insurance claims 2014–2019; mean age 67.2 years
- Participants
- 371,006 patients
- Comparison
- Logistic regression vs machine learning
- Outcome
- Discrimination (AUPRC and AUROC) and calibration for prediction of MI and stroke
- Follow-up
- 3-year risk
Methodological detailstrain-test split, health insurance claims, 287 literature-derived variables
- Train-test split
- Health insurance claims
- 287 literature-derived variables
Performance
As reported in the abstract
- AUPRC, null model: Myocardial infarction (MI) / stroke
- 0.035 / 0.034
- AUPRC for MI: between multi-layer-perceptrons (MLP) and gradient boosting (GB)
- 0.082 (MLP) to 0.092 (GB)
- AUPRC for stroke: between MLP and GB
- 0.061 (MLP) to 0.073 (GB)
- AUROC, null models
- 0.5
- AUROC for MI: Random forests (RF), MLP, feature-tokenizer transformers (FTT) 0.70; all other models 0.71
- 0.70–0.71
- AUROC for stroke: between MLP and GB
- 0.66 (MLP) to 0.69 (GB)
Further findings
Incidence of MI and stroke
- MI
- 3.5% (n = 13,030)
- Stroke
- 3.4% (n = 12,701)
Original publication
Composed by AI from the published abstract and checked against it. How Briefs are made →