atlas

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

Stress-Testing Capability Elicitation With Password-Locked Models

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

Introduces password-locked LLMs that show strong capabilities only when a password is present, and uses them to stress-test whether fine-tuning (SFT and RL) can elicit hidden capabilities without the password.

Problem

Capability elicitation via prompting often fails to reveal a model's full abilities, so dangerous capabilities may be underestimated; it is unclear when fine-tuning-based elicitation reliably recovers hidden capabilities.

Approach

Deliberately lock capabilities by fine-tuning a model to imitate a strong policy only when a password is in the prompt and a weak policy otherwise, then measure how much fine-tuning-based elicitation (SFT on demonstrations, expert iteration, DPO) recovers performance without the password.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
SFT on a few high-quality demonstrations (GPT-4 T=0) for locked modelPassword-locked Deepseek-7B-Coder (code generation) · Code generationcorrectnessHelped
SFT on high-quality demonstrations for locked modelPassword-locked models across tasks except code critiques · Code, MATH, MMLUcorrectness relative to locked-with-password ceilingHelped
SFT on reference critiques for locked modelPassword-locked model on code critiques · Code critiquescorrectnessNo clear effect
SFT on Llama7B demonstrations instead of high-quality demonstrationsPassword-locked models · Across taskscorrectness relative to ceilingHurt
Expert iteration (RL with binary correctness reward)Password-locked models starting from weak policy · Code and critiquescorrectnessHelped
Starting RL from a weak policy with very poor correctness (Pythia-1B on MATH, fixed random distribution)RL elicitation (expert iteration, DPO) · MATH and fixedrdmcorrectnessHurt
Continuous rewards instead of discrete rewards for RL (ablation)RL elicitation · Code and critiquestraining speed and final correctnessMixed
Locking from scratch instead of via SFT on pretrained model (ablation)Password-locked small transformers (function composition) · Function composition toy taskdata needed to unlock: 2-4x more data neededHurt
Supervised fine-tuning on pi_weak samples (weak sample unlocking) (ablation)Password-locked models · Across taskscorrectnessMixed
Demonstrations restricted to single domain (MATH category/difficulty) (ablation)Password-locked models · MATHcorrectnessNo clear effect
Demonstrations from a domain not locked (disjoint support) (ablation)Password-locked models · Domain generalizationsample efficiencyNo clear effect
Using a setting variation (Spanish + ASCII-Math) as trigger instead of password (ablation)Password-locked models on MATH · MATH Spanish ASCIIsample efficiency and RL performanceNo clear effect

Surprising findings

Limitations

Experimental setup

Datasets: APPS MBPP MATH MMLU

Benchmarks: APPS MBPP MATH MMLU

Metrics: correctness (greedy decoding) sample efficiency

Baselines: filtering then supervised fine-tuning DPO iterated DPO expert iteration fixedrdm

Builds on: DPO expert iteration PPO Sleeper Agents

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