针对医疗联邦学习中数据偏差问题,提出自适应客户端选择算法提升模型精度与公平性。
BACSA: A Bias-Aware Client Selection Algorithm for Privacy-Preserving Federated Learning in Wireless Healthcare Networks
- 根据客户端数据分布特征检测偏差,动态筛选最优参与方
- 在不同数据分布下收敛速度和准确率均优于现有方法
- 兼顾隐私、网络约束与公平性,适合医疗场景部署
联邦学习(FL)在医疗领域具有变革性潜力,可在保护用户隐私的同时实现跨分散数据源的协同建模。然而,由于参与客户端间存在非独立同分布(non-IID)数据偏差,实际应用中模型性能迅速下降,严重影响准确率与泛化能力。为此,本文提出偏见感知客户端选择算法(BACSA),通过分析模型参数与类别样本分布的相关性,检测用户数据偏差,并基于偏差特征与无线网络约束,构建混合整数非线性优化问题,实现客户端的智能选择。实验在多种数据分布(包括狄利克雷分布与类别约束场景)下验证了BACSA在收敛性与准确性上的优越表现。同时,研究揭示了精度、公平性与网络约束间的权衡关系,证明该算法具备强适应性与鲁棒性,适用于对服务质量、隐私与安全要求极高的医疗应用场景。
原文摘要 · Abstract (English)
Federated Learning (FL) has emerged as a transformative approach in healthcare, enabling collaborative model training across decentralized data sources while preserving user privacy. However, performance of FL rapidly degrades in practical scenarios due to the inherent bias in non Independent and Identically distributed (non-IID) data among participating clients, which poses significant challenges to model accuracy and generalization. Therefore, we propose the Bias-Aware Client Selection Algorithm (BACSA), which detects user bias and strategically selects clients based on their bias profiles. In addition, the proposed algorithm considers privacy preservation, fairness and constraints of wireless network environments, making it suitable for sensitive healthcare applications where Quality of Service (QoS), privacy and security are paramount. Our approach begins with a novel method for detecting user bias by analyzing model parameters and correlating them with the distribution of class-specific data samples. We then formulate a mixed-integer non-linear client selection problem leveraging the detected bias, alongside wireless network constraints, to optimize FL performance. We demonstrate that BACSA improves convergence and accuracy, compared to existing benchmarks, through evaluations on various data distributions, including Dirichlet and class-constrained scenarios. Additionally, we explore the trade-offs between accuracy, fairness, and network constraints, indicating the adaptability and robustness of BACSA to address diverse healthcare applications.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。