Recent advancements in pre-trained large language models (LLMs) have
significantly influenced various domains. Adapting these models for specific
tasks often involves fine-tuning (FT) with private, domain-specific data.
However, privacy concerns keep this data undisclosed, and the computational
demands for deploying LLMs pose challenges for resource-limited data holders.
This has sparked interest in split learning (SL), a Model-as-a-Service (MaaS)
paradigm that divides LLMs into smaller segments for distributed training and
deployment, transmitting only intermediate activations instead of raw data. SL
has garnered substantial interest in both industry and academia as it aims to
balance user data privacy, model ownership, and resource challenges in the
private fine-tuning of LLMs. Despite its privacy claims, this paper reveals
significant vulnerabilities arising from the combination of SL and LLM-FT: the
Not-too-far property of fine-tuning and the auto-regressive nature of LLMs.
Exploiting these vulnerabilities, we propose Bidirectional Semi-white-box
Reconstruction (BiSR), the first data reconstruction attack (DRA) designed to
target both the forward and backward propagation processes of SL. BiSR utilizes
pre-trained weights as prior knowledge, combining a learning-based attack with
a bidirectional optimization-based approach for highly effective data
reconstruction. Additionally, it incorporates a Noise-adaptive Mixture of
Experts (NaMoE) model to enhance reconstruction performance under perturbation.
We conducted systematic experiments on various mainstream LLMs and different
setups, empirically demonstrating BiSR's state-of-the-art performance.
Furthermore, we thoroughly examined three representative defense mechanisms,
showcasing our method's capability to reconstruct private data even in the
presence of these defenses.
FedSL: Federated split learning on distributed sequential data in recurrent neural networks
Ali Abedi, Shehroz S Khan
Published: 2024
NeurIPS
Lamp: Extracting text from gradients with language model priors
Mislav Balunovic, Dimitar Dimitrov, Nikola Jovanovic, Martin Vechev
Published: 2022
ACL workshop on intrinsic and extrinsic evaluation measures for machine translation and/or summarization
METEOR: An automatic metric for MT evaluation with improved correlation with human judgments
Satanjeev Banerjee, Alon Lavie
Published: 2005
Proceedings of the AAAI conference on artificial intelligence
Piqa: Reasoning about physical commonsense in natural language
Bisk, Y., Zellers, R., Gao, J., Choi, Y.
Published: 2020
OpenAI Technical Report
Language models are few-shot learners
T. B. Brown, B. Mann, N. Ryder, M. Subbiah, J. Kaplan, P. Dhariwal, A. Neelakantan, P. Shyam, G. Sastry, A. Askell, S. Agarwal, A. Herbert-Voss, G. Krueger, T. Henighan, R. Child, A. Ramesh, D. M. Ziegler, J. Wu, C. Winter, C. Hesse, M. Chen, E. Sigler, M. Litwin, S. Gray, B. Chess, J. Clark, C. Berner, S. McCandlish, A. Radford, I. Sutskever, D. Amodei