← All papers
First page of Canopy: Exploiting Piecewise Smooth Tree Priors for Multi-Fidelity Bandits

Canopy: Exploiting Piecewise Smooth Tree Priors for Multi-Fidelity Bandits

Michael Jerge, Suman Jana

cs.LG Sep 24, 2026 · v1 cs.AI
Key theoretical guarantees (bias certificate, fixed-budget and regret bounds) are mechanized in Lean 4, with sources shipped as supplementary material.
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.

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.

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.

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.

BenchmarkMetricOursBest baseline
Top-k id. (1000 models), B=600recall@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)