mlsanityData to 14 Oct 2026

AI4AI at Test-Time: Strong-to-Weak Capability Transfer via Harnesses

arXiv 2608.12307 ↗First posted 12 Aug 2026cs.LG cs.AI cs.CLcited by 7

On the map: Knowledge distillation → On-policy and self-distillation for LLMs

Abstract (arXiv)

Recent work on distillation transfers the capabilities of large models to smaller ones often by updating the latter's parameters, through teacher forcing, on-policy distillation, and related training-time methods. In this paper, we ask whether such transfer can instead occur at test time. We study strong-to-weak scaffolding: whether a stronger builder model can construct inference-time harnesses that help a weaker target model solve tasks more reliably without any parameter updates. Using four representative Theory-of-Mind benchmarks, each builder model uses 5% of the data as a validation set to iteratively refine its harness over multiple rounds, after which the finalized harness is evaluated on the full test set. Empirically, this form of test-time capability transfer is highly effective, nearly doubling average target-model performance from 0.49 to 0.91. Our analysis shows that the gains come primarily from offloading unstable model reasoning into deterministic code, benchmark-specific routing, and strict answer-format enforcement, rather than from encouraging the target model to reason more extensively or sample more broadly. We further find that builder-model reasoning effort improves harness quality monotonically, platform effects are modest relative to the builder model's own capability, and weaker target models receive the largest gains. These results suggest that inference-time harness design is an important complement to conventional training-time distillation, enabling strong models to transfer cognitive structure to weaker models without retraining.

Key facts. Cited by 7 papers in the map (data through 14 Oct 2026). Placed in On-policy and self-distillation for LLMs, one of 12 directions in Knowledge distillation. Builds on 36 of its 39 references that are in the map. First posted on arXiv 12 Aug 2026.

Builds on

36 of 39 references.

References · 36

  1. 2022Training language models to follow instructions with human feedbackcited by 15,971
  2. 2022Chain-of-Thought Prompting Elicits Reasoning in Large Language Modelscited by 12,459
  3. 2015Distilling the Knowledge in a Neural Networkcited by 10,324
  4. 2022ReAct: Synergizing Reasoning and Acting in Language Modelscited by 8,695
  5. 2022Self-Consistency Improves Chain of Thought Reasoning in Language Modelscited by 5,708
  6. 2023Toolformer: Language Models Can Teach Themselves to Use Toolscited by 4,182
  7. 2023Self-Refine: Iterative Refinement with Self-Feedbackcited by 3,442
  8. 2023Tree of Thoughts: Deliberate Problem Solving with Large Language Modelscited by 3,211
  9. 2023A Survey on Large Language Model based Autonomous Agentscited by 2,242
  10. 2024SWE-agent: Agent-Computer Interfaces Enable Automated Software Engineeringcited by 1,737
  11. 2022Least-to-Most Prompting Enables Complex Reasoning in Large Language Modelscited by 1,407
  12. 2022Program of Thoughts Prompting: Disentangling Computation from Reasoning for Numerical Reasoning Taskscited by 1,076
24 more
  1. 2023On-Policy Distillation of Language Models: Learning from Self-Generated Mistakescited by 939
  2. 2023Graph of Thoughts: Solving Elaborate Problems with Large Language Modelscited by 886
  3. 2023DSPy: Compiling Declarative Language Model Calls into Self-Improving Pipelinescited by 832
  4. 2023Distilling Step-by-Step! Outperforming Larger Language Models with Less Training Data and Smaller Model Sizescited by 737
  5. 2022PAL: Program-aided Language Modelscited by 652
  6. 2022Decomposed Prompting: A Modular Approach for Solving Complex Taskscited by 558
  7. 2023Weak-to-Strong Generalization: Eliciting Strong Capabilities With Weak Supervisioncited by 467
  8. 2024Automated Design of Agentic Systemscited by 407
  9. 2026Meta-Harness: End-to-End Optimization of Model Harnessescited by 309
  10. 2023Large Language Models Fail on Trivial Alterations to Theory-of-Mind Taskscited by 292
  11. 2023Faithful Chain-of-Thought Reasoningcited by 288
  12. 2023Understanding Social Reasoning in Language Models with Language Modelscited by 172
  13. 2025A Survey of Frontiers in LLM Reasoning: Inference Scaling, Learning to Reason, and Agentic Systemscited by 111
  14. 2024MMToM-QA: Multimodal Theory of Mind Question Answeringcited by 85
  15. 2024ToMBench: Benchmarking Theory of Mind in Large Language Modelscited by 72
  16. 2023HI-TOM: A Benchmark for Evaluating Higher-Order Theory of Mind Reasoning in Large Language Modelscited by 63
  17. 2026Harness-Bench: Measuring Harness Effects across Models in Realistic Agent Workflowscited by 57
  18. 2024MuMA-ToM: Multi-modal Multi-Agent Theory of Mindcited by 51
  19. 2026Code as Agent Harnesscited by 47
  20. 2024Position: Theory of Mind Benchmarks are Broken for Large Language Modelscited by 31
  21. 2025Theory of Mind in Large Language Models: Assessment and Enhancementcited by 25
  22. 2026From Static Templates to Dynamic Runtime Graphs: A Survey of Workflow Optimization for LLM Agentscited by 24
  23. 2025AutoToM: Scaling Model-based Mental Inference via Automated Agent Modelingcited by 23
  24. 2026UserHarness: Harnessing User Minds for Stronger Agent Theory-of-Mindcited by 1

Built on by

Cited by year (2026 to 14 Oct)

Citing papers per year: 2026: 7.

7 papers.

Similar, not cited

Close in wording, no citation either way

  1. 2026Towards Understanding On-Policy Distillation through the Lens of Test-Time Scaling
  2. 2025A Survey of Test-Time Compute: From Intuitive Inference to Deliberate Reasoning
  3. 2026CapField-OPD: Learning Continuous Capability Fields via Joint-Anchored Multi-Teacher On-Policy Distillation for Flow Models
  4. 2026Weak-to-Strong On-Policy Distillation
  5. 2026How Much Can Language Models Gain from Test-Time Computation?
  6. 2026A Survey on Self-Improving Test-Time Intelligence: Feedback-Driven Adapting, Learning, and Scaling at Inference
  7. 2026ReAD: Reinforcement-Guided Capability Distillation for Large Language Models
  8. 2025Revisiting the Test-Time Scaling of o1-like Models: Do they Truly Possess Test-Time Scaling Capabilities?