atlas

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

Privacy-Preserving Instructions for Aligning Large Language Models

ICML 2024 · Large language models › Instruction and fine-tuning · approach: Safety alignment after fine-tuning · 5 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.

Generates differentially private synthetic user instructions by DP-fine-tuning LLaMA generators and resampling them with a DP histogram filter so they match real Chatbot Arena instruction distributions, then uses them for SFT and PPO-based RLHF.

Problem

User instructions collected by LLM services contain sensitive information that is exposed to human annotators during data labeling and can be memorized by aligned models; existing DP fine-tuning only addresses memorization.

Approach

Replace real user instructions with DP synthetic instructions from a privately fine-tuned generator, then apply a novel DP histogram-based resampling step (k-means clusters of synthetic embeddings, noisy histogram of real data) to align the synthetic distribution with real instructions at small extra privacy cost.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Resample initial DP synthetic instructions with the private histogramDP synthetic instructions generated from LLaMA 13B · Chatbot Arena instructionsMAUVE unigram: 0.942 -> 0.961Helped
Resample initial DP synthetic instructions with the private histogramDP synthetic instructions generated from LLaMA 13B · Chatbot Arena instructionsMAUVE Sentence-T5: 0.912 -> 0.975Helped
Resample initial DP synthetic instructions with the private histogramDP synthetic reviews generated from GPT2-Large · IMDB reviewsMAUVE Sentence-T5: 0.647 -> 0.825Helped
Resample initial DP synthetic abstracts with the private histogramDP synthetic abstracts generated from GPT2-Large · PubMed abstractsMAUVE Sentence-T5: 0.594 -> 0.811Helped
Use filtered rather than unfiltered DP synthetic instructionsLLaMA 7B supervised fine-tuning · Chatbot Arena instructions (test)win-rate: 62.7% (± 0.34) -> 65.6% (±0.36)Helped
Use filtered DP synthetic instructions instead of DP training on real instructionsLLaMA 7B supervised fine-tuning · Chatbot Arena instructions (test)win-rate: 60.7% (± 0.42) -> 67.8% (± 0.32)Helped
Use in-domain Chatbot Arena instructions instead of out-of-domain FLAN instructionsLLaMA 7B supervised fine-tuning · Chatbot Arena instructions (test)win-rate: 50% -> 68.9% (± 0.31)Helped
Increase the gradient clipping threshold from C=0.01 to C=0.5 (ablation)DP instruction generator · Chatbot Arena instructionsMAUVE unigram: 0.667 -> 0.839Helped
Increase the gradient clipping threshold from C=0.01 to C=0.5 (ablation)DP instruction generator · Chatbot Arena instructionsMAUVE Sentence-T5: 0.843 -> 0.890Helped
Use LLaMA 13B rather than LLaMA 7B as the pretrained generator (ablation)DP instruction generator · Chatbot Arena instructionsMAUVE unigram: 0.839 -> 0.942Helped
Use LLaMA 13B rather than LLaMA 7B as the pretrained generator (ablation)DP instruction generator · Chatbot Arena instructionsMAUVE Sentence-T5: 0.890 -> 0.912Helped
Use 500 thousand rather than one million initial synthetic instructions (ablation)Private histogram resampling · Chatbot Arena instructions, large KMAUVEHurt

Surprising findings

Limitations

Experimental setup

Datasets: LMSYS-Chat-1M Chatbot Arena Conversations FLAN OpenOrca IMDB PubMed

Benchmarks: Chatbot Arena instructions test split AlpacaEval MAUVE on Chatbot Arena instructions

Metrics: win-rate MAUVE (unigram and Sentence-T5) average test reward loss rank leakage rate

Baselines: Vicuna-v1.3 LLaMA2-Chat FLAN (non-private) Chatbot Arena (non-private) Text-davinci-003

Builds on: DP-SGD DP-Adam MAUVE Private Evolution LoRA

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