Amplifying Training Data Exposure through Fine-Tuning with Pseudo-Labeled Memberships
On the map: LLM safety and security → Membership inference attacks
Abstract (arXiv)
Neural language models (LMs) are vulnerable to training data extraction attacks due to data memorization. This paper introduces a novel attack scenario wherein an attacker adversarially fine-tunes pre-trained LMs to amplify the exposure of the original training data. This strategy differs from prior studies by aiming to intensify the LM's retention of its pre-training dataset. To achieve this, the attacker needs to collect generated texts that are closely aligned with the pre-training data. However, without knowledge of the actual dataset, quantifying the amount of pre-training data within generated texts is challenging. To address this, we propose the use of pseudo-labels for these generated texts, leveraging membership approximations indicated by machine-generated probabilities from the target LM. We subsequently fine-tune the LM to favor generations with higher likelihoods of originating from the pre-training data, based on their membership probabilities. Our empirical findings indicate a remarkable outcome: LMs with over 1B parameters exhibit a four to eight-fold increase in training data exposure. We discuss potential mitigations and suggest future research directions.
Key facts. No citing papers in the map yet (data through 7 Oct 2026). Placed in Membership inference attacks, one of 12 directions in LLM safety and security. Builds on 65 of its 91 references that are in the map. First posted on arXiv 19 Feb 2024.
Builds on
65 of 91 references.
Closest ancestors · 6
Same direction, newest first
- 2023Membership Inference Attacks against Language Models via Neighbourhood Comparison29 May
- 2022RelaxLoss: Defending Membership Inference Attacks without Losing Utility12 Jul
- 2022Quantifying Privacy Risks of Masked Language Models Using Membership Inference Attacks8 Mar
- 2021Membership Inference Attacks From First Principles7 Dec
- 2021Enhanced Membership Inference Attacks against Machine Learning Models18 Nov
- 2021Mitigating Membership Inference Attacks by Self-Distillation Through a Novel Ensemble Architecture15 Oct
Other references · 59
- 2018BERT: Pre-training of Deep Bidirectional Transformers for Language Understandingcited by 40,204
- 2020Language Models are Few-Shot Learnerscited by 32,199
- 2019PyTorch: An Imperative Style, High-Performance Deep Learning Librarycited by 20,160
- 2023GPT-4 Technical Reportcited by 18,132
- 2022Training language models to follow instructions with human feedbackcited by 15,712
- 2019Exploring the Limits of Transfer Learning with a Unified Text-to-Text Transformercited by 14,491
- 2017Proximal Policy Optimization Algorithmscited by 14,441
- 2021LoRA: Low-Rank Adaptation of Large Language Modelscited by 14,084
- 2023LLaMA: Open and Efficient Foundation Language Modelscited by 13,958
- 2019RoBERTa: A Robustly Optimized BERT Pretraining Approachcited by 13,266
- 2023Llama 2: Open Foundation and Fine-Tuned Chat Modelscited by 12,865
- 2023Direct Preference Optimization: Your Language Model is Secretly a Reward Modelcited by 7,916
38 more
- 2016SGDR: Stochastic Gradient Descent with Warm Restartscited by 4,993
- 2017Deep reinforcement learning from human preferencescited by 3,846
- 2022OPT: Open Pre-trained Transformer Language Modelscited by 3,464
- 2023Universal and Transferable Adversarial Attacks on Aligned Language Modelscited by 2,914
- 2019The Curious Case of Neural Text Degenerationcited by 2,881
- 2016Deep Learning with Differential Privacycited by 2,682
- 2020Learning to summarize from human feedbackcited by 2,562
- 2020The Pile: An 800GB Dataset of Diverse Text for Language Modelingcited by 2,346
- 2016Membership Inference Attacks against Machine Learning Modelscited by 2,221
- 2020Extracting Training Data from Large Language Modelscited by 2,144
- 2023Code Llama: Open Foundation Models for Codecited by 2,105
- 2022BLOOM: A 176B-Parameter Open-Access Multilingual Language Modelcited by 1,968
- 2020mT5: A massively multilingual pre-trained text-to-text transformercited by 1,813
- 2019PubMedQA: A Dataset for Biomedical Research Question Answeringcited by 1,271
- 2023PaLM 2 Technical Reportcited by 1,154
- 2015Aligning Books and Movies: Towards Story-like Visual Explanations by Watching Movies and Reading Bookscited by 1,041
- 2022Red Teaming Language Models with Language Modelscited by 983
- 2022Quantifying Memorization Across Neural Language Modelscited by 755
- 2021Deduplicating Training Data Makes Language Models Bettercited by 705
- 2023DetectGPT: Zero-Shot Machine-Generated Text Detection using Probability Curvaturecited by 705
- 2019Defending Against Neural Fake Newscited by 553
- 2019Pseudo-Labeling and Confirmation Bias in Deep Semi-Supervised Learningcited by 507
- 2019Be Your Own Teacher: Improve the Performance of Convolutional Neural Networks via Self Distillationcited by 491
- 2018Texygen: A Benchmarking Platform for Text Generation Modelscited by 484
- 2023RRHF: Rank Responses to Align Language Models with Human Feedback without tearscited by 454
- 2023Scalable Extraction of Training Data from (Production) Language Modelscited by 415
- 2023SLiC-HF: Sequence Likelihood Calibration with Human Feedbackcited by 395
- 2023Can AI-Generated Text be Reliably Detected?cited by 372
- 2021ZeRO-Offload: Democratizing Billion-Scale Model Trainingcited by 339
- 2022Language Models Are Greedy Reasoners: A Systematic Formal Analysis of Chain-of-Thoughtcited by 334
- 2020The Pushshift Reddit Datasetcited by 289
- 2020Systematic Evaluation of Privacy Risks of Machine Learning Modelscited by 273
- 2019Automatic Detection of Generated Text is Easiest when Humans are Fooledcited by 260
- 2018A Simple Method for Commonsense Reasoningcited by 223
- 2019BERT has a Mouth, and It Must Speak: BERT as a Markov Random Field Language Modelcited by 184
- 2020Better Fine-Tuning by Reducing Representational Collapsecited by 156
- 2019Revisiting Self-Training for Neural Sequence Generationcited by 154
- 2022Measuring Forgetting of Memorized Training Examplescited by 118
50 most cited of 59 All 59 (JSON)
Built on by
Not cited yet.