English

Towards Cooperative Federated Learning over Heterogeneous Edge/Fog Networks

Distributed, Parallel, and Cluster Computing 2023-03-16 v1 Machine Learning Networking and Internet Architecture Systems and Control Systems and Control

Abstract

Federated learning (FL) has been promoted as a popular technique for training machine learning (ML) models over edge/fog networks. Traditional implementations of FL have largely neglected the potential for inter-network cooperation, treating edge/fog devices and other infrastructure participating in ML as separate processing elements. Consequently, FL has been vulnerable to several dimensions of network heterogeneity, such as varying computation capabilities, communication resources, data qualities, and privacy demands. We advocate for cooperative federated learning (CFL), a cooperative edge/fog ML paradigm built on device-to-device (D2D) and device-to-server (D2S) interactions. Through D2D and D2S cooperation, CFL counteracts network heterogeneity in edge/fog networks through enabling a model/data/resource pooling mechanism, which will yield substantial improvements in ML model training quality and network resource consumption. We propose a set of core methodologies that form the foundation of D2D and D2S cooperation and present preliminary experiments that demonstrate their benefits. We also discuss new FL functionalities enabled by this cooperative framework such as the integration of unlabeled data and heterogeneous device privacy into ML model training. Finally, we describe some open research directions at the intersection of cooperative edge/fog and FL.

Keywords

Cite

@article{arxiv.2303.08361,
  title  = {Towards Cooperative Federated Learning over Heterogeneous Edge/Fog Networks},
  author = {Su Wang and Seyyedali Hosseinalipour and Vaneet Aggarwal and Christopher G. Brinton and David J. Love and Weifeng Su and Mung Chiang},
  journal= {arXiv preprint arXiv:2303.08361},
  year   = {2023}
}

Comments

This paper has been accepted for publication in IEEE Communications Magazine