Federated Learning Framework with Straggling Mitigation and Privacy-Awareness for AI-based Mobile Application Services

Abstract

This work proposes a novel framework to address straggling and privacy issues for federated learning (FL)-based mobile application services, considering limited computing/communications resources at mobile users (MUs)/mobile application provider (MAP), privacy cost, the rationality and incentive competition among MUs in contributing data to the MAP. Particularly, the MAP first determines a set of the best MUs for the FL process based on MUs' provided information/features. Then, each selected MU can encrypt part of local data and upload the encrypted data to the MAP for an encrypted training process, in addition to the local training process. For that, the selected MU can propose a contract to the MAP according to its expected local and encrypted data. To find optimal contracts that can maximize utilities while maintaining high learning quality of the system, we develop a multi-principal one-agent contract-based problem considering the MUs' privacy cost, the MAP's limited computing resources, and asymmetric information between the MAP and MUs. Experiments with a real-world dataset show that our framework can speed up training time up to 49% and improve prediction accuracy up to 4.6 times while enhancing network's social welfare up to 114% under the privacy cost consideration compared with those of baseline methods.