HyperAIHyperAI

Command Palette

Search for a command to run...

EarlyEval: 早期結果予測によるエージェント評価の低コスト化

Yuling Shi Zhensu Sun Junsen Dong Chengcheng Wan David Lo Xiaodong Gu

概要

LLMエージェントの評価はその開発を導く上で不可欠だが、そのコストは法外なものとなっている。先端的モデルをエージェントベンチマークで1回実行するだけで数百ドルから数千ドルを要し、このコストが反復的な開発サイクルを通じて繰り返し発生する。ベンチマークの蒸留に焦点を当てた従来の取り組みは評価タスクの数を削減するが、残された各タスクの実行コストには手つかずのままである。本研究では、各タスク内のコストを削減する、効率性の相補的な軸として早期結果予測を導入する。我々の核となる洞察は、エージェントの最終結果が、実行完了よりもかなり前に、その中間的な振る舞いから明らかになることが多いという点にある。このアイデアを具現化したのがEarlyEvalである。これは軽量なフレームワークであり、行動的特徴、テキスト的特徴、参照解に基づく特徴を用いて、LightGBMによる成功分類器と失敗分類器のペアを訓練し、いずれかの分類器が較正された信頼度閾値を超えた瞬間にエージェントの実行を停止する。これにより、ステップごとのオーバーヘッドは無視できる程度に抑えられる。SWE-bench Verified、TerminalBench、Toolathlonの3つのベンチマークにおいて、EarlyEvalは89%から97%の予測精度を達成しつつ、エージェントのステップ数を13%から26%、入力トークン数を最大44.1%、出力トークン数を最大29.4%削減可能であり、エージェントごとの解決率の変動は平均してわずか1から2パーセントポイントに留まる。

One-sentence Summary

Researchers from Shanghai Jiao Tong University, Singapore Management University, East China Normal University, and Shanghai Innovation Institute propose EarlyEval, a lightweight framework that trains LightGBM success and failure classifiers on behavioral, textual, and reference-solution features to halt agent runs early at calibrated confidence thresholds, eliminating 13–26% of steps and up to 44.1% input tokens across SWE-bench Verified, TerminalBench, and Toolathlon with 89–97% accuracy and minimal resolve rate perturbation.

Key Contributions

  • Early outcome prediction is introduced as a complementary efficiency axis for LLM agent benchmarking, cutting cost within each task by terminating runs early based on intermediate behavior rather than reducing the number of tasks.
  • EarlyEval, a lightweight framework, trains a pair of LightGBM success and failure classifiers on behavioral, textual, and reference-solution features and stops an agent run when either classifier exceeds a calibrated confidence threshold.
  • Across SWE-bench Verified, TerminalBench, and Toolathlon, EarlyEval eliminates 13% to 26% of agent steps and up to 44.1% of input tokens at 89% to 97% prediction accuracy, while per-agent resolve rates shift by only one to two percentage points on average.

Introduction

Evaluating LLM agents is essential for guiding development, but the cost of running modern agentic benchmarks has surged, with a single pass on SWE-bench Verified costing hundreds of dollars and longer-rollout benchmarks reaching thousands. This expense makes frequent evaluation impractical for many teams and slows the iteration cycle. Prior efforts to reduce costs have focused on benchmark distillation, which downsizes the task set while leaving the per-task execution cost unchanged. The authors introduce EarlyEval, a complementary approach that terminates an agent’s rollout early when its final outcome can be confidently predicted from intermediate behavior. By training lightweight classifiers on historical trajectories from other agents, EarlyEval cuts execution steps and token consumption while preserving per-agent resolve rates and leaderboard rankings across multiple benchmarks.

Dataset

The authors construct a training dataset from agent trajectories collected across three multi-step benchmarks. Each trajectory is a sequence of steps with a final binary success label, and the data is used to train a model that predicts eventual success from partial execution prefixes.

