Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

SAGE: Stochastic Prompt Optimization via Agent-Guided Exploration

About

Context engineering has emerged as a primary lever for improving AI systems without parameter updates. Recent work showing that textual gradients do not function as real gradients motivates treating automatic prompt optimization (APO) as black-box search. We introduce SPO (Stochastic Prompt Optimization), a framework for stochastic search over prompt space, and compare three strategies of increasing sophistication: error-informed random search, a genetic algorithm with evolutionary operators, and SAGE (SPO via Agent-Guided Exploration), a multi-agent pipeline with diagnostic code execution. Across three benchmarks, no single strategy dominates; effectiveness depends on the interaction of landscape structure with error type. We further deploy SAGE on a mental-health chatbot under a continuous optimization paradigm, where it compounds eight cycles of individually-noisy A/B tests into a statistically robust gain in next-day retention. We argue that coupling qualitative diagnosis with quantitative validation is what makes agentic optimization effective for open-ended task-oriented dialogue.

Ziyi Zhu, Luka Smyth, Saki Shinoda, Jinghong Chen• 2026

Related benchmarks

TaskDatasetResultRank
Information ExtractionFiNER (test)
Accuracy77.5
9
ReasoningFormula (test)
Accuracy86.5
9
Agentic TaskAppWorld normal (test)
Task Goal Completion (TGC)79.8
8
Showing 3 of 3 rows

Other info

Follow for update