mlsanityData to 7 Oct 2026

Amplifying Training Data Exposure through Fine-Tuning with Pseudo-Labeled Memberships

arXiv 2402.12189 ↗First posted 19 Feb 2024cs.CL cs.CR cs.LGcited by 0

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

  1. 2023Membership Inference Attacks against Language Models via Neighbourhood Comparison29 May
  2. 2022RelaxLoss: Defending Membership Inference Attacks without Losing Utility12 Jul
  3. 2022Quantifying Privacy Risks of Masked Language Models Using Membership Inference Attacks8 Mar
  4. 2021Membership Inference Attacks From First Principles7 Dec
  5. 2021Enhanced Membership Inference Attacks against Machine Learning Models18 Nov
  6. 2021Mitigating Membership Inference Attacks by Self-Distillation Through a Novel Ensemble Architecture15 Oct

Other references · 59

  1. 2018BERT: Pre-training of Deep Bidirectional Transformers for Language Understandingcited by 40,204
  2. 2020Language Models are Few-Shot Learnerscited by 32,199
  3. 2019PyTorch: An Imperative Style, High-Performance Deep Learning Librarycited by 20,160
  4. 2023GPT-4 Technical Reportcited by 18,132
  5. 2022Training language models to follow instructions with human feedbackcited by 15,712
  6. 2019Exploring the Limits of Transfer Learning with a Unified Text-to-Text Transformercited by 14,491
  7. 2017Proximal Policy Optimization Algorithmscited by 14,441
  8. 2021LoRA: Low-Rank Adaptation of Large Language Modelscited by 14,084
  9. 2023LLaMA: Open and Efficient Foundation Language Modelscited by 13,958
  10. 2019RoBERTa: A Robustly Optimized BERT Pretraining Approachcited by 13,266
  11. 2023Llama 2: Open Foundation and Fine-Tuned Chat Modelscited by 12,865
  12. 2023Direct Preference Optimization: Your Language Model is Secretly a Reward Modelcited by 7,916
38 more
  1. 2016SGDR: Stochastic Gradient Descent with Warm Restartscited by 4,993
  2. 2017Deep reinforcement learning from human preferencescited by 3,846
  3. 2022OPT: Open Pre-trained Transformer Language Modelscited by 3,464
  4. 2023Universal and Transferable Adversarial Attacks on Aligned Language Modelscited by 2,914
  5. 2019The Curious Case of Neural Text Degenerationcited by 2,881
  6. 2016Deep Learning with Differential Privacycited by 2,682
  7. 2020Learning to summarize from human feedbackcited by 2,562
  8. 2020The Pile: An 800GB Dataset of Diverse Text for Language Modelingcited by 2,346
  9. 2016Membership Inference Attacks against Machine Learning Modelscited by 2,221
  10. 2020Extracting Training Data from Large Language Modelscited by 2,144
  11. 2023Code Llama: Open Foundation Models for Codecited by 2,105
  12. 2022BLOOM: A 176B-Parameter Open-Access Multilingual Language Modelcited by 1,968
  13. 2020mT5: A massively multilingual pre-trained text-to-text transformercited by 1,813
  14. 2019PubMedQA: A Dataset for Biomedical Research Question Answeringcited by 1,271
  15. 2023PaLM 2 Technical Reportcited by 1,154
  16. 2015Aligning Books and Movies: Towards Story-like Visual Explanations by Watching Movies and Reading Bookscited by 1,041
  17. 2022Red Teaming Language Models with Language Modelscited by 983
  18. 2022Quantifying Memorization Across Neural Language Modelscited by 755
  19. 2021Deduplicating Training Data Makes Language Models Bettercited by 705
  20. 2023DetectGPT: Zero-Shot Machine-Generated Text Detection using Probability Curvaturecited by 705
  21. 2019Defending Against Neural Fake Newscited by 553
  22. 2019Pseudo-Labeling and Confirmation Bias in Deep Semi-Supervised Learningcited by 507
  23. 2019Be Your Own Teacher: Improve the Performance of Convolutional Neural Networks via Self Distillationcited by 491
  24. 2018Texygen: A Benchmarking Platform for Text Generation Modelscited by 484
  25. 2023RRHF: Rank Responses to Align Language Models with Human Feedback without tearscited by 454
  26. 2023Scalable Extraction of Training Data from (Production) Language Modelscited by 415
  27. 2023SLiC-HF: Sequence Likelihood Calibration with Human Feedbackcited by 395
  28. 2023Can AI-Generated Text be Reliably Detected?cited by 372
  29. 2021ZeRO-Offload: Democratizing Billion-Scale Model Trainingcited by 339
  30. 2022Language Models Are Greedy Reasoners: A Systematic Formal Analysis of Chain-of-Thoughtcited by 334
  31. 2020The Pushshift Reddit Datasetcited by 289
  32. 2020Systematic Evaluation of Privacy Risks of Machine Learning Modelscited by 273
  33. 2019Automatic Detection of Generated Text is Easiest when Humans are Fooledcited by 260
  34. 2018A Simple Method for Commonsense Reasoningcited by 223
  35. 2019BERT has a Mouth, and It Must Speak: BERT as a Markov Random Field Language Modelcited by 184
  36. 2020Better Fine-Tuning by Reducing Representational Collapsecited by 156
  37. 2019Revisiting Self-Training for Neural Sequence Generationcited by 154
  38. 2022Measuring Forgetting of Memorized Training Examplescited by 118

50 most cited of 59 All 59 (JSON)

Built on by

Not cited yet.