Deep learning strategies for predicting amputation free survival in patients with peripheral artery disease

This study aimed to address the limitations of traditional Cox proportional hazards (CPH) models in predicting amputation free survival in patients with peripheral artery disease (PAD) by benchmarking alternative survival machine learning (ML) models. It evaluated the performance of ML models in capturing non-linear relationships, re-assessing key predictors, and developing a prototype for a patient level risk prediction tool.

Methods

A retrospective, observational cohort dataset from Waikato Hospital (New Zealand) was analysed, including 2 366 symptomatic patients with PAD who underwent revascularisation between 2010 and 2021. Clinical, biological, and procedural information, and outcomes (amputations and deaths) were acquired. The study investigated non-competing risk models (CPH, conditional survival forest, random survival forest, and non-linear CPH [NLCH]) and competing risk models (Fine and Gray subdistribution hazard model and DeepHit model). Models were developed using fivefold cross validation (80/20 training–validation split) stratified by median time to amputation. Performances were evaluated using the concordance index and integrated Brier score.