atlas

Research map / Large language models / Instruction and fine-tuning

DPZero: Private Fine-Tuning of Language Models without Backpropagation

ICML 2024 · Large language models › Instruction and fine-tuning · approach: Calibration and knowledge in LLM tuning · 1 citations (OpenAlex)

Read the paperCheck your idea against 108,967 papers

AI-extracted summary. Written by AI from the paper's full text. Numbers and claims can be wrong or missing context, so check the paper before you cite or build on it.

Proposes DPZero, a differentially private zeroth-order method that privatizes only a scalar finite-difference magnitude along public random directions, achieving near dimension-independent rates for fine-tuning RoBERTa and OPT without backpropagation.

Problem

Fine-tuning LLMs faces high memory cost of backpropagation and privacy risks of memorization; existing differentially private first-order methods have dimension-dependent rates and high per-sample clipping overhead, and naive combination of DP-GD with zeroth-order gradients scales poorly with dimens

Approach

Decouple the zeroth-order gradient into a public random direction u_t and a scalar magnitude; clip and add scalar Gaussian noise only to the magnitude, and use a small clipping threshold justified by the fact that |u^T grad| concentrates at scale L (sphere tail bound), giving logarithmic dependence on d under low effective rank.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Use DPZero instead of MeZO for RoBERTa fine-tuning at ε=6RoBERTa-355M · Six sentence-classification tasksaccuracy (%)Mixed
Use DPZero instead of DP-AdamW at ε=6RoBERTa-355M · Six sentence-classification tasksaccuracy (%)Mixed
Use DPZero with ε=2 rather than ε=6DPZero on RoBERTa-355M · Six sentence-classification tasksaccuracy (%)Hurt
Use DPZero with ε=6 rather than non-private MeZOOPT models · Classification and generation tasksaccuracy / F1No clear effect
Sample the update direction from a Gaussian rather than a sphere (ablation)DPZero on RoBERTa-355M · SST-2 and SST-5accuracy (%): $92.2\pm0.3$ vs $91.8\pm0.1$; $91.8\pm0.1$ vs $91.5\pm0.5$; $49.3\pm0.6$ vs $49.9\pm1.3$; $47.1\pm0.9$ vs $47.4\pm1.3$No clear effect
Use DPZero rather than DPGD-0thRoBERTa-large · TREC, ε=2accuracy (%): 67.0 to 82.0Helped
Increase the privacy budget from ε=2 to ε=6DPZero on OPT · Classification and generation tasksaccuracy / F1Helped

Surprising findings

Limitations

Experimental setup

Datasets: SST-2 SST-5 SNLI MNLI RTE TREC BoolQ SQuAD DROP

Benchmarks: SST-2 SST-5 SNLI MNLI RTE TREC BoolQ SQuAD DROP

Metrics: accuracy f1 score gradient norm runtime per iteration memory consumption (MiB)

Baselines: MeZO DP-AdamW DP-LoRA AdamW LoRA DPGD-0th DP-GD zero-shot

Builds on: DP-GD MeZO zeroth-order SGD

Closest papers by meaning

Writing a paper in this area? scime draft is a free LaTeX editor with AI included, real-time co-authors and a check of your references and venue rules before you submit. It also imports OpenAI Prism projects.

Open scime draft

Get the next scime Dispatch and a monthly field update

Free. We send a confirmation link first; every email has a one-click unsubscribe. Privacy