A team at Beijing Chaoyang Hospital builds and compares five supervised machine learning algorithms using clinical, echocardiographic, and hemodynamic features. They identify six key predictors via LASSO, train models with logistic regression, SVM, random forest, XGBoost, and decision tree, and use SHAP to interpret the best model’s decisions in predicting BPA outcomes.
Key points
- Six predictors selected by LASSO: occlusive lesion proportion, TAPSE/PASP, 6MWD, RVESA, TR severity, PVR.
- Logistic regression with L2 regularisation outperforms other ML models, achieving test AUC of 0.865, accuracy 0.848, sensitivity 0.950.
- SHAP analysis identifies occlusive lesion proportion as the most influential feature driving BPA response predictions.
Why it matters: A reliable ML tool for preoperative BPA response prediction can enhance patient selection, reduce procedural risks, and improve outcomes in CTEPH management.
Q&A
- What is CTEPH?
- How does balloon pulmonary angioplasty (BPA) work?
- What role does LASSO feature selection play?
- What are SHAP values?