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.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Information Extraction | FiNER (test) | Accuracy77.5 | 9 | |
| Reasoning | Formula (test) | Accuracy86.5 | 9 | |
| Agentic Task | AppWorld normal (test) | Task Goal Completion (TGC)79.8 | 8 |