FastContext: Training Efficient Repository Explorer for Coding Agents

· Source: cs.SE updates on arXiv.org · Field: Technology & Digital — Artificial Intelligence & Machine Learning, Software Development & Engineering · Depth: Expert, extended

Summary

FastContext is a novel exploration subagent designed to enhance Large Language Model (LLM) coding agents by decoupling repository exploration from task solving. This system, powered by specialized models ranging from 4B to 30B parameters, addresses the significant bottleneck of locating relevant code, which often consumes excessive token budgets and introduces irrelevant information into an agent's context. FastContext operates on demand, executing parallel tool calls (Read, Glob, Grep) to identify and return concise file paths and line ranges as focused context. Its models are trained using supervised fine-tuning from reference trajectories and further refined with reinforcement learning for broad initial search, multi-turn evidence gathering, and precise citation generation. Integrating FastContext into Mini-SWE-Agent has demonstrated notable improvements, boosting end-to-end resolution rates by up to 5.5% and reducing main-agent token consumption by up to 60% across SWE-bench Multilingual, SWE-bench Pro, and SWE-QA benchmarks, with minimal overhead.

Key takeaway

For Machine Learning Engineers developing or deploying LLM-powered coding agents, you should re-evaluate monolithic agent architectures. Integrating a dedicated, trained exploration subagent like FastContext can significantly reduce your main model's token consumption by up to 60% and improve task resolution rates by up to 5.5%. Consider modularizing your agent workflow to offload repository navigation to specialized, smaller models, thereby optimizing inference costs and enhancing overall performance.

Key insights

Dedicated exploration subagents improve LLM coding agent efficiency and accuracy by providing focused context.

Principles

Method

FastContext delegates repository exploration, executing parallel Read, Glob, and Grep tool calls. It is trained via SFT for initial behavior and RL for task-relevant code localization.

In practice

Topics

Code references

Best for: AI Scientist, Machine Learning Engineer, AI Engineer

Related on AIssential

Open in AIssential →

Editorial summary, takeaway, and curation by AIssential. Original article published by cs.SE updates on arXiv.org.