Canopy: Exploiting Piecewise Smooth Tree Priors for Multi-Fidelity Bandits
Michael Jerge, Suman Jana
cs.LG
Sep 24, 2026 · v1
cs.AI
TL;DR
Key theoretical guarantees (bias certificate, fixed-budget and regret bounds) are mechanized in Lean 4, with sources shipped as supplementary material.
Abstract
Many LLM inference problems, including model routing, prefix-cache management, prompt trimming, and test-time search, can be viewed as optimization over a tree. This structure arises naturally from autoregressive generation: every prefix defines a node, and its continuations form a subtree below it. Internal nodes of the tree provide cheap but biased estimates of a region's value, while leaf evaluations are expensive but accurate. Hierarchical bandit methods can exploit this structure, but typically require a specific smoothness schedule to be specified in advance, even though real objectives are often only piecewise smooth and their optima may lie near sharp boundaries. We introduce CANOPY, a multi-fidelity tree bandit that learns where the smoothness prior is valid rather than assuming it globally. CANOPY uses cheap random-path probes to construct an online certificate of local aggregation bias, then directs expensive leaf evaluations toward cells where the certificate detects a smoothness violation. We prove fixed-budget and regret guarantees whose additional cost is additive in the number of discontinuities, recovering the smooth-tree rate when no violations are present and approaching structure-blind search as violations become dense. Across routing, top-$k$ identification, test-time search, caching, and prompt trimming, CANOPY consistently improves matched-budget performance, including $2.9\times$ higher top-10 recall on a 1000-model pool, $1.6\times$ more SWE-bench Verified issues resolved than best-of-$N$, and $3.6\times$ lower median time-to-first-token with prefix caching.
Problem
Many LLM inference tasks, such as routing, prefix caching, prompt trimming and test-time search, are optimization problems over a tree. Cheap internal-node probes give biased estimates and leaf evaluations are expensive. Hierarchical bandits need a smoothness schedule fixed in advance, yet real objectives are only piecewise smooth.
Approach
CANOPY is a multi-fidelity tree bandit. It uses cheap random-path probes to build an online, data-driven certificate of each cell's aggregation bias. Expensive leaf evaluations are then directed to cells where the certificate detects a smoothness violation. Fixed-budget and regret guarantees are proved, and the main statements are machine-checked in a Lean 4 development provided as supplementary material.
Results
The extra cost in the guarantees is additive in the number of discontinuities: it recovers smooth-tree rates when there are no violations and approaches structure-blind search when violations are dense. Across twelve benchmarks CANOPY improves matched-budget performance, including 2.9x higher top-10 recall on a 1000-model pool, 1.6x more SWE-bench Verified issues resolved than best-of-N, and 3.6x lower median TTFT with prefix caching.
| Benchmark | Metric | Ours | Best baseline |
|---|
| Top-k id. (1000 models), B=600 | recall@10 | .370 | .126 |
| MATH (300, Llama-70B) | accuracy | .617 | .530 |
| SWE-bench Verified (261) | resolved | .318 | .195 |
| vLLM caching (A10G) | TTFT p50 (s) | .267 | .961 (off) |
Selected headline results at matched metric (from Table 1)