mlsanityData to 7 Oct 2026

Extracting Training Dialogue Data from Large Language Model based Task Bots

arXiv 2603.01550 ↗First posted 2 Mar 2026cs.CL cs.AIcited by 0

On the map: LLM safety and security → Machine unlearning

Abstract (arXiv)

Large Language Models (LLMs) have been widely adopted to enhance Task-Oriented Dialogue Systems (TODS) by modeling complex language patterns and delivering contextually appropriate responses. However, this integration introduces significant privacy risks, as LLMs, functioning as soft knowledge bases that compress extensive training data into rich knowledge representations, can inadvertently memorize training dialogue data containing not only identifiable information such as phone numbers but also entire dialogue-level events like complete travel schedules. Despite the critical nature of this privacy concern, how LLM memorization is inherited in developing task bots remains unexplored. In this work, we address this gap through a systematic quantitative study that involves evaluating existing training data extraction attacks, analyzing key characteristics of task-oriented dialogue modeling that render existing methods ineffective, and proposing novel attack techniques tailored for LLM-based TODS that enhance both response sampling and membership inference. Experimental results demonstrate the effectiveness of our proposed data extraction attack. Our method can extract thousands of training labels of dialogue states with best-case precision exceeding 70%. Furthermore, we provide an in-depth analysis of training data memorization in LLM-based TODS by identifying and quantifying key influencing factors and discussing targeted mitigation strategies.

Key facts. No citing papers in the map yet (data through 7 Oct 2026). Placed in Machine unlearning, one of 12 directions in LLM safety and security. Builds on 27 of its 35 references that are in the map. First posted on arXiv 2 Mar 2026.

Builds on

27 of 35 references.

References · 27

  1. 2023GPT-4 Technical Reportcited by 18,132
  2. 2021LoRA: Low-Rank Adaptation of Large Language Modelscited by 14,084
  3. 2023Llama 2: Open Foundation and Fine-Tuned Chat Modelscited by 12,865
  4. 2015Distilling the Knowledge in a Neural Networkcited by 10,076
  5. 2014Sequence to Sequence Learning with Neural Networkscited by 2,974
  6. 2019The Curious Case of Neural Text Degenerationcited by 2,881
  7. 2020Extracting Training Data from Large Language Modelscited by 2,144
  8. 2018Hierarchical Neural Story Generationcited by 1,182
  9. 2022Quantifying Memorization Across Neural Language Modelscited by 755
  10. 2023Extracting Training Data from Diffusion Modelscited by 723
  11. 2023Scalable Extraction of Training Data from (Production) Language Modelscited by 415
  12. 2020A Simple Language Model for Task-Oriented Dialoguecited by 331
15 more
  1. 2022Deduplicating Training Data Mitigates Privacy Risks in Language Modelscited by 309
  2. 2016Diverse Beam Search: Decoding Diverse Solutions from Neural Sequence Modelscited by 299
  3. 2021NeuroLogic A*esque Decoding: Constrained Text Generation with Lookahead Heuristicscited by 153
  4. 2019Reconstruction and Membership Inference Attacks against Generative Modelscited by 122
  5. 2020UBAR: Towards Fully End-to-End Task-Oriented Dialog Systems with GPT-2cited by 111
  6. 2022Locally Typical Samplingcited by 108
  7. 2021GALAXY: A Generative Pre-trained Model for Task-Oriented Dialog with Semi-Supervised Learning and Explicit Policy Injectioncited by 102
  8. 2019Comparison of Diverse Decoding Methods from Conditional Language Modelscited by 78
  9. 2024Follow My Instruction and Spill the Beans: Scalable Data Extraction from Retrieval-Augmented Generation Systemscited by 76
  10. 2023Bag of Tricks for Training Data Extraction from Language Modelscited by 67
  11. 2023Exploring Memorization in Fine-tuned Language Modelscited by 44
  12. 2021AuGPT: Auxiliary Tasks and Data Augmentation for End-To-End Dialogue with Pre-Trained Language Modelscited by 31
  13. 2022Semi-Supervised Knowledge-Grounded Pre-training for Task-Oriented Dialog Systemscited by 15
  14. 2022Mars: Modeling Context & State Representations with Contrastive Learning for End-to-End Task-Oriented Dialogcited by 11
  15. 2024TA&AT: Enhancing Task-Oriented Dialog with Turn-Level Auxiliary Tasks and Action-Tree Based Scheduled Samplingcited by 1

Built on by

Not cited yet.