Research map / Large language models / Agents, code and math
Guided Search Strategies in Non-Serializable Environments with Applications to Software Engineering Agents
Read the paperCheck your idea against 108,967 papers
AI-extracted summary. Written by AI from the paper's full text. Numbers and claims can be wrong or missing context, so check the paper before you cite or build on it.
Proposes 1-step lookahead and critic-guided trajectory selection as test-time search methods for non-serializable environments (Docker-based SWE-agent), roughly doubling SWE-bench Verified success for a fine-tuned Qwen2.5-72B policy (40.8%) and improving GPT-4o.
Problem
Test-time search methods such as MCTS cannot be applied to non-serializable RL environments (e.g., Docker containers) where states cannot be saved, restored, or revisited, so LLM agents show high variance across attempts on tasks like SWE-bench.
Approach
Two search operators that need no state rollback: sample-based 1-step lookahead and trajectory selection, both guided by a learned action-value critic (TD(lambda)-trained LLM) that is applied to a policy's sampled actions or complete trajectories.
Key results
- Qwen-based policy with 1-step lookahead and trajectory selection (N=15, K=8) reaches 40.8% success on full SWE-bench Verified, a new open-weights state of the art.
- On Verified-50, Qwen-based policy success rises from 16.2% to 41.7% with lookahead plus trajectory selection (N=10) in the default regime.
- GPT-4o policy success rises from 22.0% to 40.0% with 1-step lookahead plus trajectory selection (N=5) in the default regime.
Measured effects reported in the paper
| What was changed | Where | Measured | Effect |
|---|---|---|---|
| Add 1-step lookahead | Qwen-based policy · Verified-50 (Default) | success rate (%): 16.2 -> 26.8 | Helped |
| Add trajectory selection (N=10) | Qwen-based policy · Verified-50 (Default) | success rate (%): 16.2 -> 31.27 | Helped |
| Add 1-step lookahead and trajectory selection (N=10) | Qwen-based policy · Verified-50 (Default) | success rate (%): 16.2 -> 41.7 | Helped |
| Add 1-step lookahead and trajectory selection (N=5) | GPT-4o policy · Verified-50 (Default) | success rate (%): 22.0 -> 40.0 | Helped |
| Run until submitted rather than stopping on first completion (ablation) | Qwen-based policy · Verified-50 | success rate (%): 16.2 -> 22.8 | Helped |
| Run until submitted rather than stopping on first completion (ablation) | GPT-4o policy · Verified-50 | success rate (%): 22.0 -> 22.0 | No clear effect |
| Use Qwen2.5-72B instead of LLaMA3.1-70B as critic base model (ablation) | Qwen-based policy + 1-step lookahead · Verified-50 (Default) | success rate (%): 26.8 -> 22.8 | Hurt |
| Use LLaMA3.1-8B instead of LLaMA3.1-70B as critic base model (ablation) | Qwen-based policy + 1-step lookahead · Verified-50 (Default) | success rate (%): 26.8 -> 21.0 | Hurt |
| Use Qwen2.5-7B instead of LLaMA3.1-70B as critic base model (ablation) | Qwen-based policy + 1-step lookahead · Verified-50 (Default) | success rate (%): 26.8 -> 20.2 | Hurt |
| Remove the critic (ablation) | Qwen-based policy + 1-step lookahead · Verified-50 (Default) | success rate (%): 26.8 -> 16.2 | Hurt |
| Use TD(0.7) targets rather than Monte-Carlo (TD(1)) or one-step TD (TD(0)) (ablation) | Critic guiding Qwen-based policy with 1-step lookahead · SWE-bench Verified | success rate | Helped |
| Increase the discount factor γ toward 1 for Monte-Carlo-trained critic (ablation) | Qwen-based policy with 1-step lookahead · SWE-bench Verified | success rate | Helped |
Surprising findings
- Critic-guided 1-step lookahead causes value hacking, inflating critic values for negative trajectories.
- A GPT-4o critic was not trained on GPT-4o trajectories yet still improved its success rate substantially.
Limitations
- Non-serializable setting precludes MCTS and other rollback-based search, so only forward-only operators are studied.
- Critic quality is bounded: performance saturates at K=8 and the critic is a modest model relative to the GPT-4o policy.
Experimental setup
Datasets: SWE-bench Verified SWE-bench training issues (6,500 issue-PR pairs) bootstrapped agent trajectories (80,000)
Benchmarks: SWE-bench Verified Verified-50
Metrics: success rate standard error of the mean Pass@N
Baselines: Qwen-based policy without search GPT-4o policy SWE-Gym SWE-Fixer Lingma Agent + Lingma-SWE-GPT-72B
Builds on: SWE-agent SWE-bench TD(lambda) Monte-Carlo Tree Search CRIU
Closest papers by meaning
- SWE-Search: Enhancing Software Agents with Monte Carlo Tree Search and Iterative Refinement (ICLR 2025 · similarity 0.91)
- Fleet of Agents: Coordinated Problem Solving with Large Language Models (ICML 2025 · similarity 0.88)
- AgentSwift: Efficient LLM Agent Design via Value-Guided Hierarchical Search (AAAI 2026 · similarity 0.88)
- Demystifying and Enhancing the Efficiency of Large Language Model Based Search Agents (ICLR 2026 · similarity 0.87)
- Discovering Hierarchical Software Engineering Agents via Bandit Optimization (ICLR 2026 · similarity 0.87)
- LLMs Improving LLMs: Agentic Discovery for Test-Time Scaling (NeurIPS 2026 · similarity 0.87)
Writing a paper in this area? scime draft is a free LaTeX editor with AI included, real-time co-authors and a check of your references and venue rules before you submit. It also imports OpenAI Prism projects.
