An important Bottleneck in LLM Optimizers: Principle-Level Local Optima
Organizations: The University of Manchester · UKRI AI Centre for Doctoral Training in Decision Making for Complex Systems · University of Chinese Academy of Sciences · Zhongguancun Academy ·
Abstract
Feedback-driven LLM optimizers are increasingly used for evolutionary search, Bayesian optimization, heuristic design, and scientific discovery. Yet rapid initial gains often give way to prolonged plateaus despite substantial remaining evaluation budgets. We investigate principle-level search collapse, where an optimizer's search becomes confined to a narrow set of design principles, even though it may continue to generate distinct candidates and achieve small improvements. We assemble a diagnostic benchmark spanning molecular optimization, network sentinel-node selection, and GPU-kernel optimization, and introduce a decision-tree protocol combining principle narrowing, prolonged performance plateaus, and gaps to attainable reference solutions. Across 161 assessable runs on the three problem domains, 78.3% receive support for collapse, including 55.3% with strong support. Candidate-level variation can nevertheless conceal restricted search at the principle level. We then test diversity injection, a common response to search stagnation, such as higher t, multiple LLMs, and web search, and show that it provides limited and inconsistent benefits: among 65 molecular runs, still 73.8% and 53.8% receive support and strong support respectively. We finally test reference-principle interventions and find that they improve some runs, but their benefits vary across models and tasks: access to a better principle does not guarantee its effective use. Our benchmark and diagnostic protocol provide a basis for evaluating future optimizers on principle-level collapse. Our findings highlight the need for online collapse detection and better mechanisms for generating and maintaining alternative principles during optimization.
Figures & tables
| GPU kernel | Molecular optimization | Sentinel-node selection | |
|---|---|---|---|
| Candidate | one PyTorch/Triton source file | SMILES string | node subset of size (4 or 5) |
| Utility | (geometric-mean latency) | scalarized five-objective score | (approximation error) |
| Descriptor | executed route, blind LLM annotation against a per-problem rubric (6–9 leaf routes + families) | Bemis–Murcko scaffold | node incidence in selected sets |
| Diagnostic set | fastest 10 valid kernels | top 50 distinct molecules | top 50 distinct node sets |
| Candidate-level diversity | 5-gram Jaccard distance between source files | fingerprint diversity, uniqueness | set uniqueness |
| Reference | FlashInfer library kernel, same GPU | best known molecule, | exhaustive optimum (Dolphin); published optimized set (residence hall) |
Appendix figures & tables41 assets
Supplementary material from the paper’s appendix.
Appendix
| Method | Condition | S | L | N |
|---|---|---|---|---|
| PiEvo ( ) | Sigma 0.9 | 1 | 1 | 3 |
| PiEvo | Original | 2 | 3 | 0 |
| ExLLM(Sonnet 5) | Explore (7:2:1) | 5 | 0 | 0 |
| ExLLM(Sonnet 5) | Original | 3 | 2 | 0 |
| ExLLM(Sonnet 5) | Web search | 3 | 2 | 0 |
| ExLLM(Sonnet 5 Thinking) | Original | 5 | 0 | 0 |
| Dolphin | Residence hall | |||||
|---|---|---|---|---|---|---|
| Method | S | L | N | S | L | N |
| PiEvo | 0 | 0 | 5 | 4 | 1 | 0 |
| ExLLM(Sonnet 5) | 0 | 0 | 5 | 3 | 0 | 2 |
| ExLLM(DeepSeek) | 0 | 0 | 5 | 5 | 0 | 0 |
| ExLLM(Qwen) | 0 | 0 | 5 | 0 | 4 | 1 |
| ExLLM(GPT-5.4) | 0 | 0 | 5 | 1 | 1 | 3 |
| Method / condition | Top-1 | Top-10 | HV (50) | Remaining (%) | |||
|---|---|---|---|---|---|---|---|
| ExLLM(GPT-5.4) | |||||||
| ExLLM(Sonnet 5) | |||||||
| ExLLM(DeepSeek) | |||||||
| ExLLM(Qwen) | |||||||
| PiEvo | |||||||
| MCCE |
| Method / condition | ||||||||||
|---|---|---|---|---|---|---|---|---|---|---|
| ExLLM(GPT-5.4) | ||||||||||
| ExLLM(Sonnet 5) | ||||||||||
| ExLLM(DeepSeek) | ||||||||||
| ExLLM(Qwen) | ||||||||||
| PiEvo | ||||||||||
| MCCE |
| Method / condition | Top-1 error | Top-10 error | Remaining (%) | ||||
|---|---|---|---|---|---|---|---|
| Dolphin | |||||||
| ExLLM(GPT-5.4) | |||||||
| ExLLM(Sonnet 5) | |||||||
| ExLLM(DeepSeek) | |||||||
| ExLLM(Qwen) | |||||||
| PiEvo | |||||||
| Method / condition | ||||||||
|---|---|---|---|---|---|---|---|---|
| Dolphin | ||||||||
| ExLLM(GPT-5.4) | ||||||||
| ExLLM(Sonnet 5) | ||||||||
| ExLLM(DeepSeek) | ||||||||
| ExLLM(Qwen) | ||||||||
| PiEvo | ||||||||
| Method / condition | Top-1 | Top-10 | HV (50) | Remaining (%) | |||
|---|---|---|---|---|---|---|---|
| ExLLM(Baseline) | |||||||
| ExLLM(Explore) | |||||||
| ExLLM(Web search) | |||||||
| ExLLM(Temp 1.3) | |||||||
| Two LLMs | |||||||
| MCCE |
| Method / condition | ||||||||||
|---|---|---|---|---|---|---|---|---|---|---|
| ExLLM(Baseline) | ||||||||||
| ExLLM(Explore) | ||||||||||
| ExLLM(Web search) | ||||||||||
| ExLLM(Temp 1.3) | ||||||||||
| Two LLMs | ||||||||||
| MCCE |
| Method / condition | Top-1 | Top-10 | HV (50) | Remaining (%) | |||
|---|---|---|---|---|---|---|---|
| ExLLM(GPT-5.4) | |||||||
| ExLLM(GPT-5.4) / Explore (7:2:1) | |||||||
| ExLLM(GPT-5.4) | |||||||
| ExLLM(GPT-5.4) / Temperature 1.3 | |||||||
| ExLLM(GPT-5.4) / Web search | |||||||
| ExLLM(Sonnet 5) | |||||||
| Method / condition | Top-1 | Top-10 | Top-10 scaffolds | ||||
|---|---|---|---|---|---|---|---|
| ExLLM(GPT-5.4) | |||||||
| ExLLM(Sonnet 5) | |||||||
| ExLLM(DeepSeek) | |||||||
| ExLLM(Qwen) | |||||||
| ExLLM(GPT-4o) | |||||||
| ExLLM(Sonnet thinking) |
| Method / condition | Updates early | Updates late | Anomalies all | Anomalies early | Anomalies late | ||
|---|---|---|---|---|---|---|---|
| PiEvo | |||||||
| PiEvo / Sigma 0.9 |
| Method / problem | Updates early | Updates late | Anomalies all | Anomalies early | Anomalies late | ||
|---|---|---|---|---|---|---|---|
| PiEvo /dolphin | |||||||
| PiEvo /residence hall |
| Method / condition | Top-1 | Top-10 | Scaffold | |||
|---|---|---|---|---|---|---|
| PiEvo | ||||||
| PiEvo / Sigma 0.9 |
| Method / condition | Top-1 | Top-10 | |||
|---|---|---|---|---|---|
| ExLLM(GPT-5.4) / Hint from 40% | |||||
| ExLLM(GPT-5.4) / Hint from start | |||||
| ExLLM(GPT-5.4) | |||||
| ExLLM(Sonnet 5) / Hint from 40% | |||||
| ExLLM(Sonnet 5) / Hint from start | |||||
| ExLLM(Sonnet 5) |
| Method / condition | Top-1 error | Top-10 error | ||
|---|---|---|---|---|
| ExLLM(GPT-5.4) / Hint from 40% | ||||
| ExLLM(GPT-5.4) | ||||
| ExLLM(Sonnet 5) / Hint from 40% | ||||
| ExLLM(Sonnet 5) | ||||
| ExLLM(DeepSeek) / Hint from 40% | ||||
| ExLLM(DeepSeek) |
| Method / condition | Top-1 error | Top-10 error | ||
|---|---|---|---|---|
| ExLLM(GPT-5.4) / Hint from 40% | ||||
| ExLLM(GPT-5.4) | ||||
| ExLLM(Sonnet 5) / Hint from 40% | ||||
| ExLLM(Sonnet 5) | ||||
| ExLLM(DeepSeek) / Hint from 40% | ||||
| ExLLM(DeepSeek) |
| Task | Collapse | Late plateau | , concentrated | Not concentrated | |
|---|---|---|---|---|---|
| GQA decode | 20 | 5 | 5 | 10 | 0 |
| MLA decode | 20 | 9 | 6 | 5 | 0 |
| MLA prefill | 20 | 19 | 1 | 0 | 0 |
| MoE FP8 | 20 | 15 | 4 | 0 | 1 |
| RMSNorm | 20 | 0 | 0 | 19 | 1 |
| Method | Task | (latency / expert) | Late | Collapse / assessed | |
|---|---|---|---|---|---|
| ExLLM(GPT-5.4) | GQA decode | 5 | 3/5 | ||
| ExLLM(GPT-5.4) | MLA decode | 5 | 4/5 | ||
| ExLLM(GPT-5.4) | MLA prefill | 5 | 5/5 | ||
| ExLLM(GPT-5.4) | MoE FP8 | 5 | 5/5 | ||
| ExLLM(GPT-5.4) | RMSNorm | 5 | 0/5 | ||
| ExLLM(Claude Sonnet 5) | GQA decode | 5 | 0/5 |
| Timing | Hint | Pairs / planned | Better | Worse | Within band | Geom. gain |
|---|---|---|---|---|---|---|
| Start | L1 | 48/48 | 28 | 12 | 8 | 1.169 |
| Start | L2 | 46/48 | 14 | 21 | 11 | 0.833 |
| Start | L3 | 46/48 | 20 | 15 | 11 | 1.067 |
| Onset | L1 | 41/41 | 12 | 10 | 19 | 1.110 |
| Onset | L2 | 41/41 | 13 | 9 | 19 | 1.010 |
| Onset | L3 | 41/41 | 16 | 8 | 17 | 0.986 |
| First kernel of the route vs. the run’s best so far | events | never developed again | median | IQR | events with attempts left | their median |
|---|---|---|---|---|---|---|
| First kernel faster than the incumbent | 73 | 3% | 59 | 15–96 | 71 | 64 |
| Up to 2x slower | 69 | 23% | 5 | 1–28 | 67 | 5 |
| More than 2x slower | 44 | 36% | 1 | 0–4 | 36 | 1.5 |
| All events | 186 | 18% | 9.5 | 1–78.5 | 174 | 14 |
| Problem | Backbone | seed | attempt | best on the route / expert | ||
|---|---|---|---|---|---|---|
| GQA decode | Sonnet 5 | 42 | 22 | 1.35 | 14 | 0.98 |
| GQA decode | Sonnet 5 | 43 | 15 | 3.39 | 37 | 0.77 |
| GQA decode | Sonnet 5 | 44 | 14 | 3.95 | 63 | 0.66 |
| GQA decode | Sonnet 5 | 45 | 19 | 2.99 | 13 | 1.01 |
| GQA decode | Sonnet 5 | 46 | 13 | 1.55 | 18 | 0.79 |
| MLA decode | Sonnet 5 | 43 | 16 | 3.27 | 151 | 0.86 |
| Task | Hint |
|---|---|
| Molecular (L2) | Search around compact heteroaryl-amide chemotypes rather than only refining generic benzamide or para-methoxy aryl-amide families. Favor low-synthetic-complexity molecules with a small sulfur-containing or heteroaromatic ring, a simple aryl/benzyl partner, and light substituent tuning to keep QED high while driving the unwanted target probabilities near zero. |
| Dolphin (L2) | A direction is a heterogeneous degree profile: include one low-degree or leaf-like sentinel and combine it with low-to-mid and moderate/high-role nodes whose errors complement each other as a set. |
| Residence hall (L2) | Look for five sentinels that jointly cover different network roles across degree and neighborhood structure. Try heterogeneous complementarity, try a mixed low/mid/moderately-high profile and test replacements that preserve that role coverage. |
| Kernel, all problems (L1) | Optimization trajectories on this problem typically settle early on one overall design approach whose latency stays far above what the hardware allows, and further refinement within that approach does not close the gap. Whenever the implementations shown share the same overall algorithmic decomposition, do not keep tuning it; propose a candidate whose decomposition of the computation is fundamentally different. |
| GQA / MLA paged decode (L2) | Instead of one program per (sequence, head) that walks the whole KV sequence serially, split each sequence’s KV range into several chunks processed by different programs in parallel, each producing partial attention statistics (running max, normalizer, accumulated output), and merge the chunks afterwards with log-sum-exp rescaling. Pack the heads that share the same KV as the row dimension of a matrix product so that scores and outputs run on tensor cores and each KV block is read once. |
| MLA paged prefill (L2) | Treat the computation as a two-dimensional tiling—a block of query tokens (with all heads, which share the same compressed KV) against a block of KV tokens—so that score computation and output accumulation are matrix products on tensor cores, with an online softmax carried across KV blocks and the causal mask applied only where a block crosses the diagonal. Long sequences can additionally be split along the KV dimension across programs and merged with log-sum-exp rescaling. Avoid host-side loops over sequences. |
Large Language Model-Guided Optimization
Large Language Model-Guided Optimization uses language models to propose, steer, and refine solutions to hard search and design problems.
See all 608 papers