Selective Pre-training for Private Fine-tuning
Not on the map. Why?
Abstract (arXiv)
Text prediction models, when used in applications like email clients or word processors, must protect user data privacy and adhere to model size constraints. These constraints are crucial to meet memory and inference time requirements, as well as to reduce inference costs. Building small, fast, and private domain-specific language models is a thriving area of research. In this work, we show that a careful pre-training on a subset of the public dataset that is guided by the private dataset is crucial to train small language models with differential privacy. On standard benchmarks, small models trained with our new framework achieve state-of-the-art performance. In addition to performance improvements, our results demonstrate that smaller models, through careful pre-training and private fine-tuning, can match the performance of much larger models that do not have access to private data. This underscores the potential of private learning for model compression and enhanced efficiency.
Key facts. Cited by 23 papers in the map (data through 14 Oct 2026). Builds on 74 of its 101 references that are in the map. First posted on arXiv 23 May 2023.
Builds on
74 of 101 references.
References · 74
- 2018BERT: Pre-training of Deep Bidirectional Transformers for Language Understandingcited by 40,399
- 2014Adam: A Method for Stochastic Optimizationcited by 37,240
- 2020Language Models are Few-Shot Learnerscited by 32,419
- 2021LoRA: Low-Rank Adaptation of Large Language Modelscited by 14,380
- 2015Distilling the Knowledge in a Neural Networkcited by 10,324
- 2020Scaling Laws for Neural Language Modelscited by 6,586
- 2022PaLM: Scaling Language Modeling with Pathwayscited by 5,326
- 2017On Calibration of Modern Neural Networkscited by 4,960
- 2018GLUE: A Multi-Task Benchmark and Analysis Platform for Natural Language Understandingcited by 4,899
- 2019DistilBERT, a distilled version of BERT: smaller, faster, cheaper and lightercited by 4,307
- 2016Deep Learning with Differential Privacycited by 2,728
- 2016Membership Inference Attacks against Machine Learning Modelscited by 2,281
38 more
- 2020Extracting Training Data from Large Language Modelscited by 2,189
- 2015Learning both Weights and Connections for Efficient Neural Networkscited by 2,031
- 2020Don't Stop Pretraining: Adapt Language Models to Domains and Taskscited by 1,565
- 2022Quantifying Memorization Across Neural Language Modelscited by 763
- 2021Deduplicating Training Data Makes Language Models Bettercited by 714
- 2022Beyond neural scaling laws: beating power law scaling via data pruningcited by 528
- 2021Large Language Models Can Be Strong Differentially Private Learnerscited by 412
- 2021Differentially Private Fine-tuning of Language Modelscited by 400
- 2020Label-Only Membership Inference Attackscited by 388
- 2016Semi-supervised Knowledge Transfer for Deep Learning from Private Training Datacited by 348
- 2019Selection via Proxy: Efficient Data Selection for Deep Learningcited by 339
- 2023Data Selection for Language Models via Importance Resamplingcited by 329
- 2023SemDeDup: Data-efficient learning at web-scale through semantic deduplicationcited by 303
- 2018Scalable Private Learning with PATEcited by 269
- 2020Differentially Private Learning Needs Better Features (or Much More Data)cited by 235
- 2022Unlocking High-Accuracy Differentially Private Image Classification through Scalecited by 224
- 2021Numerical Composition of Differential Privacycited by 180
- 2022Structured Pruning Learns Compact and Accurate Modelscited by 170
- 2021Counterfactual Memorization in Neural Language Modelscited by 156
- 2022Scaling Laws and Interpretability of Learning from Repeated Datacited by 152
- 2021Large-Scale Differentially Private BERTcited by 118
- 2014Differentially Private Empirical Risk Minimization: Efficient Algorithms and Tight Error Boundscited by 107
- 2021Hyperparameter Tuning with Renyi Differential Privacycited by 104
- 2020Bypassing the Ambient Dimension: Private SGD with Gradient Subspace Identificationcited by 90
- 2021Large Scale Private Learning via Low-rank Reparametrizationcited by 88
- 2022Position: Considerations for Differentially Private Learning with Large-Scale Public Pretrainingcited by 88
- 2022Toward Training at ImageNet Scale with Differential Privacycited by 87
- 2023Membership Inference Attacks against Diffusion Modelscited by 86
- 2018Private Selection from Private Candidatescited by 75
- 2020Enabling Fast Differentially Private SGD via Just-in-Time Compilation and Vectorizationcited by 58
- 2022Mixed Differential Privacy in Computer Visioncited by 54
- 2019Transfer Adaptation Learning: A Decade Surveycited by 52
- 2022TAN Without a Burn: Scaling Laws of DP-SGDcited by 52
- 2022Exploring the Limits of Differentially Private Deep Learning with Group-wise Clippingcited by 52
- 2021Public Data-Assisted Mirror Descent for Private Model Trainingcited by 49
- 2022Scalable and Efficient Training of Large Convolutional Neural Networks with Differential Privacycited by 48
- 2022When Does Differentially Private Learning Not Suffer in High Dimensions?cited by 48
- 2022Differentially Private Bias-Term Fine-tuning of Foundation Modelscited by 48
50 most cited of 74 All 74 (JSON)
Built on by
Citing papers per year: 2023: 4; 2024: 10; 2025: 6; 2026: 3.
23 papers.
Sort:
By area:
- 2023Textbooks Are All You Needcited by 569
- 2023Privacy in Large Language Models: Attacks, Defenses and Future Directionscited by 51
- 2024LLM-PBE: Assessing Data Privacy in Large Language Modelscited by 41
- 2024What is the Role of Small Models in the LLM Era: A Surveycited by 41
- 2024PrivacyRestore: Privacy-Preserving Inference in Large Language Models via Privacy Removal and Restorationcited by 28
- 2023Grounding Foundation Models through Federated Transfer Learning: A General Frameworkcited by 24
- 2025Ten Challenging Problems in Federated Foundation Modelscited by 20
- 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
- 2025Differentially Private Synthetic Data via APIs 3: Using Simulators Instead of Foundation Modelcited by 13
- 2025DPImageBench: A Unified Benchmark for Differentially Private Image Synthesiscited by 11
- 2024Efficiently Computing Similarities to Private Datasetscited by 9
11 more
- 2024Towards More Realistic Extraction Attacks: An Adversarial Perspectivecited by 8
- 2023Optimal Locally Private Nonparametric Classification with Public Datacited by 7
- 2025AugFL: Augmenting Federated Learning with Pretrained Modelscited by 6
- 2024Small Language Models for Application Interactions: A Case Studycited by 5
- 2024Differentially Private Kernel Density Estimationcited by 3
- 2024Federated Domain-Specific Knowledge Transfer on Large Language Models Using Synthetic Datacited by 2
- 2025Public Data Assisted Differentially Private In-Context Learningcited by 1
- 2025Towards Reliable and Generalizable Differentially Private Machine Learning (Extended Version)cited by 0
- 2026Curation Leaks: Membership Inference Attacks against Data Curation for Machine Learningcited by 0
- 2026Privacy-Preserving RAG via Multi-Agent Semantic Rewriting: Achieving Confidentiality Without Compromising Contextual Fidelitycited by 0
- 2026Differentially Private Mixing of Public Datasets Improves Private Learningcited by 0
- 2026Differentially Private Mixing of Public Datasets Improves Private Learning5 Oct 2026
- 2026Privacy-Preserving RAG via Multi-Agent Semantic Rewriting: Achieving Confidentiality Without Compromising Contextual Fidelity23 Jun 2026
- 2026Curation Leaks: Membership Inference Attacks against Data Curation for Machine Learning28 Feb 2026
- 2025Public Data Assisted Differentially Private In-Context Learning13 Sep 2025
- 2025Towards Reliable and Generalizable Differentially Private Machine Learning (Extended Version)21 Aug 2025
- 2025DPImageBench: A Unified Benchmark for Differentially Private Image Synthesis18 Mar 2025
- 2025AugFL: Augmenting Federated Learning with Pretrained Models4 Mar 2025
- 2025Ten Challenging Problems in Federated Foundation Models14 Feb 2025
- 2025Differentially Private Synthetic Data via APIs 3: Using Simulators Instead of Foundation Model8 Feb 2025
- 2024What is the Role of Small Models in the LLM Era: A Survey10 Sep 2024
- 2024Differentially Private Kernel Density Estimation3 Sep 2024
- 2024LLM-PBE: Assessing Data Privacy in Large Language Models23 Aug 2024
11 more
- 2024Towards More Realistic Extraction Attacks: An Adversarial Perspective2 Jul 2024
- 2024PrivacyRestore: Privacy-Preserving Inference in Large Language Models via Privacy Removal and Restoration3 Jun 2024
- 2024Small Language Models for Application Interactions: A Case Study23 May 2024
- 2024Federated Domain-Specific Knowledge Transfer on Large Language Models Using Synthetic Data23 May 2024
- 2024Prompt Public Large Language Models to Synthesize Data for Private On-device Applications5 Apr 2024
- 2024Efficiently Computing Similarities to Private Datasets13 Mar 2024
- 2024Differentially Private Knowledge Distillation via Synthetic Text Generation1 Mar 2024
- 2023Grounding Foundation Models through Federated Transfer Learning: A General Framework29 Nov 2023
- 2023Optimal Locally Private Nonparametric Classification with Public Data19 Nov 2023
- 2023Privacy in Large Language Models: Attacks, Defenses and Future Directions16 Oct 2023
- 2023Textbooks Are All You Need20 Jun 2023
- 2023Textbooks Are All You Needcited by 569
- 2023Privacy in Large Language Models: Attacks, Defenses and Future Directionscited by 51
- 2024LLM-PBE: Assessing Data Privacy in Large Language Modelscited by 41
- 2024What is the Role of Small Models in the LLM Era: A Surveycited by 41
- 2024PrivacyRestore: Privacy-Preserving Inference in Large Language Models via Privacy Removal and Restorationcited by 28
- 2023Grounding Foundation Models through Federated Transfer Learning: A General Frameworkcited by 24
- 2024Prompt Public Large Language Models to Synthesize Data for Private On-device Applicationscited by 17
- 2025DPImageBench: A Unified Benchmark for Differentially Private Image Synthesiscited by 11
- 2024Efficiently Computing Similarities to Private Datasetscited by 9
- 2024Towards More Realistic Extraction Attacks: An Adversarial Perspectivecited by 8
- 2023Optimal Locally Private Nonparametric Classification with Public Datacited by 7
- 2025AugFL: Augmenting Federated Learning with Pretrained Modelscited by 6
12 most cited of 17 All 17 (JSON)
- 2026Differentially Private Mixing of Public Datasets Improves Private Learning5 Oct 2026
- 2025Public Data Assisted Differentially Private In-Context Learning13 Sep 2025
- 2025Towards Reliable and Generalizable Differentially Private Machine Learning (Extended Version)21 Aug 2025
- 2025DPImageBench: A Unified Benchmark for Differentially Private Image Synthesis18 Mar 2025
- 2025AugFL: Augmenting Federated Learning with Pretrained Models4 Mar 2025
- 2024What is the Role of Small Models in the LLM Era: A Survey10 Sep 2024
- 2024Differentially Private Kernel Density Estimation3 Sep 2024
- 2024LLM-PBE: Assessing Data Privacy in Large Language Models23 Aug 2024
- 2024Towards More Realistic Extraction Attacks: An Adversarial Perspective2 Jul 2024
- 2024PrivacyRestore: Privacy-Preserving Inference in Large Language Models via Privacy Removal and Restoration3 Jun 2024
- 2024Small Language Models for Application Interactions: A Case Study23 May 2024
- 2024Prompt Public Large Language Models to Synthesize Data for Private On-device Applications5 Apr 2024
12 newest of 17 All 17 (JSON)