arXiv:2501.09822cs.LGcs.NI2025-01被引 5

针对无线网络中数据异构问题,提出个性化联邦学习框架pFedWN

pFedWN: A Personalized Federated Learning Framework for D2D Wireless Networks with Heterogeneous Data

  • 基于设备间通信信道状态选择邻居,实现自适应协同学习
  • 采用EM算法计算客户端数据相似度,优化权重分配策略
  • 在动态无线环境下比传统方法更稳定高效,适合移动边缘场景

传统联邦学习在客户端数据异构时性能下降。为此,本文提出无服务器的个性化联邦学习框架pFedWN,融合设备到设备(D2D)无线信道条件,解决非独立同分布(non-IID)和不平衡数据带来的挑战。通过将问题分解为邻域选择与权重分配两部分:前者利用ISM等免许可频段的信道感知机制筛选邻居;后者采用期望最大化(EM)算法评估客户端数据相似性,实现最优权重分配。实验表明,pFedWN在非IID和不平衡数据下具备高效个性化学习能力,在动态不可预测的无线信道中显著优于现有FL与PFL方法,提升学习效能与鲁棒性。

原文摘要 · Abstract (English)

Traditional Federated Learning (FL) approaches often struggle with data heterogeneity across clients, leading to suboptimal model performance for individual clients. To address this issue, Personalized Federated Learning (PFL) emerges as a solution to the challenges posed by non-independent and identically distributed (non-IID) and unbalanced data across clients. Furthermore, in most existing decentralized machine learning works, a perfect communication channel is considered for model parameter transmission between clients and servers. However, decentralized PFL over wireless links introduces new challenges, such as resource allocation and interference management. To overcome these challenges, we formulate a joint optimization problem that incorporates the underlying device-to-device (D2D) wireless channel conditions into a server-free PFL approach. The proposed method, dubbed pFedWN, optimizes the learning performance for each client while accounting for the variability in D2D wireless channels. To tackle the formulated problem, we divide it into two sub-problems: PFL neighbor selection and PFL weight assignment. The PFL neighbor selection is addressed through channel-aware neighbor selection within unlicensed spectrum bands such as ISM bands. Next, to assign PFL weights, we utilize the Expectation-Maximization (EM) method to evaluate the similarity between clients' data and obtain optimal weight distribution among the chosen PFL neighbors. Empirical results show that pFedWN provides efficient and personalized learning performance with non-IID and unbalanced datasets. Furthermore, it outperforms the existing FL and PFL methods in terms of learning efficacy and robustness, particularly under dynamic and unpredictable wireless channel conditions.

联邦学习无线网络个性化D2D通信

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。