atlas

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

Localized Zeroth-Order Prompt Optimization

NeurIPS 2024 · Large language models › Language, safety and interpretability · approach: LLM unlearning and benchmarks · 2 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 ZOPO, a localized zeroth-order prompt optimizer that runs gradient descent in a continuous embedding space of LLM-generated prompts using an NTK-derived Gaussian process surrogate and uncertainty-informed local exploration.

Problem

Black-box discrete prompt optimization for LLMs is query-inefficient because global optimization spends budget searching for a rare global optimum, while local optima are prevalent and well-performing.

Approach

Empirically motivated shift from global to local optimization: maps prompts to continuous embeddings (via a generator LLM plus embedding model), then does zeroth-order gradient descent with a gradient estimator from an NTK-based derived Gaussian process, plus local queries where predictive uncertainty is high.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Use ZOPO to optimize the chain-of-thought prompt instead of the hand-crafted promptzero-shot arithmetic reasoning prompts · GSM8Kaccuracy: 71.797 -> 75.360Helped
Use ZOPO to optimize the chain-of-thought prompt instead of the hand-crafted promptzero-shot arithmetic reasoning prompts · AQUA-RATaccuracy: 52.362 -> 54.724Helped
Use ZOPO to optimize the chain-of-thought prompt instead of the hand-crafted promptzero-shot arithmetic reasoning prompts · SVAMPaccuracy: 76.25 -> 81.0Helped
Use ZOPO rather than the strongest baselineinstruction induction · 20 instruction induction taskstest accuracyMixed
Generate candidates with ChatGPT rather than Vicuna-13B, using SBERT for both (ablation)ZOPO · instruction inductiontest accuracyMixed
Use structured last-token, OpenAI, or SBERT embeddings instead of random embeddings (ablation)ZOPO with fixed Vicuna-13B prompts · instruction inductiontest accuracyHelped
Replace the NTK kernel with a Matérn kernel (ablation)ZOPO · instruction induction taskstest accuracyHurt
Remove uncertainty-informed local exploration (ablation)ZOPO · instruction induction taskstest accuracyHurt
Increase the generated prompt candidate set from 500 to 1000 or 2000 (ablation)ZOPO and ZOPO-GPT · instruction inductiontest accuracyNo clear effect
Use ChatGPT-generated candidates with SBERT rather than Vicuna-generated candidates with SBERT (ablation)ZOPO · second_word_lettertest accuracy: 24.3_{\pm6.0} -> 96.7_{\pm2.4}Helped
Use ChatGPT-generated candidates with SBERT rather than Vicuna-generated candidates with SBERT (ablation)ZOPO · cause_and_effecttest accuracy: 86.7_{\pm7.5} -> 80.0_{\pm14.2}Hurt
Use ZOPO-GPT rather than the best baselineinstruction induction · second_word_lettertest accuracy: 74.7_{\pm20.3} -> 96.7_{\pm2.4}Helped

Surprising findings

Limitations

Experimental setup

Datasets: instruction induction (30 tasks) GSM8K AQUA-RAT SVAMP

Benchmarks: instruction induction (20 and 30 tasks) GSM8K AQUA-RAT SVAMP

Metrics: test accuracy performance profile rho(tau) query efficiency

Baselines: APE InstructZero INSTINCT EvoPrompt

Builds on: LoZO InstructZero INSTINCT APE

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