Resource-Efficient Federated Learning

Ahmed M. Abdelmoniem, Atal Narayan Sahu, Marco Canini, Suhaib A. Fahmy

Keywords: Federated Learning, Heterogeneity, Resource Efficiency

Abstract: Federated Learning (FL) is a distributed training paradigm that avoids sharing the users’ private data. FL has presented unique challenges in dealing with data, device and user heterogeneity which impact both model quality and training time. The impact is exacerbated by the scale of the deployments. More importantly, existing FL methods result in inefficient use of resources and prolonged training times. In this work, we propose, REFL, to systematically address the question of resource efficiency in FL, showing the benefits of intelligent participant selection, and incorporation of updates from straggling participants. REFL is a resource-efficient federated learning system that maximizes FL systems’ resource efficiency without compromising statistical and system efficiency. REFL is released as open source at https://github.com/ahmedcs/REFL.

Submission Number: 13