Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

Test-Time Compute for Frozen Embedding Models through Agentic Program Search

About

Test-time compute is widely believed to benefit only large reasoning models, leaving small models with nothing to gain. We argue the opposite for dense retrieval, since modern small embedding models are distilled or adapted from large language model backbones and can inherit their latent test-time-compute potential. We ask how much retrieval quality a frozen embedding model gains at inference alone, with no auxiliary model and no parameters trained at deployment. An agentic loop in which a large language model writes programs over a frozen encoder API explores 144 candidates and yields twelve Pareto-optimal programs that trade inference compute for quality across cost ratios from $c{=}1.2$ to $14.7$, every one improving nDCG@10 on all 14 discovery tasks. The programs use no trainable parameters and recover classical retrieval primitives, among them reciprocal rank fusion, the Fisher linear discriminant, Rocchio pseudo-relevance feedback, and sentence-level MaxSim. Applied unmodified to nineteen held-out tasks and three unseen encoder families, a single fixed program improves the majority of tasks, with a positive median $\Delta$nDCG@10 and a 54 to 57% win-rate at $c{\ge}4$, and the gains are largest on encoder families never seen during discovery. A matched-budget learned projection head trained on the same tasks does not transfer this way, improving in-domain retrieval by $+0.20$ to $+0.25$ nDCG@10 yet falling below baseline on every held-out encoder. Small embedding models therefore inherit usable test-time-compute potential, and a frozen encoder converts inference compute into retrieval gains that transfer to new corpora and encoders with no per-domain labels.

Han Xiao• 2026

Related benchmarks

TaskDatasetResultRank
Information RetrievalFIQA BEIR (test)
nDCG@1040.32
44
Information RetrievalArguana BEIR
NDCG@1068.82
33
Information RetrievalSciFact BEIR
NDCG@1072.41
24
Document RetrievalSciFact BEIR
Delta nDCG@100.8
16
Document RetrievalNFCorpus BEIR
Delta nDCG@101.63
15
Document RetrievalFiQA BEIR 2018
Delta nDCG@101.36
15
Information RetrievalNFCorpus Full BEIR
nDCG@1042.14
11
Information RetrievalNFCorpus nanoBEIR slice
nDCG@1042.19
4
Information RetrievalArguAna nanoBEIR slice
nDCG@1072.97
4
Information RetrievalFiQA nanoBEIR 2018 (slice)
nDCG@1064.63
4
Showing 10 of 15 rows

Other info

Follow for update