mlsanityData to 13 Oct 2026

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

arXiv 2605.11374 ↗First posted 12 May 2026cs.LG cs.CL cs.IRcited by 0

On the map: LLM reasoning → Knowledge- and tool-augmented reasoning

also in 1 more
Abstract (arXiv)

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 DeltanDCG@10 and a 54 to 57% win-rate at c≥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.

Key facts. No citing papers in the map yet (data through 13 Oct 2026). Placed in Knowledge- and tool-augmented reasoning, one of 17 directions in LLM reasoning. Builds on 19 of its 38 references that are in the map. First posted on arXiv 12 May 2026.

Builds on

19 of 38 references.

References · 19

  1. 2024Scaling LLM Test-Time Compute Optimally can be More Effective than Scaling Model Parameterscited by 1,954
  2. 2025Qwen3 Embedding: Advancing Text Embedding and Reranking Through Foundation Modelscited by 1,222
  3. 2024Large Language Monkeys: Scaling Inference Compute with Repeated Samplingcited by 970
  4. 2025AlphaEvolve: A coding agent for scientific and algorithmic discoverycited by 819
  5. 2024NV-Embed: Improved Techniques for Training LLMs as Generalist Embedding Modelscited by 467
  6. 2023Improving Text Embeddings with Large Language Modelscited by 297
  7. 2024Inference Scaling Laws: An Empirical Analysis of Compute-Optimal Inference for Problem-Solving with Language Modelscited by 286
  8. 2023Fine-Tuning LLaMA for Multi-Stage Text Retrievalcited by 271
  9. 2024Generative Representational Instruction Tuningcited by 229
  10. 2025EmbeddingGemma: Powerful and Lightweight Text Representationscited by 132
  11. 2024Making Text Embedders Few-Shot Learnerscited by 79
  12. 2025MetaEmbed: Scaling Multimodal Retrieval at Test-Time with Flexible Late Interactioncited by 36
7 more
  1. 2026jina-embeddings-v5-text: Task-Targeted Embedding Distillationcited by 21
  2. 2021Improving Query Representations for Dense Retrieval with Pseudo Relevance Feedback: A Reproducibility Studycited by 13
  3. 2025Guided Query Refinement: Multimodal Hybrid Retrieval with Test-Time Optimizationcited by 10
  4. 2023Generative and Pseudo-Relevant Feedback for Sparse, Dense and Learned Sparse Retrievalcited by 9
  5. 2025Pseudo Relevance Feedback is Enough to Close the Gap Between Small and Large Dense Retrieval Modelscited by 4
  6. 2025Generalized Pseudo-Relevance Feedbackcited by 4
  7. 2025LLM-VPRF: Large Language Model Based Vector Pseudo Relevance Feedbackcited by 2

Built on by

Not cited yet.

Similar, not cited

Close in wording, no citation either way

  1. 2025Soft Reasoning: Navigating Solution Spaces in Large Language Models through Controlled Embedding Exploration
  2. 2026Compute Allocation for Reasoning-Intensive Retrieval Agents
  3. 2026Navigation-Informed Embeddings: Dense-Retriever Adaptation from Agent Search Traces
  4. 2024RAR-b: Reasoning as Retrieval Benchmark
  5. 2026EvoEmbedding: Evolvable Representations for Long-Context Retrieval and Agentic Memory
  6. 2025Exploring Reasoning-Infused Text Embedding with Large Language Models for Zero-Shot Dense Retrieval
  7. 2026MemRetriever: Learning to Search, Reflect, and Retrieve from Long-Term Memory ↗
  8. 2026Critic-R: Improving Agentic Search using Instruction-tuned Retrievers with Natural Language Introspective Feedback