Track papers across labs, topics, and trends.
Search papers, labs, and topics across Lattice.
This paper explores a unified framework for over-the-air federated learning (FL) with energy-harvesting mobile devices, addressing challenges posed by heterogeneous data distributions and limited communication resources. By leveraging cluster information, the approach enhances global model representation and enables personalized model training, optimizing both energy use and communication efficiency. Numerical results indicate significant improvements in model fairness and personalization while reducing communication overhead compared to traditional FL methods.
Clustering energy-harvesting devices can simultaneously enhance global model accuracy and enable personalized learning, transforming federated learning dynamics.
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. In practical wireless networks, mobile devices (MDs) often exhibit diverse data and learning objectives, naturally forming clusters of users with jointly trainable models. When devices rely on energy harvesting (EH), stochastic energy arrivals further complicate participation and scheduling under communication constraints. In this work, we study over-the-air (OTA) FL with EH MDs under heterogeneous data distributions, and investigate two closely related learning objectives within a unified framework: one aiming for a more representative global model by reducing data bias, and the other learning more personalized cluster-specific models by exploiting this bias. In the global training mode, cluster information guides energy- and diversity-aware scheduling, ensuring that the scheduled active users provide a more representative aggregate update. In the personalization mode, the same cluster structure defines cluster-level learning objectives and OTA recovery targets, enabling the parameter server to train multiple cluster-specific models through simultaneous transmissions over the wireless multiple-access channel. Numerical results demonstrate that the proposed unified framework improves fairness or personalization, depending on the operating mode, while reducing communication overhead.