Federated Learning of Gboard Language Models with Differential Privacy
Not on the map. Why?
Abstract (arXiv)
We train language models (LMs) with federated learning (FL) and differential privacy (DP) in the Google Keyboard (Gboard). We apply the DP-Follow-the-Regularized-Leader (DP-FTRL)~citepkairouz21b algorithm to achieve meaningfully formal DP guarantees without requiring uniform sampling of client devices. To provide favorable privacy-utility trade-offs, we introduce a new client participation criterion and discuss the implication of its configuration in large scale systems. We show how quantile-based clip estimation~citepandrew2019differentially can be combined with DP-FTRL to adaptively choose the clip norm during training or reduce the hyperparameter tuning in preparation for training. With the help of pretraining on public data, we train and deploy more than twenty Gboard LMs that achieve high utility and ρ-zCDP privacy guarantees with ρ ∈ (0.2, 2), with two models additionally trained with secure aggregation~citepbonawitz2017practical. We are happy to announce that all the next word prediction neural network LMs in Gboard now have DP guarantees, and all future launches of Gboard neural network LMs will require DP guarantees. We summarize our experience and provide concrete suggestions on DP training for practitioners.
Key facts. Cited by 66 papers in the map (data through 14 Oct 2026). Builds on 28 of its 46 references that are in the map. First posted on arXiv 29 May 2023.
Builds on
28 of 46 references.
References · 28
- 2019Exploring the Limits of Transfer Learning with a Unified Text-to-Text Transformercited by 14,599
- 2016Communication-Efficient Learning of Deep Networks from Decentralized Datacited by 7,669
- 2019Advances and Open Problems in Federated Learningcited by 3,046
- 2016Deep Learning with Differential Privacycited by 2,728
- 2020mT5: A massively multilingual pre-trained text-to-text transformercited by 1,818
- 2018The Secret Sharer: Evaluating and Testing Unintended Memorization in Neural Networkscited by 864
- 2019Towards Federated Learning at Scale: System Designcited by 832
- 2018Federated Learning for Mobile Keyboard Predictioncited by 572
- 2017Differentially Private Federated Learning: A Client Level Perspectivecited by 459
- 2021Large Language Models Can Be Strong Differentially Private Learnerscited by 412
- 2021Differentially Private Fine-tuning of Language Modelscited by 400
- 2016Concentrated Differential Privacy: Simplifications, Extensions, and Lower Boundscited by 346
16 more
- 2021A Field Guide to Federated Optimizationcited by 315
- 2022Unlocking High-Accuracy Differentially Private Image Classification through Scalecited by 224
- 2019Differentially Private Learning with Adaptive Clippingcited by 220
- 2018Applied Federated Learning: Improving Google Keyboard Query Suggestionscited by 204
- 2021Practical and Private (Deep) Learning without Sampling or Shufflingcited by 170
- 2023How to DP-fy ML: A Practical Guide to Machine Learning with Differential Privacycited by 153
- 2021The Distributed Discrete Gaussian Mechanism for Federated Learning with Secure Aggregationcited by 128
- 2020Training Production Language Models without Memorizing User Datacited by 68
- 2021The Skellam Mechanism for Differentially Private Federated Learningcited by 68
- 2022Multi-Epoch Matrix Factorization Mechanisms for Private Machine Learningcited by 48
- 2020Privacy Amplification via Random Check-Inscited by 39
- 2022The Fundamental Price of Secure Aggregation in Differentially Private Federated Learningcited by 39
- 2017Compressing Word Embeddings via Deep Compositional Code Learningcited by 35
- 2023On the Convergence of Federated Averaging with Cyclic Client Participationcited by 31
- 2023Can Public Large Language Models Help Private Cross-device Federated Learning?cited by 30
- 2022Learning to Generate Image Embeddings with User-level Differential Privacycited by 23
Built on by
Citing papers per year: 2023: 16; 2024: 26; 2025: 17; 2026: 7.
66 papers.
Sort:
By area:
- 2025Multi-Agent Collaboration Mechanisms: A Survey of LLMscited by 431
- 2023(Amplified) Banded Matrix Factorization: A unified approach to private trainingcited by 49
- 2023User Inference Attacks on Large Language Modelscited by 36
- 2025Privacy in Action: Towards Realistic Privacy Mitigation and Evaluation for LLM-Powered Agentscited by 36
- 2024Teach LLMs to Phish: Stealing Private Information from Language Modelscited by 32
- 2024A Survey on Efficient Federated Learning Methods for Foundation Model Trainingcited by 29
- 2024PrE-Text: Training Language Models on Private Federated Data in the Age of LLMscited by 29
- 2024Fine-Tuning Large Language Models with User-Level Differential Privacycited by 29
- 2023How to Protect Copyright Data in Optimization of Large Language Models?cited by 26
- 2023Grounding Foundation Models through Federated Transfer Learning: A General Frameworkcited by 24
- 2023Correlated Noise Provably Beats Independent Noise for Differentially Private Learningcited by 23
- 2023Enabling On-Device Large Language Model Personalization with Self-Supervised Data Selection and Synthesiscited by 20
38 more
- 2024Differentially Private Knowledge Distillation via Synthetic Text Generationcited by 18
- 2024Prompt Public Large Language Models to Synthesize Data for Private On-device Applicationscited by 17
- 2024Efficient and Near-Optimal Noise Generation for Streaming Differential Privacycited by 17
- 2024Empirical Guidelines for Deploying LLMs onto Resource-constrained Edge Devicescited by 17
- 2025Synthesizing Privacy-Preserving Text Data via Finetuning without Finetuning Billion-Scale LLMscited by 16
- 2024Federated Learning in Practice: Reflections and Projectionscited by 15
- 2023Private Distribution Learning with Public Data: The View from Sample Compressioncited by 14
- 2024Synergizing Foundation Models and Federated Learning: A Surveycited by 14
- 2025POPri: Private Federated Learning using Preference-Optimized Synthetic Datacited by 14
- 2024The Federation Strikes Back: A Survey of Federated Learning Privacy Attacks, Defenses, Applications, and Policy Landscapecited by 13
- 2023Unmasking Transformers: A Theoretical Approach to Data Recovery via Attention Weightscited by 12
- 2024Private Gradient Descent for Linear Regression: Tighter Error Bounds and Instance-Specific Uncertainty Estimationcited by 12
- 2023Information Flow Control in Machine Learning through Modular Model Architecturecited by 11
- 2023FedTherapist: Mental Health Monitoring with User-Generated Linguistic Expressions on Smartphones via Federated Learningcited by 10
- 2024DP-DyLoRA: Fine-Tuning Transformer-Based Models On-Device under Differentially Private Federated Learning using Dynamic Low-Rank Adaptationcited by 9
- 2024A Hassle-free Algorithm for Private Learning in Practice: Don't Use Tree Aggregation, Use BLTscited by 9
- 2024pfl-research: simulation framework for accelerating research in Private Federated Learningcited by 8
- 2024Thinking Racial Bias in Fair Forgery Detection: Models, Datasets and Evaluationscited by 8
- 2024A Statistical Viewpoint on Differential Privacy: Hypothesis Testing, Representation and Blackwell's Theoremcited by 8
- 2023When Federated Learning meets Watermarking: A Comprehensive Overview of Techniques for Intellectual Property Protectioncited by 7
- 2024Byzantine-Robust Aggregation for Securing Decentralized Federated Learningcited by 7
- 2025Mitigating Unintended Memorization with LoRA in Federated Learning for LLMscited by 7
- 2023Samplable Anonymous Aggregation for Private Federated Data Analysiscited by 6
- 2023A Theoretical Insight into Attack and Defense of Gradient Leakage in Transformercited by 6
- 2024On the Convergence of DP-SGD with Adaptive Clippingcited by 6
- 2023Private Federated Learning with Autotuned Compressioncited by 5
- 2024Confidential Federated Computationscited by 4
- 2024Improved Communication-Privacy Trade-offs in L2 Mean Estimation under Streaming Differential Privacycited by 4
- 2024Momentum Approximation in Asynchronous Private Federated Learningcited by 3
- 2024Fingerprinting Codes Meet Geometry: Improved Lower Bounds for Private Query Release and Adaptive Data Analysiscited by 3
- 2025Empirical Privacy Variancecited by 3
- 2025ACTG-ARL: Differentially Private Conditional Text Generation with RL-Boosted Controlcited by 3
- 2024Efficient Language Model Architectures for Differentially Private Federated Learningcited by 2
- 2025Synthesizing and Adapting Error Correction Data for Mobile Large Language Model Applicationscited by 2
- 2025Constant Stepsize Local GD for Logistic Regression: Acceleration by Instabilitycited by 2
- 2026LLM Agents Enable User-Governed Personalization Beyond Platform Boundariescited by 2
- 2023Robust and Actively Secure Serverless Collaborative Learningcited by 1
- 2023Venn: Resource Management for Collaborative Learning Jobscited by 1
50 most cited of 66 All 66 (JSON)
- 2026Can R'enyi Accountants Beat Lossless Accountants for DP-SGD?12 Oct 2026
- 2026On-Device Commercial Intent Retrieval Under Size, Latency, and Privacy Constraints: A 3 MiB Retrieval System with Typed Egress Boundaries16 Sep 2026
- 2026Govern the Model, Not Only the Data: Storage, Circulation, and Learning in Creative AI3 Sep 2026
- 2026The Privacy-Hallucination Tradeoff in Differentially Private Language Models31 Aug 2026
- 2026LLM Agents Enable User-Governed Personalization Beyond Platform Boundaries10 May 2026
- 2026MAPLE: Metadata Augmented Private Language Evolution26 Feb 2026
- 2026SoK: Enhancing Cryptographic Collaborative Learning with Differential Privacy14 Jan 2026
- 2025Differential privacy from axioms26 Nov 2025
- 2025ACTG-ARL: Differentially Private Conditional Text Generation with RL-Boosted Control21 Oct 2025
- 2025Cocoon: A System Architecture for Differentially Private Training with Correlated Noises8 Oct 2025
- 2025Privacy in Action: Towards Realistic Privacy Mitigation and Evaluation for LLM-Powered Agents22 Sep 2025
- 2025Query-Efficient Locally Private Hypothesis Selection via the Scheffe Graph19 Sep 2025
38 more
- 2025Securing Private Federated Learning in a Malicious Setting: A Scalable TEE-Based Approach with Client Auditing10 Sep 2025
- 2025Constant Stepsize Local GD for Logistic Regression: Acceleration by Instability16 Jun 2025
- 2025Synthesizing and Adapting Error Correction Data for Mobile Large Language Model Applications24 May 2025
- 2025UnifyFL: Enabling Decentralized Cross-Silo Federated Learning26 Apr 2025
- 2025POPri: Private Federated Learning using Preference-Optimized Synthetic Data23 Apr 2025
- 2025Accelerating Differentially Private Federated Learning via Adaptive Extrapolation14 Apr 2025
- 2025Empirical Analysis of Privacy-Fairness-Accuracy Trade-offs in Federated Learning: A Step Towards Responsible AI20 Mar 2025
- 2025Synthesizing Privacy-Preserving Text Data via Finetuning without Finetuning Billion-Scale LLMs16 Mar 2025
- 2025Empirical Privacy Variance16 Mar 2025
- 2025Privacy-Preserving Dataset Combination9 Feb 2025
- 2025Mitigating Unintended Memorization with LoRA in Federated Learning for LLMs7 Feb 2025
- 2025Multi-Agent Collaboration Mechanisms: A Survey of LLMs10 Jan 2025
- 2024On the Convergence of DP-SGD with Adaptive Clipping27 Dec 2024
- 2024Fingerprinting Codes Meet Geometry: Improved Lower Bounds for Private Query Release and Adaptive Data Analysis18 Dec 2024
- 2024Federated Learning in Practice: Reflections and Projections11 Oct 2024
- 2024Debiasing Federated Learning with Correlated Client Participation2 Oct 2024
- 2024Byzantine-Robust Aggregation for Securing Decentralized Federated Learning26 Sep 2024
- 2024A Statistical Viewpoint on Differential Privacy: Hypothesis Testing, Representation and Blackwell's Theorem14 Sep 2024
- 2024A Hassle-free Algorithm for Private Learning in Practice: Don't Use Tree Aggregation, Use BLTs16 Aug 2024
- 2024Universally Harmonizing Differential Privacy Mechanisms for Federated Learning: Boosting Accuracy and Convergence20 Jul 2024
- 2024Thinking Racial Bias in Fair Forgery Detection: Models, Datasets and Evaluations19 Jul 2024
- 2024Fine-Tuning Large Language Models with User-Level Differential Privacy10 Jul 2024
- 2024Synergizing Foundation Models and Federated Learning: A Survey18 Jun 2024
- 2024Empirical Guidelines for Deploying LLMs onto Resource-constrained Edge Devices6 Jun 2024
- 2024PrE-Text: Training Language Models on Private Federated Data in the Age of LLMs5 Jun 2024
- 2024DP-DyLoRA: Fine-Tuning Transformer-Based Models On-Device under Differentially Private Federated Learning using Dynamic Low-Rank Adaptation10 May 2024
- 2024The Federation Strikes Back: A Survey of Federated Learning Privacy Attacks, Defenses, Applications, and Policy Landscape6 May 2024
- 2024Improved Communication-Privacy Trade-offs in L2 Mean Estimation under Streaming Differential Privacy2 May 2024
- 2024Efficient and Near-Optimal Noise Generation for Streaming Differential Privacy25 Apr 2024
- 2024Confidential Federated Computations16 Apr 2024
- 2024pfl-research: simulation framework for accelerating research in Private Federated Learning9 Apr 2024
- 2024Prompt Public Large Language Models to Synthesize Data for Private On-device Applications5 Apr 2024
- 2024Efficient Language Model Architectures for Differentially Private Federated Learning12 Mar 2024
- 2024Differentially Private Knowledge Distillation via Synthetic Text Generation1 Mar 2024
- 2024Teach LLMs to Phish: Stealing Private Information from Language Models1 Mar 2024
- 2024Private Gradient Descent for Linear Regression: Tighter Error Bounds and Instance-Specific Uncertainty Estimation21 Feb 2024
- 2024Momentum Approximation in Asynchronous Private Federated Learning14 Feb 2024
- 2024A Survey on Efficient Federated Learning Methods for Foundation Model Training9 Jan 2024
50 newest of 66 All 66 (JSON)
- 2023(Amplified) Banded Matrix Factorization: A unified approach to private trainingcited by 49
- 2023User Inference Attacks on Large Language Modelscited by 36
- 2024Teach LLMs to Phish: Stealing Private Information from Language Modelscited by 32
- 2024A Survey on Efficient Federated Learning Methods for Foundation Model Trainingcited by 29
- 2024PrE-Text: Training Language Models on Private Federated Data in the Age of LLMscited by 29
- 2024Fine-Tuning Large Language Models with User-Level Differential Privacycited by 29
- 2023How to Protect Copyright Data in Optimization of Large Language Models?cited by 26
- 2023Grounding Foundation Models through Federated Transfer Learning: A General Frameworkcited by 24
- 2023Correlated Noise Provably Beats Independent Noise for Differentially Private Learningcited by 23
- 2023Enabling On-Device Large Language Model Personalization with Self-Supervised Data Selection and Synthesiscited by 20
- 2024Prompt Public Large Language Models to Synthesize Data for Private On-device Applicationscited by 17
- 2024Efficient and Near-Optimal Noise Generation for Streaming Differential Privacycited by 17
12 most cited of 61 All 61 (JSON)
- 2026Can R'enyi Accountants Beat Lossless Accountants for DP-SGD?12 Oct 2026
- 2026On-Device Commercial Intent Retrieval Under Size, Latency, and Privacy Constraints: A 3 MiB Retrieval System with Typed Egress Boundaries16 Sep 2026
- 2026Govern the Model, Not Only the Data: Storage, Circulation, and Learning in Creative AI3 Sep 2026
- 2026The Privacy-Hallucination Tradeoff in Differentially Private Language Models31 Aug 2026
- 2026MAPLE: Metadata Augmented Private Language Evolution26 Feb 2026
- 2026SoK: Enhancing Cryptographic Collaborative Learning with Differential Privacy14 Jan 2026
- 2025Differential privacy from axioms26 Nov 2025
- 2025ACTG-ARL: Differentially Private Conditional Text Generation with RL-Boosted Control21 Oct 2025
- 2025Cocoon: A System Architecture for Differentially Private Training with Correlated Noises8 Oct 2025
- 2025Query-Efficient Locally Private Hypothesis Selection via the Scheffe Graph19 Sep 2025
- 2025Securing Private Federated Learning in a Malicious Setting: A Scalable TEE-Based Approach with Client Auditing10 Sep 2025
- 2025Constant Stepsize Local GD for Logistic Regression: Acceleration by Instability16 Jun 2025
12 newest of 61 All 61 (JSON)