AdaGEPA: Adaptive Feedback Allocation for Reflective Prompt Optimization
Organizations: The Chinese University of Hong Kong, Shenzhen · Shenzhen Loop Area Institute
Abstract
Prompt optimization improves the performance of language-model systems on downstream tasks by refining their prompts. Classical methods evaluate prompts on task examples and use the resulting feedback to guide prompt revisions through reflection. However, when feedback selection does not account for the prompt's weaknesses, these revisions may improve performance on selected examples without yielding broader task improvements. To address this issue, we propose AdaGEPA, an adaptive feedback-allocation method that uses the prompt's performance and task structure to select examples for the next prompt revision. Our method replaces at most one example in each feedback minibatch to target an identified weakness while preserving the remaining feedback context. Across our main experiments on six downstream benchmarks, AdaGEPA achieves higher mean validation scores than non-adaptive feedback selection under matched rollout budgets. AdaGEPA also finds high-performing prompts earlier across several tasks. In the initial Schema-Guided Dialogue (SGD) study, its half-budget prompts outperform the non-adaptive baseline's full-budget prompts in joint goal accuracy on new dialogues from services seen and unseen during search. Overall, our findings highlight the potential of adaptive feedback allocation to improve both the effectiveness and rollout-budget efficiency of reflective prompt optimization.
Figures & tables
| Earlier checkpoint | Endpoint | ||||||||
| Task / study | Metric | GEPA default | AdaGEPA ( ) | GEPA default | AdaGEPA ( ) | Gain-AUC | |||
| SGD initial | Balanced DST | 5 | 560 | 51.23 | 55.21 | 1120 | 53.53 | 55.99 | |
| JGA | 23.65 | 28.46 | 26.15 | 29.42 | |||||
| SGD extension | Balanced DST | 5 | 560 | 51.80 | 53.59 | 1120 | 55.59 | 55.18 | |
| JGA | 23.65 | 26.54 | 27.50 | 28.27 | |||||
| SGD replication | Balanced DST | 10 | 560 | 52.17 | 53.74 | 1120 | 55.35 | 55.33 | |
| Earlier checkpoint | Endpoint | ||||||||
| Task / study | Panel | Metric | GEPA default | AdaGEPA ( ) | GEPA default | AdaGEPA ( ) | |||
| SGD initial | In-domain | Balanced DST | 5 | 560 | 50.04 | 56.95 | 1120 | 51.47 | 56.85 |
| JGA | 22.31 | 31.15 | 23.46 | 31.73 | |||||
| Unseen service | Balanced DST | 56.29 | 61.03 | 57.33 | 59.69 | ||||
| JGA | 28.08 | 34.04 | 28.85 | 32.31 | |||||
| SGD extension | In-domain | Balanced DST | 5 | 560 | 51.24 | 53.85 | 1120 | 55.27 | 55.40 |
Appendix figures & tables38 assets
Supplementary material from the paper’s appendix.
Appendix
| Scope | Proposals | Positive | Zero | Negative | Positive rate |
|---|---|---|---|---|---|
| AIME | 142 | 77 | 16 | 49 | 54.23% |
| LiveBench-Math | 226 | 111 | 28 | 87 | 49.12% |
| IFBench | 90 | 45 | 10 | 35 | 50.00% |
| All proposals | 458 | 233 | 54 | 171 | 50.87% |
| GEPA default subset | 150 | 85 | 14 | 51 | 56.67% |
| Task | Experiment | Repeats |
|---|---|---|
| SGD | Ordering formation sets | 5 / 10 searches |
| SGD | Initial study (5 formation searches) | 5 paired runs |
| SGD | Extension study (5 formation searches) | 5 paired runs |
| SGD | Replication study (10 formation searches) | 10 paired runs |
| SGD | History-and-coverage and formation controls | 5 paired runs |
| SGD | Random Legal three-arm control | 10 paired runs |
| Benchmark | Task-specific adapter |
|---|---|
| SGD | Hierarchical service–component adapter |
| MultiWOZ | Service-aware routing adapter |
| HoVer | Stage-progress replay adapter |
| IFBench | Mean-deficit adapter |
| PUPA | Observed-deficit replay adapter |
| TACO | Coarse-skill adapter |
| Task | Evidence used | Proposal rule | Question examined |
|---|---|---|---|
| SGD | Search progress, service exposure, static task profiles, and selected-parent component deficits. | Frozen models rank legal service-level proposals and, when supported, refine the choice within the selected service. | Component-aware proposal ordering and search-external candidate quality. |
| MultiWOZ | Selected-parent component deficits, service exposure, and frozen service metadata. | A frozen service-aware rule ranks legal service-level proposals using search context, intervention exposure, and candidate-wide DST features, then selects an eligible occurrence within the chosen service. | Service-aware routing versus generic coverage and fitted within-service occurrence ordering. |
| HoVer | Intermediate retrieval progress, unresolved final outcomes, and replay history. | A fixed replay rule prioritizes an eligible occurrence whose trajectory made retrieval progress but did not resolve the task. | Stage-progress-guided feedback reuse. |
| AIME | Failure history, repeated exposure, and coarse topic clusters. | The bounded rule applies the history-and-coverage signal to at most one supported replacement. | History-and-coverage allocation under a bounded one-slot intervention. |
| IFBench | Selected-parent verifier deficits and prior exposure under overlapping objectives. | The fixed mean-deficit rule prioritizes supported examples from weak verifier groups. | Feedback allocation under overlapping verifier objectives. |
| PUPA | Update type, prior feedback observations, and visible privacy-leakage signals. | On privacy-facing updates, a fixed rule reuses eligible feedback with previously observed privacy leakage. Other update types retain the default minibatch. | Feedback reuse and downstream candidate outcomes under coupled objectives. |
| Component | Specification |
|---|---|
| Formation evidence | The reported configurations use five or ten designated development searches under the default feedback rule to collect proposal states and one-step targets. |
| Ordering target | The target is the observed one-step Balanced-DST frontier gain after a complete proposal transition. Rejected or non-advancing proposals receive zero. |
| Frozen models | Two ridge models rank legal service-level proposals and eligible within-service refinements using search progress, service exposure, static task profiles, and selected-parent deficits. |
| Runtime selection | The service-level model ranks legal service-level proposals, and the refinement model ranks eligible occurrences within the selected service. A distinct refinement is used when available; otherwise the service-level proposal is used, with the default minibatch as the final fallback. |
| Control | Preserved structure | Changed factor | Purpose |
|---|---|---|---|
| SGD Random Legal | One-slot capacity, the legal replacement pool, and the search schedule | Samples uniformly from legal replacements without task-aware ordering | Isolate the value of legal replacement alone. |
| SGD Structured Unordered | AdaGEPA’s supported proposal set and one-slot capacity | Selects from the same supported proposal set without fitted occurrence ordering | Separate structured replacement support from fitted ordering. |
| SGD service-level-only | One-slot capacity, service-level targeting, and the search transition | Omits within-service component refinement | Isolate component-level targeting beyond service selection. |
| MultiWOZ coverage-only | One-slot capacity and prior exposure information | Selects unused examples without task-specific service routing | Compare generic unused-example coverage with service-aware routing. |
| MultiWOZ fitted ordering | One-slot capacity and service-aware proposal routing | Ranks supported occurrences within the selected service | Measure the value of fitted ordering beyond rank-neutral service-aware routing. |
| AIME bounded one-slot | The history-and-coverage signal and surrounding search transition | Applies the same signal to at most one replacement instead of resampling the full minibatch | Separate selection direction from intervention capacity. |
| Panel | Source states | Actions compared | Repeats |
|---|---|---|---|
| Four-arm comparison | 15 states from five development runs | GEPA default, Random Legal, Structured Unordered, and AdaGEPA (5 formation runs) | 3 per distinct action |
| Ordering comparison | 15 states from a separate set of five development runs | Structured Unordered and AdaGEPA (10 formation runs) | 3 per distinct action |
| Task | Rollout cap | Development-validation size | Key reported checkpoints |
|---|---|---|---|
| SGD | 1120 | 104 | 104, 392, 560, 840, 1120 |
| MultiWOZ | 1120 | 104 | 104, 300, 392, 560, 840, 1120 |
| HoVer | 1120 | 300 | 300, 624, 1120 |
| AIME | 560 | 45 | 45, 150, 300, 336, 392, 560 |
| LiveBench-Math | 560 | 50 | 50, 150, 300, 560 |
| IFBench | 560 | 54 | 54, 300, 392, 560 |
| Task | Development evaluation | Search-external evaluation |
|---|---|---|
| SGD | Search validation | In-domain (104); unseen-service (104) |
| MultiWOZ | Search validation | Search-external panel (104; official-test split) |
| HoVer | Search validation | Search-external panel (253); full evaluation panel (300) secondary |
| AIME | Held-out development split | — |
| LiveBench-Math | Search validation | — |
| IFBench | Search validation | — |
| Task | Metric | Definition and aggregation |
|---|---|---|
| SGD / MultiWOZ | Balanced DST; JGA | Balanced DST averages a per-example score with weight on exact joint-state correctness and each on active-intent correctness, requested-slot F1, and slot-value F1. Checkpoint candidates are selected by Balanced DST. JGA is the fraction of examples with a completely correct dialogue state; both metrics are reported for the same frozen candidate. |
| HoVer | Document Exact | Fraction of examples for which the cumulative retrieval after the final hop covers all three normalized gold supporting-document titles. |
| AIME | Formatted-answer accuracy | Fraction of responses containing the formatted reference answer (‘ ### <integer> ’); the number of solved problems is also reported. |
| LiveBench-Math | Task-specific score | Mean score assigned by the pinned task-specific deterministic scorers; the number of full-credit problems is also reported. |
| IFBench | Strict pass rate | Mean of binary item scores, where an item receives one only when the response passes every instruction-specific verifier. |
| PUPA | Privacy–quality score | Let indicate that the generated response is judged at least as good as the reference, and let indicate the reverse comparison. The per-example quality score is , with as a stricter secondary measure. If is the fraction of unique private-information entries leaked into the request sent to the external model, privacy is . The overall score is , averaged across validation examples. |
| Task | Task-LM or task evaluator | Reflection-LM |
|---|---|---|
| SGD, MultiWOZ | openai/gpt-4.1-mini ; , , , | openai/gpt-5.1 ; , , , |
| HoVer | openai/gpt-4.1-mini ; , , , | openai/gpt-5.1 ; , , , |
| AIME | openai/gpt-4.1-mini ; , , , | openai/gpt-5.1 ; , , , |
| LiveBench-Math | openai/gpt-4.1-mini ; , , , | openai/gpt-5.1 ; , , , |
| IFBench | openai/gpt-4.1-mini ; , , , | openai/gpt-5.1 ; , , , |
| PUPA | openai/gpt-4.1-mini ; task , evaluator ; , | openai/gpt-5.1 ; , , , |
| JGA (%) | Balanced DST (%) | |||||||
|---|---|---|---|---|---|---|---|---|
| Arm | ||||||||
| GEPA default | 27.88 | 28.08 | 28.27 | 28.85 | 55.81 | 56.29 | 56.14 | 57.33 |
| AdaGEPA | 31.92 | 34.04 | 31.92 | 32.31 | 58.28 | 61.03 | 59.35 | 59.69 |
| Evaluation | Contrast and metric | Mean | W/T/L | 95% paired-seed interval |
|---|---|---|---|---|
| Development | Gain-AUC, Balanced DST | 3/0/2 | ||
| Development | vs. , Balanced DST | 4/0/1 | ||
| Development | vs. , Balanced DST | 3/0/2 | ||
| In-domain panel | vs. , JGA | 3/0/2 | ||
| In-domain panel | vs. , JGA | 2/1/2 | ||
| In-domain panel | vs. default , JGA | 2/0/3 |
| Evaluation | Contrast and metric | Mean | Median | W/T/L | 95% paired-seed interval |
|---|---|---|---|---|---|
| Development | Gain-AUC | 5/0/5 | |||
| Development | vs. , Balanced DST | 6/0/4 | |||
| Development | vs. , Balanced DST | 5/0/5 | |||
| Development | vs. default , Balanced DST | 4/0/6 | |||
| In-domain panel | vs. , JGA | 7/0/3 | |||
| In-domain panel | vs. , JGA | 7/0/3 |
| Balanced DST | JGA | ||||
|---|---|---|---|---|---|
| SGD study | Run | ||||
| Initial | 1 2 3 4 5 | ||||
| Extension | 1 2 3 4 5 | ||||
| Replication | 1 2 3 4 5 6 7 8 9 10 | ||||
| Evaluation | Formation prompt | AdaGEPA | Difference | 95% interval |
|---|---|---|---|---|
| Unseen-service JGA | 36.54 | 32.69 |
| Balanced DST | JGA | |||
|---|---|---|---|---|
| Run | ||||
| 1 | ||||
| 2 | ||||
| 3 | ||||
| 4 | ||||
| 5 | ||||
| Run | GEPA default | AdaGEPA | Difference | Replays |
|---|---|---|---|---|
| 1 | 43.00 | 43.00 | 1 | |
| 2 | 43.33 | 43.67 | 2 | |
| 3 | 43.67 | 42.00 | 2 | |
| 4 | 38.33 | 49.67 | 4 | |
| 5 | 44.00 | 46.33 | 3 | |
| Mean | 42.47 | 44.93 | 2.4 |
| Run | ||
|---|---|---|
| 1 | ||
| 2 | ||
| 3 | ||
| 4 | ||
| 5 |
| Arm | Gain-AUC | ||||
|---|---|---|---|---|---|
| GEPA default | 51.12 | 53.87 | 55.95 | 56.30 | 9.44 |
| History-and-coverage | 52.41 | 53.80 | 53.80 | 53.96 | 8.99 |
| Contrast | Gain-AUC / W/T/L | Endpoint / W/T/L |
| One-slot replacement and ordering | ||
| AdaGEPA GEPA default | / | / |
| Structured Unordered GEPA default | / | / |
| AdaGEPA Structured Unordered | / | / |
| AdaGEPA Random Legal | / | / |
| Random Legal GEPA default | / | / |
| Arm | ||
|---|---|---|
| GEPA default | 28.08 | 28.85 |
| Service-level-only | 29.23 | 31.92 |
| AdaGEPA | 34.04 | 32.31 |
| Contrast | Mean | Median | W/T/L | 95% paired-seed interval |
|---|---|---|---|---|
| AdaGEPA Structured Unordered | 2/0/3 | |||
| Structured Unordered Random Legal | 2/0/3 | |||
| Random Legal GEPA default | 4/0/1 | |||
| AdaGEPA GEPA default | 3/0/2 |
| Source run | AdaGEPA | Structured Unordered | Difference |
|---|---|---|---|
| 1 | 0.000 | 0.144 | |
| 2 | 0.000 | 0.000 | 0.000 |
| 3 | 1.739 | 0.167 | |
| 4 | 0.413 | 0.132 | |
| 5 | 0.958 | 0.000 | |
| Mean | 0.622 | 0.089 |
| Arm | Endpoint | Gain-AUC |
|---|---|---|
| GEPA default | 53.59 | 11.46 |
| Coverage-only | 53.24 | 11.39 |
| Service-aware rank-neutral | 55.39 | 12.94 |
| Service-aware + fitted ordering | 52.56 | 11.68 |
| Method | Gain-AUC | endpoint |
|---|---|---|
| GEPA default | 7.33 | 20.4 |
| History-and-coverage, full minibatch | 5.45 | 18.8 |
| History-and-coverage, one slot | 6.43 | 20.0 |
| APEX-style, full minibatch | 5.54 | 18.6 |
| Arm | Endpoint | Gain-AUC | Targeted repl. | ||
|---|---|---|---|---|---|
| GEPA default | 48.89 | – | 2.92 | – | – |
| Mean-deficit | 51.85 | +2.96 | 4.15 | +1.23 | 8 |
| All-pass-risk | 49.63 | +0.74 | 3.81 | +0.89 | 39 |
| Run | GEPA default | Coverage-only | Replay | Replay default |
|---|---|---|---|---|
| 1 | 80.80 | 81.13 | 85.10 | |
| 2 | 86.94 | 86.75 | 84.59 | |
| 3 | 85.10 | 85.14 | 86.64 | |
| 4 | 83.64 | 86.40 | 90.75 | |
| 5 | 86.70 | 86.64 | 82.39 | |
| Mean | 84.64 | 85.21 | 85.90 |
| Task | Arm | Dec. | Nom. | Sup. | Act | Refl. | Local eval | Local pass | Full eval |
|---|---|---|---|---|---|---|---|---|---|
| SGD | Service–component | 92 | 92 | 92 | 92 | 92 | 92 | 90 | 90 |
| SGD | Random Legal | 95 | 95 | 95 | 95 | 95 | 95 | 90 | 90 |
| SGD | default | 92 | – | – | – | 92 | 92 | 90 | 90 |
| MultiWOZ | default | 47 | – | – | – | 46 | 46 | 45 | 45 |
| MultiWOZ | Service-aware routing | 47 | 47 | 47 | 47 | 47 | 47 | 45 | 45 |
| HoVer | default | 24 | – | – | – | 23 | 23 | 10 | 10 |
| Task | Arm | Full eval | Parent gain | Frontier | Lineage | Both |
|---|---|---|---|---|---|---|
| SGD | Service–component | 90 | 60 | 26 | 19 | 17 |
| SGD | Random Legal | 90 | 60 | 30 | 22 | 18 |
| SGD | default | 90 | 66 | 32 | 25 | 21 |
| MultiWOZ | default | 45 | 31 | 19 | 16 | 12 |
| MultiWOZ | Service-aware routing | 45 | 31 | 20 | 19 | 17 |
| HoVer | default | 10 | 9 | 9 | 6 | 6 |
| Method | Gain-AUC | |||
|---|---|---|---|---|
| GEPA default | 37.6 | 39.4 | 40.7 | 4.877 |
| Cluster-aware | 38.0 | 40.2 | 41.2 | 6.274 |
| Task / intervention | Evaluation setting | Task metric | Summary relative to the GEPA default |
|---|---|---|---|
| SGD / one-slot | Three paired studies on frozen in-domain and unseen-service panels | JGA | At , in-domain differences were , , and points across the initial, extension, and replication studies; unseen-service differences were , , and points. |
| MultiWOZ / one-slot | Development search and a frozen official-test panel | Balanced DST | Development endpoint points; search-external at . |
| HoVer / one-slot | Development search and frozen search-external panel | Document Exact | Development endpoint points and search-external endpoint points at . |
| AIME / full and one-slot | Five paired optimization runs | Correct answers out of 45 | The full-minibatch rule trailed the GEPA default by correct answers. Its bounded one-slot variant reduced this gap to answers, while its Gain-AUC remained points lower. |
| LiveBench-Math / full-minibatch | Ten paired optimization runs | Correct answers out of 50 | The cluster-aware sampler led by correct answers at , compared with at . |
| IFBench / one-slot | Five paired optimization runs | Truncation-robust strict endpoint | Mean-deficit targeting percentage points. |
| Task | Allocation signal | Observed search behavior |
|---|---|---|
| SGD | Hierarchical service and component deficits | Structured replacement improves early discovery, while the advantage may narrow as the default schedule reaches more feedback. |
| MultiWOZ | Service-aware routing | Selecting the relevant service helps, while fitted within-service ordering does not preserve the same gain. |
| HoVer | Intermediate retrieval-stage progress | Replay opportunities are sparse, but some replay-induced candidates persist into winner lineages. |
| AIME | Coarse clusters and failure history | Additional parent improvements do not consistently accumulate as frontier gains. |
| LiveBench-Math | Coarse mathematical-topic clusters and loss gaps | The allocation advantage narrows as the default schedule approaches complete coverage of the small feedback pool. |
| IFBench | Overlapping verifier deficits | A shared prompt update can redistribute gains across verifier objectives. |
| Task / study | Successful requests | Tokens (M) | Configured USD | ||
|---|---|---|---|---|---|
| SGD initial | 1120 | 5 | 1107.2 / 1105.8 | 5.102 / 4.532 | 2.242 / 1.974 |
| SGD extension | 1120 | 5 | 1110.4 / 1103.6 | 4.731 / 4.195 | 2.085 / 1.828 |
| SGD replication | 1120 | 10 | 1103.7 / 1104.4 | 4.728 / 4.590 | 2.059 / 2.005 |
| MultiWOZ | 1120 | 5 | 1103.0 / 1104.4 | 4.723 / 5.072 | 2.050 / 2.209 |
| HoVer | 1120 | 5 | 3775.8 / 3715.0 | 3.589 / 3.436 | 1.645 / 1.537 |
| IFBench | 560 | 5 | 550.2 / 549.0 | 1.637 / 1.573 | 1.349 / 1.414 |
| SGD study | Usage pairs | Successful requests | Tokens (M) | Configured USD |
|---|---|---|---|---|
| Initial | 5/5 | 550.8 / 1105.8 | 1.869 / 4.532 | 0.826 / 1.974 |
| Extension | 5/5 | 551.4 / 1103.6 | 1.728 / 4.195 | 0.758 / 1.828 |
| Replication | 9/10 | 548.0 / 1104.6 | 1.807 / 4.627 | 0.783 / 2.022 |
| Task / study | Requests | Tokens (M) | Configured USD |
|---|---|---|---|
| SGD initial | 4,264 | 19.579 | 7.136 |
| SGD extension | 5,824 | 21.461 | 7.902 |
| SGD replication, in-domain | 3,536 | 15.679 | 5.703 |
| SGD replication, unseen services | 3,536 | 15.607 | 5.692 |
| MultiWOZ, and | 1,872 | 13.099 | 4.705 |
| HoVer, and | 12,000 | 17.164 | 6.890 |
Automatic Prompt Optimization
Automatic prompt optimization uses model or task feedback to revise instructions for stronger, more reliable language-model performance.
See all 103 papers