Joint Optimization for Federated Learning and Transmission over Unreliable Wireless Networks with Heterogeneous Data

📅 2026-09-12
📈 Citations: 0
Influential: 0
📄 PDF
🤖 AI Summary
为解决无线联邦学习中数据异构性和传输错误问题,提出FedRW框架,并通过联合优化学习、路径选择和传输参数来提高训练质量和收敛速度。
📝 Abstract
In wireless federated learning (FL), data heterogeneity and multiple local updates induce client drift, degrading model convergence. It is further affected by unreliable wireless links, as transmission errors may invalidate model updates. To address these challenges, we propose a federated random walk averaging (FedRW) framework, which is a variant of federated averaging (FedAvg) that mitigates data heterogeneity by updating models along random walk (RW) paths and aggregating them at the server. Model parameters are transmitted in packets with retransmission support to improve training quality by mitigating wireless errors along RW paths. Meanwhile, wireless transmission delays hinder the exploration of FedRW. To this end, we formulate a joint optimization problem that integrates learning, RW path selection, and transmission parameter tuning, aiming to minimize the training loss under delay constraints. By deriving an upper bound on the expected convergence of FedRW over unreliable wireless networks, we reduce the problem to a general form agnostic to task type and model architecture. A distributed solution is then proposed, in which the server or clients optimize packet size and maximum number of retransmissions locally, and efficiently select reliable and expandable next-hop nodes via a resilience-aware beam search with dynamic pruning. Simulation results show that FedRW achieves 2.26%-9% higher accuracy than state-of-the-art baselines under high data heterogeneity. Furthermore, the jointly optimized FedRW yields at least 2.78% higher accuracy and faster convergence compared to baselines.
Problem

Research questions and friction points this paper is trying to address.

Federated Learning
Data Heterogeneity
Wireless Transmission
Client Drift
Transmission Errors
Innovation

Methods, ideas, or system contributions that make the work stand out.

Federated Random Walk Averaging (FedRW)
Joint Optimization
Wireless Transmission Errors
Heterogeneous Data
Resilience-Aware Beam Search
C
Changheng Wang
Key Laboratory of Universal Wireless Communications, Ministry of Education, School of Information and Communication Engineering, Beijing University of Posts and Telecommunications, Beijing 100876, China
X
Xianchao Zhang
Provincial Key Laboratory of Multimodal Perceiving and Intelligent Systems, Jiaxing University, Jiaxing 314001, China
Z
Zhiqing Wei
Key Laboratory of Universal Wireless Communications, Ministry of Education, School of Information and Communication Engineering, Beijing University of Posts and Telecommunications, Beijing 100876, China
L
Lingzhu Zhao
School of Electronics and Information, Northwestern Polytechnical University, Xi’an 710000, China
Z
Zhongming Yang
School of Information Science and Engineering, Southeast University, Nanjing 210096, China
Z
Zhiyong Feng
Key Laboratory of Universal Wireless Communications, Ministry of Education, School of Information and Communication Engineering, Beijing University of Posts and Telecommunications, Beijing 100876, China