用联邦近端优化在不泄露隐私前提下,提升异构医疗数据的心脏病预测准确率。
Federated Proximal Optimization for Privacy-Preserving Heart Disease Prediction: A Controlled Simulation Study on Non-IID Clinical Data
- 采用联邦近端优化缓解不同医院数据分布差异带来的模型漂移问题。
- 在非独立同分布数据上达到85.00%准确率,优于中心化学习和本地模型。
- 适合关注医疗数据隐私与协同建模的医院技术人员参考。
医疗单位拥有宝贵的患者数据,可助力诊断模型优化,但受HIPAA、GDPR等隐私法规限制,无法直接共享数据。联邦学习通过无需集中原始数据即可协同训练模型,提供了解决方案。然而,临床数据因人口差异、疾病流行率及机构实践不同,天然具有非独立同分布(non-IID)特性。本文基于UCI心脏病数据集开展仿真研究,从克利夫兰诊所数据中模拟四个异构医院客户端,通过基于人口统计学的分层生成真实感非独立同分布数据划分。实验表明,使用近端参数μ=0.05的联邦近端优化(FedProx)可实现85.00%的准确率,优于中心化学习(83.33%)和孤立本地模型(平均78.45%),且不暴露患者隐私。通过50次独立运行的充分消融实验与统计验证,证明近端正则化在异构环境中有效抑制客户端漂移。本概念验证研究为真实世界联邦医疗系统提供了算法洞见与部署指导,结果可直接应用于医院信息化管理者推进隐私保护协作学习。
原文摘要 · Abstract (English)
Healthcare institutions have access to valuable patient data that could be of great help in the development of improved diagnostic models, but privacy regulations like HIPAA and GDPR prevent hospitals from directly sharing data with one another. Federated Learning offers a way out to this problem by facilitating collaborative model training without having the raw patient data centralized. However, clinical datasets intrinsically have non-IID (non-independent and identically distributed) features brought about by demographic disparity and diversity in disease prevalence and institutional practices. This paper presents a comprehensive simulation research of Federated Proximal Optimization (FedProx) for Heart Disease prediction based on UCI Heart Disease dataset. We generate realistic non-IID data partitions by simulating four heterogeneous hospital clients from the Cleveland Clinic dataset (303 patients), by inducing statistical heterogeneity by demographic-based stratification. Our experimental results show that FedProx with proximal parameter mu=0.05 achieves 85.00% accuracy, which is better than both centralized learning (83.33%) and isolated local models (78.45% average) without revealing patient privacy. Through generous sheer ablation studies with statistical validation on 50 independent runs we demonstrate that proximal regularization is effective in curbing client drift in heterogeneous environments. This proof-of-concept research offers algorithmic insights and practical deployment guidelines for real-world federated healthcare systems, and thus, our results are directly transferable to hospital IT-administrators, implementing privacy-preserving collaborative learning.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。