✗ Mechanism failed 2026

Two-Solve Robust Expert Router

Implementation & benchmark of arXiv:2608.29560 — Which LLM for Which Work? Budgeted Model Allocation under Uncertain Evaluation

Usefulness7/10
Difficulty6/10
Novelty7/10

Source paper: Which LLM for Which Work? Budgeted Model Allocation under Uncertain Evaluation arXiv:2608.29560 ⓘ · analyzed Sep 1, 2026

AI-generated research hypothesis, automatically tested. Not peer-reviewed.

Idea description

Replace point-estimate expert routing with a nominal allocation and a least-favourable allocation under uncertainty in expert quality. If both allocations agree, use that route confidently; if they disagree, profile or evaluate only the expert-input pairs responsible for the disagreement. The same mechanism can be used offline to assign workloads to LLMs or online to choose among heterogeneous experts under a latency or FLOP budget.

Formulas

$$X^{\star}[\mu]=\arg\max_{x}\sum_{j}\sum_{k\in\mathcal{K}_{j}}w_{j}\mu_{kj}x_{kj}\quad\text{s.t.}\quad N\sum_{j}\sum_{k\in\mathcal{K}_{j}}w_{j}p_{kj}x_{kj}\leq\Gamma,\quad \sum_{k\in\mathcal{K}_{j}}x_{kj}=1\ \forall j.$$
$$M_{kj}=\mu^{\mathrm{ref}}_{kj}+B_{kj}+C^{o}_{kj},\qquad R_{kj}=\mu^{\mathrm{ref}}_{kj}+C^{r}_{kj}.$$
$$\mathcal U=\{R-C\mid C_{kj}\in[0,\eta_j]\ \text{for every }k\in\mathcal K_j\text{ and }j\}.$$
$$\mu^{\mathrm{pess}}_{kj}=R_{kj}-\eta_j,\qquad x^{\mathrm{nom}}=X^{\star}[R],\qquad x^{\mathrm{pess}}=X^{\star}[\mu^{\mathrm{pess}}].$$

Mathematical statement

The paper defines binary allocation variables $x_{kj}\in\{0,1\}$, where $k$ indexes models or experts and $j$ indexes workloads or input clusters, with workload weight $w_j$, quality estimate $\mu_{kj}$, cost $p_{kj}$, and total budget $\Gamma$. The nominal allocation is $X^{\star}[\mu]=\arg\max_x\sum_j\sum_{k\in\mathcal K_j}w_j\mu_{kj}x_{kj}$ subject to $N\sum_j\sum_{k\in\mathcal K_j}w_jp_{kj}x_{kj}\leq\Gamma$ and $\sum_{k\in\mathcal K_j}x_{kj}=1$ for every $j$. The observed randomized score is decomposed as $R_{kj}=\mu^{\mathrm{ref}}_{kj}+C^r_{kj}$, while observational measurements also contain selection and evaluator gaps: $M_{kj}=\mu^{\mathrm{ref}}_{kj}+B_{kj}+C^o_{kj}$. The uncertainty set is generated by evaluator-error tables $C$ satisfying $C_{kj}\in[0,\eta_j]$ for every expert-workload pair, giving $\mathcal U=\{R-C:C_{kj}\in[0,\eta_j]\}$. Here $\eta_j$ is the maximum evaluator error assumed for workload cluster $j$. Adapt the paper's certificate by computing the nominal optimum at $R$ and a least-favourable optimum over $\mu\in\mathcal U$; agreement means the route is invariant over the assumed uncertainty set. For a fixed candidate allocation $x$, its pessimistic score is $\min_{\mu\in\mathcal U}\sum_{j,k}w_j\mu_{kj}x_{kj}=\sum_{j,k}w_j(R_{kj}-\eta_j)x_{kj}$ under the independent interval model, while the adversarial allocation used for certification should be selected to maximize the opportunity for an alternative allocation to beat the nominal one.

Implementation notes

Integrate this at the offline router or model-profile-selection layer, rather than inside every token-level forward pass initially. Partition requests into $J$ workload clusters using embeddings, task IDs, sequence length, or predicted difficulty. For each cluster $j$ and candidate expert or model $k$, collect a quality estimate $R_{kj}$, cost $p_{kj}$, traffic weight $w_j$, and uncertainty radius $\eta_j$; quality can be validation accuracy, reward, pass rate, or negative loss, while cost can combine latency and FLOPs. First solve the paper's 0-1 multiple-choice knapsack with $\mu=R$ to obtain $x^{nom}$. Then solve it again with $\mu=R-\eta_j$ to obtain a pessimistic route. In a stronger certificate, search for an adversarial table in the interval box: for each candidate alternative allocation $y$, assign each uncertain entry to an endpoint that maximizes $\mathrm{score}(y,\mu)-\mathrm{score}(x^{nom},\mu)$, and retain the alternative with the largest margin. For large systems, approximate this by sampling endpoint tables and repeatedly solving the knapsack, or formulate the endpoint choices as a mixed-integer linear program. Freeze routing only when nominal and pessimistic allocations agree and the worst sampled alternative margin is non-positive. Otherwise, send only disputed pairs $(k,j)$ to an evaluation queue, update $R_{kj}$ and $\eta_j$ from new measurements, and repeat. The quantities computed from the paper's mathematics are the budget-constrained solves and interval adversary; quantities estimated empirically are $R$, $\eta$, costs, and cluster frequencies. The first experiment should use a 4-8 expert MoE or a cascade of small and large language models on GSM8K, MMLU subsets, or a synthetic mixture of easy and hard tasks. Compare nominal routing, always-largest-model routing, and uncertainty-aware routing at equal average FLOPs. Measure task accuracy, average cost, number of profiling evaluations, and the fraction of routes certified. Success is higher accuracy than nominal routing at the same cost, or equal accuracy with fewer expert evaluations; a useful secondary signal is that targeted evaluation resolves routing disagreements faster than uniform profiling.

Verification

Mechanism failed

Stage 1 · Toy mechanism gate: Failed ✗

Stage 2 · Mechanism transferred to benchmark: Not tested

Stage 2 · Practical benchmark result: Not run

Methodology: Toy-system gate first; the benchmark stage runs only after a pass. How verification works

Stage 1 — Mechanism check agent confidence 10/10

Built an exact multiple-choice knapsack nominal/pessimistic router and verified the interval math over 160 random instances plus an eta sweep. The stated uncertainty model subtracts the same cluster-dependent constant from every feasible allocation, so nominal and pessimistic routes always agree; the objective shift matched eta times the total workload weight to 8.88e-16 and realized quality was identical. Therefore the disagreement-triggered profiling mechanism produces no benefit under the authoritative formulas; expert-specific uncertainty would be needed for a meaningful extension.

Agent confidence
10/10
Baseline
Always-cheapest routing: weighted true quality 4.84976, cost 6.40577
Idea
Nominal/pessimistic routing: weighted true quality 5.21174, cost 8.39366; 0/160 random-sweep disagreements and 0 disputed clusters across eta=0..0.8
Math predictions
Zero disagreements confirmed; objective shift linear in eta with maximum error 8.88e-16 confirmed; identical realized quality confirmed

Limitations: Synthetic exact-enumeration experiment only; no neural MoE or LLM workload was tested. Expert-specific uncertainty radii, noisy online profiling, latency distributions, and the stronger adversarial alternative-allocation formulation were not implemented.

How to run: python3 two_solve_router.py

Verdict computed by deterministic test code from paired-seed statistics — not by the language model.

Artifacts

Implementation overview ⬇ Download all as ZIP 3 files · code, reports and structured results