HyperAIHyperAI

Command Palette

Search for a command to run...

2時間前
LLM

テスト時におけるAI4AI:ハーネスを介した強から弱への能力転移

Cheng Qian Wenting Zhao Liangwei Yang Heng Wang Jielin Qiu Heng Ji Silvio Savarese Huan Wang Shelby Heinecke

概要

近年の蒸留に関する研究は、教師強制、オンポリシー蒸留、および関連する訓練時手法を通じて、多くの場合、小規模モデルのパラメータを更新することにより、大規模モデルの能力を小規模モデルへと転移させている。本論文では、そのような転移がテスト時に起こり得るかどうかを問う。我々は、強から弱へのスキャフォールディング、すなわち、より強力なビルダーモデルが、パラメータを一切更新することなく、より弱いターゲットモデルがタスクをより確実に解決するのを支援する推論時ハーネスを構築できるかどうかを研究する。4つの代表的な心の理論ベンチマークを用い、各ビルダーモデルはデータの5%を検証セットとして使用し、複数ラウンドにわたってハーネスを反復的に改良し、最終化されたハーネスを全テストセットで評価する。経験的に、この形式のテスト時能力転移は非常に効果的であり、ターゲットモデルの平均性能を0.49から0.91へとほぼ倍増させる。我々の分析によれば、この向上は主に、不安定なモデルの推論を決定論的コードへとオフロードすること、ベンチマーク固有のルーティング、そして厳格な回答形式の強制に起因しており、ターゲットモデルがより広範に推論したり、より広くサンプリングしたりすることを促した結果ではないことが示された。さらに、ビルダーモデルの推論努力がハーネスの品質を単調に向上させること、プラットフォーム効果はビルダーモデル自身の能力と比較して中程度であること、そしてより弱いターゲットモデルが最大の向上を得ることを発見した。これらの結果は、推論時ハーネス設計が従来の訓練時蒸留に対する重要な補完であり、強力なモデルが再訓練なしに認知構造をより弱いモデルへと転移することを可能にすることを示唆している。

One-sentence Summary

Salesforce AI Research and the University of Illinois Urbana-Champaign propose strong-to-weak scaffolding, a test-time capability transfer method in which a stronger builder model constructs inference-time harnesses that allow a weaker target model to nearly double its average performance on Theory-of-Mind benchmarks from 0.490.490.49 to 0.910.910.91 without any parameter updates, by offloading unstable model reasoning into deterministic code, benchmark-specific routing, and strict answer-format enforcement.

Key Contributions

  • The paper studies strong-to-weak scafolding, a test-time paradigm where a stronger builder model constructs an inference-time harness that helps a weaker target model solve tasks without parameter updates.
  • On four Theory-of-Mind benchmarks, using only 5% of data as a validation set, builder models iteratively refine harnesses and nearly double average target-model accuracy from 0.49 to 0.91.
  • Analysis shows that the gains come from ofloading unstable model reasoning into deterministic code, benchmark-specific routing, and strict answer-format enforcement. Builder reasoning effort improves harness quality monotonically, and weaker target models receive the largest benefits.

Introduction

The authors study a practical alternative to model distillation: instead of training a weak model to match a stronger one, they let a strong builder model construct an inference-time harness (routing, prompts, deterministic solvers, verification, format enforcement) that makes the task easier for a fixed weak target model. Prior work either updates model parameters through distillation or fine-tuning, or improves a single model's reasoning via prompting and tool use, but does not transfer capability across models without training. The key contribution is formalizing strong-to-weak scaffolding as a distinct inference-time transfer setting and providing a systematic empirical analysis on Theory-of-Mind benchmarks, showing that well-designed scaffolds can boost a weak model's accuracy dramatically (e.g., from 0.49 to 0.91) by externalizing structured reasoning rather than relying on more training data or longer model reasoning.

Method

The authors propose an automatic harness-building framework in which a strong builder model MbuildM_{\mathrm{build}}Mbuild constructs an inference-time scaffold for a fixed, weaker target model MtarM_{\mathrm{tar}}Mtar. The goal is to improve the target model's performance on a collection of benchmarks by wrapping it with a custom inference procedure that can include prompt engineering, routing, pre/post-processing, verification, or other strategies.

