Training Large-Vocabulary Neural Language Models by Private Federated Learning for Resource-Constrained Devices
Not on the map. Why?
Abstract (arXiv)
Federated Learning (FL) is a technique to train models using data distributed across devices. Differential Privacy (DP) provides a formal privacy guarantee for sensitive data. Our goal is to train a large neural network language model (NNLM) on compute-constrained devices while preserving privacy using FL and DP. However, the DP-noise introduced to the model increases as the model size grows, which often prevents convergence. We propose Partial Embedding Updates (PEU), a novel technique to decrease noise by decreasing payload size. Furthermore, we adopt Low Rank Adaptation (LoRA) and Noise Contrastive Estimation (NCE) to reduce the memory demands of large models on compute-constrained devices. This combination of techniques makes it possible to train large-vocabulary language models while preserving accuracy and privacy.
Key facts. Cited by 16 papers in the map (data through 7 Oct 2026). Builds on 26 of its 40 references that are in the map. First posted on arXiv 18 Jul 2022.
Builds on
26 of 40 references.
References · 26
- 2017Attention Is All You Needcited by 59,932
- 2018BERT: Pre-training of Deep Bidirectional Transformers for Language Understandingcited by 40,204
- 2020Language Models are Few-Shot Learnerscited by 32,199
- 2021LoRA: Low-Rank Adaptation of Large Language Modelscited by 14,084
- 2016Communication-Efficient Learning of Deep Networks from Decentralized Datacited by 7,592
- 2016Deep Learning with Differential Privacycited by 2,682
- 2020Extracting Training Data from Large Language Modelscited by 2,144
- 2019Megatron-LM: Training Multi-Billion Parameter Language Models Using Model Parallelismcited by 1,829
- 2020Adaptive Federated Optimizationcited by 1,036
- 2019Deep Leakage from Gradientscited by 1,004
- 2018The Secret Sharer: Evaluating and Testing Unintended Memorization in Neural Networkscited by 847
- 2019Measuring the Effects of Non-Identical Data Distribution for Federated Visual Classificationcited by 836
14 more
- 2017Renyi Differential Privacycited by 643
- 2018Federated Learning for Mobile Keyboard Predictioncited by 565
- 2018Exploiting Unintended Feature Leakage in Collaborative Learningcited by 509
- 2014New insights and perspectives on the natural gradient methodcited by 435
- 2021Large Language Models Can Be Strong Differentially Private Learnerscited by 405
- 2021Differentially Private Fine-tuning of Language Modelscited by 392
- 2016Using the Output Embedding to Improve Language Modelscited by 266
- 2019R'enyi Differential Privacy of the Sampled Gaussian Mechanismcited by 199
- 2021Large-Scale Differentially Private BERTcited by 117
- 2019Federated Learning for Emoji Prediction in a Mobile Keyboardcited by 115
- 2014Differentially Private Empirical Risk Minimization: Efficient Algorithms and Tight Error Boundscited by 104
- 2020Training Production Language Models without Memorizing User Datacited by 68
- 2019SNDCNN: Self-normalizing deep CNNs with scaled exponential linear units for speech recognitioncited by 13
- 2021A Discriminative Entity-Aware Language Model for Virtual Assistantscited by 7
Built on by
Citing papers per year: 2022: 2; 2023: 5; 2024: 8; 2025: 1.
16 papers.
Sort:
- 2024Federated Fine-tuning of Large Language Models under Heterogeneous Tasks and Client Resourcescited by 93
- 2022When Federated Learning Meets Pre-trained Language Models' Parameter-Efficient Tuning Methodscited by 74
- 2023Identifying and Mitigating Privacy Risks Stemming from Language Models: A Surveycited by 30
- 2024The Future of Large Language Model Pre-training is Federatedcited by 30
- 2024PrE-Text: Training Language Models on Private Federated Data in the Age of LLMscited by 29
- 2023Integration of Large Language Models and Federated Learningcited by 28
- 2023FedBPT: Efficient Federated Black-box Prompt Tuning for Large Language Modelscited by 25
- 2022CANIFE: Crafting Canaries for Empirical Privacy Measurement in Federated Learningcited by 24
- 2024Deconstructing The Ethics of Large Language Models from Long-standing Issues to New-emerging Dilemmas: A Surveycited by 23
- 2024DP-DyLoRA: Fine-Tuning Transformer-Based Models On-Device under Differentially Private Federated Learning using Dynamic Low-Rank Adaptationcited by 9
- 2024pfl-research: simulation framework for accelerating research in Private Federated Learningcited by 8
- 2023Modeling Spoken Information Queries for Virtual Assistants: Open Problems, Challenges and Opportunitiescited by 4
4 more
- 2024Momentum Approximation in Asynchronous Private Federated Learningcited by 3
- 2023Jointly Reparametrized Multi-Layer Adaptation for Efficient and Private Tuningcited by 2
- 2024Enhancing CTC-based speech recognition with diverse modeling unitscited by 1
- 2025Private Federated Multiclass Post-hoc Calibrationcited by 0
- 2025Private Federated Multiclass Post-hoc Calibration2 Oct 2025
- 2024Deconstructing The Ethics of Large Language Models from Long-standing Issues to New-emerging Dilemmas: A Survey8 Jun 2024
- 2024Enhancing CTC-based speech recognition with diverse modeling units5 Jun 2024
- 2024PrE-Text: Training Language Models on Private Federated Data in the Age of LLMs5 Jun 2024
- 2024The Future of Large Language Model Pre-training is Federated17 May 2024
- 2024DP-DyLoRA: Fine-Tuning Transformer-Based Models On-Device under Differentially Private Federated Learning using Dynamic Low-Rank Adaptation10 May 2024
- 2024pfl-research: simulation framework for accelerating research in Private Federated Learning9 Apr 2024
- 2024Federated Fine-tuning of Large Language Models under Heterogeneous Tasks and Client Resources18 Feb 2024
- 2024Momentum Approximation in Asynchronous Private Federated Learning14 Feb 2024
- 2023FedBPT: Efficient Federated Black-box Prompt Tuning for Large Language Models2 Oct 2023
- 2023Identifying and Mitigating Privacy Risks Stemming from Language Models: A Survey27 Sep 2023
- 2023Integration of Large Language Models and Federated Learning18 Jul 2023
4 more
- 2023Jointly Reparametrized Multi-Layer Adaptation for Efficient and Private Tuning30 May 2023
- 2023Modeling Spoken Information Queries for Virtual Assistants: Open Problems, Challenges and Opportunities25 Apr 2023
- 2022When Federated Learning Meets Pre-trained Language Models' Parameter-Efficient Tuning Methods20 Dec 2022
- 2022CANIFE: Crafting Canaries for Empirical Privacy Measurement in Federated Learning6 Oct 2022
- 2024Federated Fine-tuning of Large Language Models under Heterogeneous Tasks and Client Resourcescited by 93
- 2022When Federated Learning Meets Pre-trained Language Models' Parameter-Efficient Tuning Methodscited by 74
- 2023Identifying and Mitigating Privacy Risks Stemming from Language Models: A Surveycited by 30
- 2024The Future of Large Language Model Pre-training is Federatedcited by 30
- 2024PrE-Text: Training Language Models on Private Federated Data in the Age of LLMscited by 29
- 2023Integration of Large Language Models and Federated Learningcited by 28
- 2023FedBPT: Efficient Federated Black-box Prompt Tuning for Large Language Modelscited by 25
- 2022CANIFE: Crafting Canaries for Empirical Privacy Measurement in Federated Learningcited by 24
- 2024Deconstructing The Ethics of Large Language Models from Long-standing Issues to New-emerging Dilemmas: A Surveycited by 23
- 2024DP-DyLoRA: Fine-Tuning Transformer-Based Models On-Device under Differentially Private Federated Learning using Dynamic Low-Rank Adaptationcited by 9
- 2024pfl-research: simulation framework for accelerating research in Private Federated Learningcited by 8
- 2023Modeling Spoken Information Queries for Virtual Assistants: Open Problems, Challenges and Opportunitiescited by 4
12 most cited of 16 All 16 (JSON)
- 2025Private Federated Multiclass Post-hoc Calibration2 Oct 2025
- 2024Deconstructing The Ethics of Large Language Models from Long-standing Issues to New-emerging Dilemmas: A Survey8 Jun 2024
- 2024Enhancing CTC-based speech recognition with diverse modeling units5 Jun 2024
- 2024PrE-Text: Training Language Models on Private Federated Data in the Age of LLMs5 Jun 2024
- 2024The Future of Large Language Model Pre-training is Federated17 May 2024
- 2024DP-DyLoRA: Fine-Tuning Transformer-Based Models On-Device under Differentially Private Federated Learning using Dynamic Low-Rank Adaptation10 May 2024
- 2024pfl-research: simulation framework for accelerating research in Private Federated Learning9 Apr 2024
- 2024Federated Fine-tuning of Large Language Models under Heterogeneous Tasks and Client Resources18 Feb 2024
- 2024Momentum Approximation in Asynchronous Private Federated Learning14 Feb 2024
- 2023FedBPT: Efficient Federated Black-box Prompt Tuning for Large Language Models2 Oct 2023
- 2023Identifying and Mitigating Privacy Risks Stemming from Language Models: A Survey27 Sep 2023
- 2023Integration of Large Language Models and Federated Learning18 Jul 2023
12 newest of 16 All 16 (JSON)