English

Mind the Privacy Unit! User-Level Differential Privacy for Language Model Fine-Tuning

Computation and Language 2024-08-19 v3 Cryptography and Security Machine Learning

Abstract

Large language models (LLMs) have emerged as powerful tools for tackling complex tasks across diverse domains, but they also raise privacy concerns when fine-tuned on sensitive data due to potential memorization. While differential privacy (DP) offers a promising solution by ensuring models are 'almost indistinguishable' with or without any particular privacy unit, current evaluations on LLMs mostly treat each example (text record) as the privacy unit. This leads to uneven user privacy guarantees when contributions per user vary. We therefore study user-level DP motivated by applications where it necessary to ensure uniform privacy protection across users. We present a systematic evaluation of user-level DP for LLM fine-tuning on natural language generation tasks. Focusing on two mechanisms for achieving user-level DP guarantees, Group Privacy and User-wise DP-SGD, we investigate design choices like data selection strategies and parameter tuning for the best privacy-utility tradeoff.

Keywords

Cite

@article{arxiv.2406.14322,
  title  = {Mind the Privacy Unit! User-Level Differential Privacy for Language Model Fine-Tuning},
  author = {Lynn Chua and Badih Ghazi and Yangsibo Huang and Pritish Kamath and Ravi Kumar and Daogao Liu and Pasin Manurangsi and Amer Sinha and Chiyuan Zhang},
  journal= {arXiv preprint arXiv:2406.14322},
  year   = {2024}
}

Comments

Published as a conference paper at COLM 2024

R2 v1 2026-06-28T17:13:27.331Z