融合参数与非参数模型,实现高效隐私保护的心血管风险预测
FedCVD++: Communication-Efficient Federated Learning for Cardiovascular Risk Prediction with Parametric and Non-Parametric Model Optimization
- 采用树模型子集采样与特征提取,降低通信开销70%
- 联邦XGBoost F1达0.80,超中心化模型,且通信量减少3.2倍
- 首次实现非参数模型在联邦医疗系统中的实用集成,适合多机构协作
心血管疾病每年导致全球超1700万人死亡,亟需隐私保护的预测系统。本文提出FedCVD++,一种融合参数模型(逻辑回归、SVM、神经网络)与非参数模型(随机森林、XGBoost)的增强型联邦学习框架,用于冠心病风险预测。为应对关键挑战,提出:(1) 树子集采样技术,使随机森林通信开销降低70%;(2) 基于XGBoost的特征提取,支持轻量级联邦集成;(3) 联邦SMOTE同步机制,解决跨机构类别不平衡问题。在Framingham数据集(4,238条记录)上评估显示:联邦XGBoost F1=0.80,优于中心化模型(F1=0.78),联邦随机森林F1=0.81,达到非联邦性能水平。同时,通信策略使带宽消耗降低3.2倍,保持95%准确率。相比现有框架,FedCVD++在多机构部署中实现最高15%的F1提升,具备更强可扩展性。该工作首次实现非参数模型在联邦医疗系统的实际集成,提供符合真实临床约束的隐私保护解决方案。
原文摘要 · Abstract (English)
Cardiovascular diseases (CVD) cause over 17 million deaths annually worldwide, highlighting the urgent need for privacy-preserving predictive systems. We introduce FedCVD++, an enhanced federated learning (FL) framework that integrates both parametric models (logistic regression, SVM, neural networks) and non-parametric models (Random Forest, XGBoost) for coronary heart disease risk prediction. To address key FL challenges, we propose: (1) tree-subset sampling that reduces Random Forest communication overhead by 70%, (2) XGBoost-based feature extraction enabling lightweight federated ensembles, and (3) federated SMOTE synchronization for resolving cross-institutional class imbalance. Evaluated on the Framingham dataset (4,238 records), FedCVD++ achieves state-of-the-art results: federated XGBoost (F1 = 0.80) surpasses its centralized counterpart (F1 = 0.78), and federated Random Forest (F1 = 0.81) matches non-federated performance. Additionally, our communication-efficient strategies reduce bandwidth consumption by 3.2X while preserving 95% accuracy. Compared to existing FL frameworks, FedCVD++ delivers up to 15% higher F1-scores and superior scalability for multi-institutional deployment. This work represents the first practical integration of non-parametric models into federated healthcare systems, providing a privacy-preserving solution validated under real-world clinical constraints.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。