# Developing an explainable machine learning model using body composition to predict cardiovascular mortality in initial dialysis patients: a multicenter study

## TL;DR

This study created a machine learning model using body composition data from CT scans to predict cardiovascular death in patients starting dialysis.

## Contribution

The model integrates CT-based body composition and clinical features for explainable cardiovascular mortality prediction in dialysis patients.

## Key findings

- CatBoost achieved high predictive accuracy with an AUC of 0.843 in internal validation and 0.799 in external validation.
- Key predictors included age, diabetes, CVD, skeletal muscle density, and hemoglobin.
- SHAP analysis revealed CVD, skeletal muscle density, and hemoglobin as major contributors to predictions.

## Abstract

Cardiovascular disease (CVD) is the leading cause of death in patients receiving dialysis, and accurate risk prediction at dialysis initiation remains limited. We developed and validated a machine learning model integrating CT-derived body composition features to predict CVD-related mortality in initial dialysis patients.

Patients initiating dialysis between 2014 and 2020 from three tertiary hospitals were used for model training and internal validation, with patients from a fourth center for external validation. Clinical characteristics and laboratory variables were collected, and body composition parameters were assessed using opportunistic CT scans. Feature selection was performed using univariable logistic regression and LASSO regression. Eight machine learning algorithms were trained, and model performance was assessed using discrimination, calibration, and decision curve analysis. Model interpretability was evaluated using Shapley Additive Explanations (SHAP), and a web-based risk calculator was developed.

Among 1051 incident dialysis patients, 645 were assigned to the training and internal validation cohorts and 406 to the external validation cohort. Eight key predictors were identified, including age, diabetes, CVD, history of cardiac intervention, dialysis modality, skeletal muscle density, hemoglobin, and serum creatinine. CatBoost demonstrated the best performance, with an area under the receiver operating characteristic curve of 0.843 in internal validation and 0.799 in external validation, along with good calibration and clinical net benefit. SHAP analysis identified CVD, skeletal muscle density, and hemoglobin as major contributors.

An explainable machine learning model incorporating CT-derived body composition features accurately predicts CVD-related mortality in initial dialysis patients. This model may facilitate early risk stratification and targeted prevention strategies at dialysis initiation.