针对无线网络不稳定导致的联邦学习偏差,提出客户端选择新方法提升鲁棒性。
Robust Federated Learning in Unreliable Wireless Networks: A Client Selection Approach
- 通过分析传输失败如何扭曲本地标签分布,揭示偏差来源。
- 无需资源调度,仅优化客户端选择概率即可缓解偏差问题。
- 适合在多标准网络(4G/5G/Wi-Fi)混合环境下部署的联邦学习场景。
联邦学习(FL)作为无线边缘训练深度神经网络的分布式学习范式,其性能常受无线传输不可靠和客户端间数据异质性影响。现有方法多依赖上行资源分配优化,假设客户端与服务器使用统一网络标准,但现实中移动客户端可能通过不同网络标准(如4G、5G、Wi-Fi)接入,配置各异,限制了服务器端修改的灵活性,难以适用于真实商业网络。本文首次理论分析了不可靠网络中传输失败如何扭曲本地样本的有效标签分布,导致偏离全局数据分布并引入收敛偏差。研究发现,合理设计客户端选择策略可有效缓解由网络不可靠性和数据异质性引发的偏差。基于此,我们提出FedCote客户端选择方法,无需依赖无线资源调度,仅通过优化客户端选择概率实现鲁棒性提升。实验表明,在频繁传输失败的不可靠网络下,FedCote在基于DNN的分类任务中表现稳定且有效。
原文摘要 · Abstract (English)
Federated learning (FL) has emerged as a promising distributed learning paradigm for training deep neural networks (DNNs) at the wireless edge, but its performance can be severely hindered by unreliable wireless transmission and inherent data heterogeneity among clients. Existing solutions primarily address these challenges by incorporating wireless resource optimization strategies, often focusing on uplink resource allocation across clients under the assumption of homogeneous client-server network standards. However, these approaches overlooked the fact that mobile clients may connect to the server via diverse network standards (e.g., 4G, 5G, Wi-Fi) with customized configurations, limiting the flexibility of server-side modifications and restricting applicability in real-world commercial networks. This paper presents a novel theoretical analysis about how transmission failures in unreliable networks distort the effective label distributions of local samples, causing deviations from the global data distribution and introducing convergence bias in FL. Our analysis reveals that a carefully designed client selection strategy can mitigate biases induced by network unreliability and data heterogeneity. Motivated by this insight, we propose FedCote, a client selection approach that optimizes client selection probabilities without relying on wireless resource scheduling. Experimental results demonstrate the robustness of FedCote in DNN-based classification tasks under unreliable networks with frequent transmission failures.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。