Explore More, Validate Less: Efficient Reflective Prompt Optimization
Organizations: Pennsylvania State University, USA
Abstract
Reflective prompt optimization improves instructions using task feedback, but coupling candidate generation with full-validation-based selection can make search costly and sequential. We propose a two-stage framework that decouples reflective candidate-pool construction from validation allocation. During construction, sampled parents first answer training questions, and groups of their saved responses guide parallel reflection and revision. Each child is screened on its group's questions against the corresponding parent scores. Improving, nonduplicate prompts enter the pool and can become parents without full validation. After construction, an exploration policy selects candidates from the frozen pool under a fixed full-validation budget, and the prompt with the highest measured validation score is returned. This separation allows candidate-pool size and validation budget to be set independently. We evaluate random selection and a multilayer perceptron with similarity-based exploration (MLP + similarity) on five benchmarks using Qwen3.5-9B and Llama-3.1-8B-Instruct. With 100 retained candidates and 20 full validations, our framework improves over the initial solver by up to 11.22 percentage points. MLP + similarity improves over the initial solver in all ten comparisons, although neither selection policy consistently outperforms the other. These results support separating reflective search from budgeted validation while leaving room for stronger candidate-selection policies.
Figures & tables
| Method | SuperGPQA | LiveBench-Math | IFBench | BBEH | HotpotQA | Average |
|---|---|---|---|---|---|---|
| Llama-3.1-8B-Instruct | ||||||
| Initial solver | ||||||
| MIPROv2 | ||||||
| GEPA | ||||||
| DPO | ||||||
| Ours (Random) | ||||||
| Ours (Random) | 20 rounds of GEPA | |||
| Dataset | Build | Validation | Build | Validation |
| Llama-3.1-8B-Instruct | ||||
| SuperGPQA | 53.92 | 54.42 | 85.91 | 56.58 |
| LiveBench-Math | 219.51 | 35.54 | 287.26 | 37.24 |
| IFBench | 38.77 | 43.33 | 46.02 | 40.97 |
| BBEH | 173.17 | 76.37 | 367.25 | 78.51 |
| Mean depth | Mean branching | |||
| Dataset | Ours | GEPA | Ours | GEPA |
| Llama-3.1-8B-Instruct | ||||
| SuperGPQA | 4.37 | 3.47 | 2.41 | 1.81 |
| LiveBench-Math | 4.09 | 3.00 | 2.15 | 2.23 |
| IFBench | 3.37 | 5.73 | 3.19 | 1.93 |
| BBEH | 5.24 | 3.80 | 2.11 | 1.93 |
Prompt Optimization
Prompt optimization automates the revision or selection of instructions and other prompt components to improve model behavior for specific tasks.
See all 141 papers