Architectural Blueprint For Heterogeneity-Resilient Federated Learning
Satwat Bashir, Tasos Dagiuklas, Kasra Kassai, Muddesar Iqbal
TL;DR
This paper tackles the challenge of data and compute heterogeneity in Federated Learning for edge environments. It introduces a novel three-tier architecture consisting of a Client Layer, Edge Layer, and Fedge Layer, enabling multi-global models and dual-level aggregation to manage non-IID data efficiently while preserving privacy. The approach demonstrates improved model accuracy, reduced communication overhead, and scalability on MNIST under non-IID scenarios, compared to standard FedAvg. The work offers a practical blueprint for deploying heterogeneous FL at the edge, with implications for privacy-preserving distributed learning and broader adoption of FL technologies.
Abstract
This paper proposes a novel three tier architecture for federated learning to optimize edge computing environments. The proposed architecture addresses the challenges associated with client data heterogeneity and computational constraints. It introduces a scalable, privacy preserving framework that enhances the efficiency of distributed machine learning. Through experimentation, the paper demonstrates the architecture capability to manage non IID data sets more effectively than traditional federated learning models. Additionally, the paper highlights the potential of this innovative approach to significantly improve model accuracy, reduce communication overhead, and facilitate broader adoption of federated learning technologies.
