atlas

Research map / Large language models / Language, safety and interpretability

PICLe: Eliciting Diverse Behaviors from Large Language Models with Persona In-Context Learning

ICML 2024 · Large language models › Language, safety and interpretability · approach: LLM pruning and scaling data · 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.

Proposes PICLe, a Bayesian-motivated in-context example selection method that ranks persona statements by the log-likelihood ratio between a persona-finetuned and the original LLM, to elicit target personas from LLMs.

Problem

Eliciting a specified persona from LLMs by probing their yes/no behavioral responses to persona statements; base prompting is only moderately better than chance (65.5% on Llama-2), so the question is how to select in-context demonstrations that elicit a target persona effectively.

Approach

Select ICL demonstrations by the likelihood ratio log p_persona(x) - log p_original(x), where the persona model is a LoRA fine-tune on the persona's statement pool; this favors statements indicative of the persona but not already well represented by the base LLM.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Use likelihood-ratio selection instead of similarity-based example selectionLlama-2-7B-Chat PICLe · Anthropic Persona test setAction Consistency: 84.6 -> 88.1Helped
Use likelihood-ratio selection instead of action-entropy selectionVicuna-7B-v1.5 · Anthropic Persona test setAction Consistency: 76.1 -> 78.6Helped
Use likelihood-ratio selection instead of the strongest baseline, Certainty-tokenGPT-J-6B · Anthropic Persona test setAction Consistency: 61.4 -> 67.0Helped
Add three PICLe in-context examplesLlama-2-7B-Chat · Anthropic Persona test setAction Consistency: 65.5 -> 88.1Helped
Use positive-labeled statements for Persona SFT and ICL selectionLlama-2-7B-Chat PICLe · Anthropic Persona test set, label-awareAction Consistency: 88.1 -> 93.1Helped
Increase the number of ICL examples to 10 (ablation)Llama-2-7B-Chat PICLe · Anthropic Persona test setAction Consistency: 88.1 -> 92.3Helped
Replace likelihood-ratio scoring with persona-model likelihood alone (ablation)Llama-2-7B-Chat PICLe · Anthropic Persona test set, label-agnosticAction Consistency: 88.1 -> 72.1Hurt
Replace likelihood-ratio scoring with base-model likelihood alone (ablation)Llama-2-7B-Chat PICLe · Anthropic Persona test set, label-agnosticAction Consistency: 88.1 -> 71.8Hurt
Replace likelihood-ratio scoring with persona-model likelihood alone (ablation)Llama-2-7B-Chat PICLe+ · Anthropic Persona test set, label-awareAction Consistency: 93.1 -> 87.8Hurt
Replace likelihood-ratio scoring with base-model likelihood alone (ablation)Llama-2-7B-Chat PICLe+ · Anthropic Persona test set, label-awareAction Consistency: 93.1 -> 87.0Hurt
Vary Persona SFT epochs from 1 to 10 instead of fixing 4 (ablation)Llama-2-7B-Chat PICLe · Anthropic Persona test setAction Consistency: 87.6; 87.6; 88.7; 88.1; 88.6; 88.6; 88.4; 88.3; 88.4; 88.5No clear effect
Reduce Persona SFT and ICL data to 40% of the full data (ablation)Llama-2-7B-Chat PICLe · Anthropic Persona test setAction Consistency: 88.1 -> 87.0No clear effect

Surprising findings

Limitations

Experimental setup

Datasets: Anthropic model-written-evals persona dataset

Benchmarks: Anthropic persona dataset (99 personas, 300 test statements each)

Metrics: action consistency action confidence action uncertainty token-level uncertainty degree of alteration (KL divergence)

Baselines: Base Instructive prompting Descriptive prompting Random ICL Similarity ICL Uncertainty ICL Certainty ICL Diversity ICL Likelihood ICL

Builds on: in-context learning LoRA Anthropic model-written evals persona dataset

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