For each benchmark D(j)\mathcal{D}^{(j)}D(j), a small validation split V(j)\mathcal{V}^{(j)}V(j) containing 5% of the examples is randomly sampled, and the remaining examples are set aside as a hidden test split T(j)\mathcal{T}^{(j)}T(j). The union of all validation splits is denoted V\mathcal{V}V, and the union of all test splits is T\mathcal{T}T. The builder model has access only to V\mathcal{V}V during scaffold construction, while T\mathcal{T}T is strictly held out for final evaluation.

At the start of a build run, the builder is placed inside an existing agentic coding harness Hbuild\mathcal{H}_{\mathrm{build}}Hbuild and given an initial workspace

W0={R,Cdemo,V},\mathcal{W}_0 = \{\mathcal{R}, \mathcal{C}_{\mathrm{demo}}, \mathcal{V}\},W0={R,Cdemo,V},

where R\mathcal{R}R is a rule file describing task instructions and submission format, Cdemo\mathcal{C}_{\mathrm{demo}}Cdemo is a demonstration file showing how to call MtarM_{\mathrm{tar}}Mtar, and V\mathcal{V}V is the labeled validation set. The builder is not constrained to a fixed scaffold architecture. It may implement any inference-time procedure, such as prompt templates, benchmark routing, deterministic pre- or post-processing, answer-format enforcement, verification passes, few-shot retrieval, or direct symbolic solvers. The only requirement is that the final scaffold exposes a single entry point that can be applied to unseen test examples.

Conceptually, the builder searches over a space of possible scaffolds S\mathcal{S}S to find the scaffold that maximizes accuracy on the hidden test set:

S=argmaxSSAcc(S,Mtar;T).S^{\star} = \arg\max_{S \in \mathcal{S}} \mathrm{Acc}\big(S, M_{\mathrm{tar}}; \mathcal{T}\big).S=argSSmaxAcc(S,Mtar;T).

Since T\mathcal{T}T is hidden, the builder instead uses validation performance as a proxy and selects a scaffold S^\hat{S}S^ by optimizing

S^=argmaxSSbuildAcc(S,Mtar;V).\hat{S} = \arg\max_{S \in \mathcal{S}_{\text{build}}} \operatorname{Acc}\bigl(S, M_{\text{tar}}; \mathcal{V}\bigr).S^=argSSbuildmaxAcc(S,Mtar;V).

The success of a scaffold therefore depends on identifying reusable task structure from the small validation slice and transferring it to the hidden test set. After the builder finishes, a human evaluator runs the exported entry point on T\mathcal{T}T without further builder intervention. The overall process is summarized in Algorithm 1.

Experiment

The study evaluates strong-to-weak scaffolding, where a builder model writes a test-time harness to improve a weaker target model’s performance on Theory of Mind benchmarks. The experiments systematically vary the builder, platform, target, and builder reasoning effort, revealing that scaffolding consistently lifts the weak target, with the builder model and its reasoning budget being the dominant drivers of improvement, while the coding platform and the number of validation iterations have little effect. The gains arise from offloading structured sub-tasks into deterministic rules and from imposing reliable output formats, thereby reducing the target model’s cognitive load; the approach is most effective when the target has recoverable latent competence, and the remaining errors cluster in tasks that resist compilation, such as deep recursive belief tracking and Bayesian goal inference.

Scaffolding consistently lifts GPT-5.4-mini's macro-average accuracy from 0.488 to a mean of 0.763 across all runs, with every builder configuration outperforming the baseline. The best scaffolded run reaches 0.912, exceeding even the larger unscaffolded GPT-5.4 model, and the automated scaffold matches or surpasses a human-designed harness on the structured BigToM task. Residual errors remain concentrated in tasks demanding nested belief tracking or Bayesian inference, where the builder cannot fully compile away the reasoning burden. All 11 builder configurations improved over the no-scaffold baseline, and 100% of scaffolded runs exceeded it. The best run (GPT-5.5 on GPT Codex) achieved a 0.423 absolute improvement, an 87% relative uplift. The strongest scaffold outperformed both vanilla GPT-5.4 and GPT-OSS-120B baselines on every benchmark. On BigToM, the automated scaffold reached near-ceiling accuracy (1.00) and slightly exceeded the human-inspired harness (0.95). Top scaffolds fixed 83% of baseline mistakes while breaking only 7% of baseline-correct items, showing near-Pareto improvement. Remaining errors clustered in harder subtasks: Hi-ToM accuracy declined with recursion depth, and MMToM-QA errors concentrated in Bayesian goal-inference questions.

