DOI 10.1007/s12265-026-10842-z
Cross-cohort Generalization for Heart Disease Prediction with Explainable AI.
Аннотация
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.
Предметные рубрики MeSH
Подготовка и перевод публикации
Обработка статьи · этапы, ошибки и управление
Ошибки и ограничения
История · было → стало
Состояние обновляется каждые 5 секунд, пока блок открыт. Процент показывается только там, где известен объём работы. Пауза останавливает продолжение этапа; текущий запрос к источнику может завершиться.
Тип публикации указан по данным источника. Уровень доказательности автоматически не присваивается.