From Prompts to Responses: Dual-Sided Data Leakage and Defense in Split Large Language Models
Not on the map. Why?
Abstract (arXiv)
Large language models (LLMs) are increasingly deployed in privacy-sensitive domains, where users must balance the risk of data exposure through external APIs against the high computational cost of local deployment. Split learning has therefore emerged as a promising paradigm for LLM fine-tuning and inference under limited local resources. However, it introduces new privacy risks. Prior work primarily studies leakage of private input prompts, typically via inversion attacks on intermediate representations, while the potential for sensitive information leakage through generative response outputs remains largely unexplored. In this work, we unveil novel vulnerabilities of Split-LLM by presenting Patched Model Inversion with Dual-Sided Initialization (PIDI), a two-stage attack that simultaneously targets both private input prompts and output responses in Split-LLM settings. It combines dual-sided initialization with a patched inversion strategy to tackle long sequences, substantially outperforming prior inversion methods. To counter threats from both sides, we further propose the Adapter-based DualGuard with Mutual Information Defense (ADMI), which integrates an adapter-based local warmup strategy and mutual information regularization to provide a strong empirical privacy protection with minimal impact on task performance. Extensive experiments across diverse tasks and models demonstrate that ADMI effectively defends against PIDI and other state-of-the-art inversion attacks. Our code is publicly available at https://github.com/FLAIR-THU/VFLAIR-LLM.
Key facts. Cited by 2 papers in the map (data through 14 Oct 2026). Builds on 20 of its 32 references that are in the map. First posted on arXiv 12 Jun 2026.
Builds on
20 of 32 references.
References · 20
- 2019BERTScore: Evaluating Text Generation with BERTcited by 5,512
- 2020BLEURT: Learning Robust Metrics for Text Generationcited by 1,203
- 2019Deep Leakage from Gradientscited by 1,021
- 2016Deep Variational Information Bottleneckcited by 930
- 2023MedAlpaca -- An Open-Source Collection of Medical Conversational AI Models and Training Datacited by 286
- 2018Distributed learning of deep neural network over multiple agentscited by 279
- 2017Sparse Communication for Distributed Gradient Descentcited by 249
- 2020Information Leakage in Embedding Modelscited by 203
- 2021Label Leakage and Protection in Two-party Split Learningcited by 90
- 2021Differential Privacy for Text Analytics via Natural Text Sanitizationcited by 77
- 2023DP-Forward: Fine-tuning and Inference on Language Models with Differential Privacy in Forward Passcited by 67
- 2021TAG: Gradient Attack on Transformer-based Language Modelscited by 63
8 more
- 2024SplitLoRA: A Split Parameter-Efficient Fine-Tuning Framework for Large Language Modelscited by 62
- 2022A Customized Text Sanitization Mechanism with Differential Privacycited by 52
- 2023Split-and-Denoise: Protect large language model inference with local differential privacycited by 32
- 2024Unveiling the Vulnerability of Private Fine-Tuning in Split-Based Frameworks for Large Language Models: A Bidirectionally Enhanced Attackcited by 18
- 2025Prompt Inversion Attack against Collaborative Inference of Large Language Modelscited by 15
- 2023A Split-and-Privatize Framework for Large Language Model Fine-Tuningcited by 11
- 2025VFLAIR-LLM: A Comprehensive Framework and Benchmark for Split Learning of LLMscited by 5
- 2023PrivaScissors: Enhance the Privacy of Collaborative Inference through the Lens of Mutual Informationcited by 4
Built on by
Citing papers per year: 2026: 2.
2 papers.