<p>This paper considers the challenge of clinical reliability and generalization in heart disease diagnosis, where traditional ensembles often overlook feature heterogeneity and lack uncertainty quantification. We propose an uncertainty-aware feature-weighted ensemble (UAFE) framework designed for robust cardiovascular prediction. UAFE integrates three core components: (1) Feature Stratification to train specialized learners on importance-based subgroups; (2) Dynamic uncertainty weighting to adaptively adjust model contributions based on prediction disagreement; and (3) Neighborhood refinement to rectify decisions for high-uncertainty samples using local geometric priors. Comprehensive experiments on a multi-center dataset demonstrate that UAFE outperforms state-of-the-art baselines with an accuracy of 0.8660. Furthermore, leave-one-center-out (LOCO) validation across four international clinical sites shows that UAFE maintains superior stability (mean accuracy 0.8332) against institutional distribution shifts, establishing its effectiveness for deployment in unseen clinical environments.</p>

错误:搜索内容不能为空,请输入英文关键词
错误:关键词超出字数限制,请精简
高级检索

Uncertainty-aware feature-weighted ensemble framework for heart disease prediction

  • Xinling Wang,
  • Yangrui Fan,
  • Minglian Yu,
  • Fujiang Yuan

摘要

This paper considers the challenge of clinical reliability and generalization in heart disease diagnosis, where traditional ensembles often overlook feature heterogeneity and lack uncertainty quantification. We propose an uncertainty-aware feature-weighted ensemble (UAFE) framework designed for robust cardiovascular prediction. UAFE integrates three core components: (1) Feature Stratification to train specialized learners on importance-based subgroups; (2) Dynamic uncertainty weighting to adaptively adjust model contributions based on prediction disagreement; and (3) Neighborhood refinement to rectify decisions for high-uncertainty samples using local geometric priors. Comprehensive experiments on a multi-center dataset demonstrate that UAFE outperforms state-of-the-art baselines with an accuracy of 0.8660. Furthermore, leave-one-center-out (LOCO) validation across four international clinical sites shows that UAFE maintains superior stability (mean accuracy 0.8332) against institutional distribution shifts, establishing its effectiveness for deployment in unseen clinical environments.