Dataset sources and composition

  • SWE-bench Verified

    • 500 human-validated GitHub issues across 12 Python repositories.
    • Trajectories generated by mini-SWE-agent paired with 16 LLMs (Claude, GPT-5, Gemini, GLM, DeepSeek, Devstral, Kimi, MiniMax families).
    • Total: 7,805 trajectories.
    • Each task includes a gold patch, enabling reference-solution features.
  • TerminalBench

    • 89 command-line automation tasks.
    • 37 distinct agent configurations (scaffolds: mini-SWE-agent, Codex CLI, Claude Code, Gemini CLI, OpenHands, Terminus-2; models: GPT-5, GPT-5-mini, Claude-Haiku-4.5, Claude-Opus-4.5, Gemini-2.5-Pro).
    • Total: 6,757 trajectories (multiple rollouts per task).
    • No per-task reference solutions; only behavioral and textual features are used.
  • Toolathlon

    • 108 complex API and tool-use tasks.
    • Native Toolathlon scaffold with 22 LLMs, three rollouts per task.
    • Total: 7,116 trajectories.
    • No reference solutions; same feature restrictions as TerminalBench.

Data processing pipeline

  • Trajectories shorter than 10 steps are discarded (insufficient signal).
  • For each trajectory of length T, all prefixes of lengths 0 through T are extracted and paired with the trajectory’s final outcome label (success/failure).
  • Each prefix is converted to a fixed-length feature vector composed of three families:
    • Behavioral features: run progression metrics (volume, pacing, milestone timing, error/test signals, stalling patterns).
    • Textual features: semantic blocks (task prompt, full action history, most recent action, full environment feedback, most recent feedback) are vectorized independently with TF-IDF over word n-grams, then compressed to 64 dimensions (prompt) or 128 dimensions (action/feedback groups) via Truncated SVD, preserving block boundaries while keeping dimensionality low.
    • Reference-solution features (SWE-bench only): measures structural overlap between the current prefix and the provided gold patch (files, symbols, tests). Omitted for TerminalBench and Toolathlon.

Usage in the model The resulting feature vectors and binary labels serve as the training set. The model learns from these prefix-label pairs to predict final success at intermediate steps, enabling early termination decisions. No explicit train/validation split or mixture ratios are detailed in the provided text.

Method

The authors propose EarlyEval, a framework designed to predict an agent’s final outcome on a benchmark task from a partial trajectory and halt execution as soon as the eventual outcome becomes statistically evident. The system operates through a sequential inference workflow comprising two primary stages: offline predictor construction and online step-by-step inference.

As shown in the figure below:

In the offline phase, the authors leverage historical agent runs that have already been evaluated to construct the training data. For a given benchmark, they collect a pool of trajectories τ=(e1,...,eT)\tau = (e_1, ..., e_T)τ=(e1,...,eT), each associated with a binary evaluation score y{0,1}y \in \{0, 1\}y{0,1}. Trajectories shorter than 10 steps are discarded to ensure sufficient signal for optimization. Each trajectory is decomposed into a sequence of labeled prefixes τ:k=(e1,,ek)\tau_{:k} = (e_1, \ldots, e_k)τ:k=(e1,,ek) for k=0,1,,Tk = 0, 1, \ldots, Tk=0,1,,T, where each prefix is paired with the final outcome label yyy. These prefixes are mapped to a fixed-length multimodal feature vector ϕ(τ:k)Rd\phi(\tau_{:k}) \in \mathbb{R}^dϕ(τ:k)Rd. The feature extraction process captures three distinct families of signals. Behavioral features capture run progression invariants, including volume metrics, structural composition, milestone timing, and environment feedback signals. Textual features encode the natural-language context by isolating the task prompt, action history, and environment feedback into distinct semantic blocks. Each block is vectorized using TF-IDF over word n-grams and compressed via Truncated Singular Value Decomposition to maintain computational efficiency. Reference-solution features are optionally leveraged when ground-truth human patches are available, measuring the structural overlap between the current prefix and the gold solution.

To judge these partial trajectories, the authors train a pair of agent-agnostic predictors using gradient-boosted decision tree ensembles via LightGBM. This architecture is selected for its ability to evaluate high-dimensional feature vectors in under a millisecond on a single CPU core, ensuring negligible computational overhead during step-by-step inference. EarlyEval optimizes two separate ensembles over the feature representation ϕ\phiϕ: a success predictor h+h_+h+ and a failure predictor hh_-h. Both models ingest the same feature vector but are optimized against inverted target sets. The success predictor targets y=1y = 1y=1, while the failure predictor targets 1y=11 - y = 11y=1 (i.e., y=0y = 0y=0). Training two distinct predictors allows positive and negative evidence to accumulate independently, reflecting the asymmetric behaviors signaling success and failure. This design also creates an explicit unconfident region where both predictors output low probabilities, allowing the agent to continue execution when the outcome remains ambiguous. To prevent data leakage, the trajectory pool is partitioned by task into training and validation folds. Furthermore, each prefix instance is weighted by 1/(T+1)1 / (T + 1)1/(T+1) to ensure that every trajectory contributes identical total mass to the objective function, preventing prolonged trajectories from dominating the optimization loss.

