首个真实心血管病数据联邦学习基准,解决医疗数据隐私与异构难题
FedCVD: The First Real-World Federated Learning Benchmark on Cardiovascular Disease Data
- 基于七家机构真实数据构建联邦学习基准,包含心电图分类与超声心动图分割任务
- 发现真实数据中非独立同分布和长尾分布使联邦学习性能显著下降
- 为医疗领域联邦学习提供可复现的评估平台,适合算法研究者与临床数据团队
心血管疾病(CVDs)是全球首要死因,亟需早期诊断。机器学习可辅助早期识别,但依赖高质量大规模数据。由于医疗数据敏感性,各医疗机构难以共享数据以训练通用、无偏模型。联邦学习(FL)通过跨机构协作训练而不暴露原始数据,提供潜在解决方案。然而,目前针对心血管领域的联邦学习研究极少,且现有基准多为模拟数据,无法反映真实数据中的复杂异质性。本文提出首个真实世界心血管疾病检测联邦学习基准FedCVD,涵盖心电图(ECG)分类与超声心动图(ECHO)分割两大任务,基于七家机构的真实散点数据构建。大量实验表明,真实数据中的非独立同分布(non-IID)与长尾分布给联邦学习带来新挑战。项目代码与数据已开源:https://github.com/SMILELab-FL/FedCVD。
原文摘要 · Abstract (English)
Cardiovascular diseases (CVDs) are currently the leading cause of death worldwide, highlighting the critical need for early diagnosis and treatment. Machine learning (ML) methods can help diagnose CVDs early, but their performance relies on access to substantial data with high quality. However, the sensitive nature of healthcare data often restricts individual clinical institutions from sharing data to train sufficiently generalized and unbiased ML models. Federated Learning (FL) is an emerging approach, which offers a promising solution by enabling collaborative model training across multiple participants without compromising the privacy of the individual data owners. However, to the best of our knowledge, there has been limited prior research applying FL to the cardiovascular disease domain. Moreover, existing FL benchmarks and datasets are typically simulated and may fall short of replicating the complexity of natural heterogeneity found in realistic datasets that challenges current FL algorithms. To address these gaps, this paper presents the first real-world FL benchmark for cardiovascular disease detection, named FedCVD. This benchmark comprises two major tasks: electrocardiogram (ECG) classification and echocardiogram (ECHO) segmentation, based on naturally scattered datasets constructed from the CVD data of seven institutions. Our extensive experiments on these datasets reveal that FL faces new challenges with real-world non-IID and long-tail data. The code and datasets of FedCVD are available https://github.com/SMILELab-FL/FedCVD.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。