All scaffold builders substantially improve average accuracy over the no-scaffold baseline, with absolute gains of roughly a third or more. BigToM is near ceiling across builder models, while Hi-ToM, MMToM, and MuMA-ToM remain lower and more variable. The strongest builder configurations reach average accuracy above 0.85, and residual errors cluster in harder recursive, deceptive, and goal-inference cases. Every scaffold builder raises average accuracy well above the baseline, with absolute improvements ranging from roughly 0.31 to 0.39. BigToM is the most solved task component, reaching near-perfect or perfect accuracy for every listed builder. Top builder models such as GPT-5.5 and Opus-4.7 x-high achieve the highest overall averages, while no listed scaffold builder falls below 0.79.

Refinement improves validation accuracy for all builders, but the amount of improvement varies widely. Builders that start lower such as Sonnet and Gemini gain the most, while GPT-5.5 reaches the highest final validation accuracy and Opus variants tend to start stronger with smaller subsequent gains. Validation-full optimism gaps are mostly small and positive, indicating validation modestly overestimates full-set performance. GPT-5.5 attains the highest best validation accuracy despite having a lower first validation accuracy than the strongest-starting Opus variant. Sonnet and Gemini show the largest accuracy gains and also have the highest validation run counts. Opus variants generally begin with stronger validation accuracy but improve less than the largest-gain builders. GPT-5.4-mini records the smallest accuracy gain among the listed builders. Most validation-full optimism gaps are small and positive, while Gemini is effectively neutral.

Increasing the builder model's reasoning effort consistently improves scaffold quality, with pooled macro accuracy rising from 0.71 at low effort to 0.86 at extra-high effort. The largest gain occurs between low and medium effort, while further increases bring smaller but positive returns, and scaffold code length grows substantially with more deliberation. Pooled macro accuracy climbs monotonically with effort tier, from 0.71 (low) to 0.86 (extra-high), with a strong rank correlation between effort and per-run accuracy. Scaffold size expands from roughly 500–650 lines of Python at low effort to over 1000 lines at higher effort, indicating that more reasoning compiles additional decision logic into the harness.

Scaffolds targeting the weak GPT-5.4-mini model achieve higher final accuracy when a larger share of evaluation items are answered by deterministic code or rules, with a strong correlation (r=0.72) between determinism fraction and accuracy. This offloading reduces the reasoning burden on the target model, but its feasibility varies sharply across benchmarks: BigToM is almost fully offloadable, while MuMA-ToM resists compilation into structured code. Higher determinism fractions are strongly associated with higher accuracy (r=0.72): runs with determinism above 0.98 reach accuracies above 0.87, whereas the lowest-determinism run (0.75) yields only 0.749 accuracy. The ability to offload tasks to deterministic code is highly benchmark-dependent, with mean determinism ranging from ~0.94 in BigToM to ~0.36 in MuMA-ToM, where free-form dialogue reasoning is hardest to compile away.

the paper evaluate automated scaffold builders that compile reasoning strategies into code to assist a target model on theory-of-mind benchmarks. Scaffolding consistently improves accuracy over the no-scaffold baseline, with the best scaffolds surpassing larger unscaffolded models and achieving near-ceiling performance on structured tasks like BigToM. Residual errors cluster in nested belief tracking, Bayesian inference, and free-form dialogue reasoning, which resist compilation into deterministic rules. Builder model capability, reasoning effort, and the fraction of items offloaded to deterministic code all correlate with scaffold quality, though gains vary across benchmarks and builder types.


AIでAIを構築

アイデアからローンチまで — 無料のAIコーディング支援、すぐに使える環境、最高のGPU価格でAI開発を加速。

AI コーディング補助
すぐに使える GPU
最適な料金体系

HyperAI Newsletters

最新情報を購読する
北京時間 毎週月曜日の午前9時 に、その週の最新情報をメールでお届けします
メール配信サービスは MailChimp によって提供されています