arXiv Machine Learning

Uncertainty-Aware Federated Learning for Joint Communication and Computation in XR Systems

The paper proposes an AI-native, robust network for sixth-generation (6G) communication systems, where multiple base stations serve extended reality (XR) devices. It introduces a distributed deep learning approach that jointly manages transmit power, computing power, and computation allocation while accounting for uncertainties in channel and computing states. The method uses a deep neural network trained centrally or via federated learning, injecting uncertainty samples after the DNN output to minimize worst-case delay under power and computation constraints, achieving near-centralized performance with improved privacy and scalability.

arXiv Machine Learning
Aug 31

Ampere: Communication-Efficient and High-Accuracy Split Federated Learning

Ampere is a new split federated learning system that reduces both on‑device computation and device‑server communication while improving accuracy. It trains device and server blocks sequentially with local losses, eliminating gradient transfers, and uses a lightweight auxiliary network to consolidate activations into a single transfer. Experiments on CNNs and Transformers show up to 11.70 pp accuracy gains, 18.6× faster training, 911× less communication, and 14.5× less computation compared to state‑of‑the‑art SFL baselines.

By Zihan Zhang, Leon Wong, Blesson Varghese
arXiv Machine Learning
Aug 20

GQ-FSL: Green Quantized Federated Split Learning Framework for Wireless Edge Networks

The paper introduces GQ-FSL, a green quantized federated split learning framework designed for wireless edge networks. It uses stochastic quantization for both local training and wireless transmissions, allowing asymmetric precision between client and server submodels to balance device energy limits with global convergence. The authors develop energy models and a convergence bound for heterogeneous data, then formulate an optimization problem to set the DNN split point and precision levels, achieving lower energy consumption while meeting latency and accuracy targets.

By Idan Roth, Lutz Lampe
Hugging Face Trending Papers
Jul 22

Convergence-Latency-Aware Adaptive Modulation and Resource Allocation in RIS-Assisted Wireless Federated Learning

Federated learning (FL) over wireless networks suffers from significant training latency and degraded convergence due to unreliable wireless transmission, especially under blocked propagation environments. Although reconfigurable intelligent surfaces (RISs) can improve communication reliability, existing wireless FL studies rarely characterize the trade-off between learning convergence and communication delay under modulation-dependent transmission errors.

arXiv Machine Learning
Oct 28

Unified Robust and Convergent Over-The-Air Federated Learning under Channel Noise and Heterogeneity

The paper introduces RoCo-FL, a robust and convergent framework for Over-the-Air Federated Learning that addresses channel noise, fading, and data/system heterogeneity. RoCo-FL models local updates with inexact optimization and a proximal constraint to stabilize updates under wireless impairments and stragglers. The authors provide convergence guarantees for non-convex objectives and demonstrate improved stability and accuracy on CIFAR10, CIFAR100, and FEMNIST compared to existing baselines.

By Zubair Shaban, Nazreen Shah, Ranjitha Prasad