arXiv Machine Learning
Oct 3

StoCFL: A Stochastically Clustered Federated Learning Framework for Non-IID Data with Dynamic Client Participation

StoCFL is a clustered federated learning framework designed to address Non-IID data and dynamic client participation. It introduces a flexible clustering mechanism that allows arbitrary client participation and accommodates newly joined clients, improving data efficiency and model performance. Experiments on four Non-IID settings and a real-world dataset demonstrate that StoCFL achieves promising cluster results even when the number of clusters is unknown, outperforming baseline approaches across various scenarios.

By Dun Zeng, Xiangjing Hu, Shiyu Liu, Yue Yu, Qifan Wang, Zenglin Xu
arXiv Machine Learning
Aug 4

Cluster-Aware Over-the-Air Federated Learning with Energy-Harvesting Devices: From Global Training to Model Personalization

arXiv:2608. 01426v1 Announce Type: new Abstract: Federated learning (FL) enables distributed optimization and learning across decentralized edge devices while preserving data privacy, but its performance is fundamentally constrained by heterogeneous data distributions, limited communication resources, and energy availability.

By Furkan Bagci, Busra Tegin, Mohammad Kazemi, Tolga M. Duman
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