科研速览 · Science Skim继续刷下去 · Keep skimming →
◆ Journal of cardiovascular translational research2026-09-23

Cross-cohort Generalization for Heart Disease Prediction with Explainable AI.

Prosper Ughakpoteni, Yaseen Akhtar, Ahmad Chaddad, Sarah Alkhodair, Tareef Daqqaq

原始摘要(英文原文)· Original abstract
We propose CardioTransfer-X, a cross-cohort transfer learning framework within a related clinical benchmark family for tabular CVD risk prediction that yields performance comparable to training from scratch while preserving transparency. Predictive models are pre-trained on a composite multi-hospital heart disease dataset to learn generalized risk patterns, then fine-tuned on the smaller, distributionally distinct Cleveland Heart Disease dataset. To mitigate dataset shift and class imbalance, we introduce a stability-aware, correlation-based feature selection strategy and apply SMOTE only during training. CardioTransfer-X is instantiated with XGBoost, TabNet, and a MLP, enabling systematic comparison across tree-based and neural models. SHAP-based explanations provide global interpretability across architectures, with consistent feature importance rankings before and after fine-tuning. Fine-tuned XGBoost achieves 80.65±4.08% accuracy and 81.74±3.71% F1-score on the Cleveland dataset while maintaining stable source-domain performance, showing performance competitive with training from scratch, while shifting the decision boundary toward higher sensitivity and maintaining clinically meaningful feature attributions.
读原文 · Read the paper ↗

AI 追问PRO

登录后使用 AI 追问

讨论区

登录后参与讨论

相关论文 · Related

Cross-cohort Generalization for Heart Disease Prediction with Explainable AI. — 科研速览 Science Skim