Federated Learning
Description
Federated learning trains one shared model across several parties, such as phones or hospitals, without collecting their data in one place. Each party trains the current model on its own data and sends back only an update, such as the changed weights (the model's internal numbers) or the gradients (the direction of change). A central coordinator combines the updates, typically by averaging, into an improved shared model and sends it out again, and the cycle repeats over many rounds. Raw records stay with their owners, which limits how much sensitive data is stored or moved and can support data protection rules that favour collecting less. It is not a privacy guarantee on its own, because updates can still reveal information about the data behind them and the coordinator has to be trusted.
Example Use Cases
Privacy
A mobile keyboard team trains a next-word prediction model on users' phones and receives only model updates, so typed text never reaches its servers. The team uses this to decide whether personalised suggestions can ship without collecting message content centrally.
Hospitals in several countries train a model that predicts COVID-19 outcomes on their own patient records and share only model updates with a coordinating centre, so no patient record leaves a hospital. Each hospital's data governance board uses this to decide whether it can join the collaboration.
Several banks jointly train a fraud-detection model on their own transaction data and share only model updates, so no bank's customer records reach a competitor. Each bank's compliance team uses the setup to decide whether joint training fits its data-sharing restrictions.
Limitations
- Communication overhead can be substantial, especially with frequent model updates and large models, potentially limiting scalability and increasing training time compared to centralised approaches.
- Statistical heterogeneity across participants (non-IID data distributions) can lead to training instability, slower convergence, and reduced model performance compared to centralised training on pooled data.
- System heterogeneity in computational capabilities, network connectivity, and availability of participating devices can create bottlenecks and introduce bias towards more capable participants.
- Privacy vulnerabilities remain through gradient leakage attacks, model inversion, and membership inference attacks that can potentially reconstruct sensitive information from shared model updates.
- A faulty or malicious participant can submit harmful updates that the coordinator cannot check against the participant's data, for example to plant a hidden backdoor in the shared model. Defences such as robust aggregation add cost, and the coordinator must also be trusted with the updates it receives.
Resources
Research Papers
Communication-Efficient Learning of Deep Networks from Decentralized Data
Modern mobile devices have access to a wealth of data suitable for learning models, which in turn can greatly improve the user experience on the device. For example, language models can improve speech recognition and text entry, and image models can automatically select good photos. However, this rich data is often privacy sensitive, large in quantity, or both, which may preclude logging to the data center and training there using conventional approaches. We advocate an alternative that leaves the training data distributed on the mobile devices, and learns a shared model by aggregating locally-computed updates. We term this decentralized approach Federated Learning. We present a practical method for the federated learning of deep networks based on iterative model averaging, and conduct an extensive empirical evaluation, considering five different model architectures and four datasets. These experiments demonstrate the approach is robust to the unbalanced and non-IID data distributions that are a defining characteristic of this setting. Communication costs are the principal constraint, and we show a reduction in required communication rounds by 10-100x as compared to synchronized stochastic gradient descent.