During online inference, the agent interacts with the environment step by step. At each step, EarlyEval extracts the feature vector ϕ\phiϕ from the accumulated partial trajectory and inputs it into both ensembles. Because regularized tree ensembles can distort output probability scales, the authors recalibrate the raw scores using Platt scaling. A one-dimensional logistic regression maps the raw ensemble score s^\hat{s}s^ to a calibrated probability:

p=σ(a\logit(s^)+b)p = \sigma \big(a \logit(\hat{s}) + b \big)p=σ(a\logit(s^)+b)

where the scalar parameters aaa and bbb are fitted on the held-out validation split. This monotonic transformation rescales the outputs so that confidence thresholds carry a consistent meaning across both predictors. The calibrated probabilities p+p_+p+ and pp_-p parameterize a dual-threshold decision mechanism. The system compares p+p_+p+ against a success threshold sss and pp_-p against a failure threshold fff. A run is intercepted and marked with a predicted outcome at the first step where p+sp_+ \ge sp+s or pfp_- \ge fpf. If both probabilities remain below their respective thresholds, the system defers commitment and allows the agent to proceed to subsequent steps. This threshold-based logic dictates the stringency of evidence required before intervention, enabling a tunable trade-off between prediction accuracy and compute savings.

Experiment

EarlyEval is evaluated on three multi-step agentic benchmarks (SWE-bench Verified, TerminalBench, Toolathlon) using a leave-one-agent-out protocol to predict task outcomes and halt unpromising trajectories early. The system achieves substantial reductions in execution steps and token usage while maintaining high fidelity to full-run resolve rates and preserving agent rankings with strong rank correlations. Ablation studies show that EarlyEval's predictions are robust to missing feature families, with behavioral features being most critical, and that a LightGBM backbone outperforms neural and linear alternatives in the cost-fidelity trade-off.

Evaluation costs vary substantially across benchmarks and models, with no single model consistently cheapest. GPT-5.5 offers the lowest cost on three of five benchmarks, while Gemini 3.1 Pro is most economical on the multimodal suite and Claude 5 leads on SWE-bench Verified. GPT-5.5 achieves the lowest cost on SWT-bench, Commit0, and GAIA, where its GAIA run costs only 122comparedto122 compared to122comparedto1,305 for Claude 5. Gemini 3.1 Pro evaluates SWE-bench Multimodal for 641,significantlyundercuttingGPT5.5(641, significantly undercutting GPT-5.5 (641,significantlyundercuttingGPT5.5(1,453) and Claude 5 ($2,270).

The LightGBM backbone uniquely combines high early-stopping accuracy with substantial coverage, achieving the largest step reduction and minimal metric distortion. In contrast, direct neural and linear models either suffer from low accuracy or remain too passive to save meaningful compute, while a fine-tuned LLM judge offers competitive fidelity but introduces prohibitive inference overhead that cancels out its own savings. LightGBM reaches 95.0% accuracy and 34.8% coverage, cutting execution steps by 26.0% while keeping metric distortion to just 1.1 points. The fine-tuned Qwen judge attains 90.7% accuracy but saves only 17.9% of steps and requires a costly model forward pass at every trajectory step, offsetting the computational savings early stopping aims to provide.

Two experiments examine efficiency in model evaluation. The first compares evaluation costs across five benchmarks, revealing that no single model is universally cheapest and that cost efficiency depends on the benchmark. The second evaluates early-stopping strategies, where a LightGBM backbone achieves the best balance of high accuracy, substantial step reduction, and minimal metric distortion, while fine-tuned LLM judges incur prohibitive inference overhead that negates their savings.


AIでAIを構築

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

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

HyperAI Newsletters

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