BOAD: Discovering Hierarchical Software Engineering Agents via Bandit Optimization

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: Xu, Iris, Zeng, Guangtao, He, Zexue, Jin, Charles, Pareja, Aldo, Gutfreund, Dan, Gan, Chuang, Hong, Zhang-Wei
Formato: Preprint
Publicado: 2025
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866909979660779520
author Xu, Iris
Zeng, Guangtao
He, Zexue
Jin, Charles
Pareja, Aldo
Gutfreund, Dan
Gan, Chuang
Hong, Zhang-Wei
author_facet Xu, Iris
Zeng, Guangtao
He, Zexue
Jin, Charles
Pareja, Aldo
Gutfreund, Dan
Gan, Chuang
Hong, Zhang-Wei
contents Large language models (LLMs) have shown strong reasoning and coding capabilities, yet they struggle to generalize to real-world software engineering (SWE) problems that are long-horizon and out of distribution. Existing systems often rely on a single agent to handle the entire workflow-interpreting issues, navigating large codebases, and implementing fixes-within one reasoning chain. Such monolithic designs force the model to retain irrelevant context, leading to spurious correlations and poor generalization. Motivated by how human engineers decompose complex problems, we propose structuring SWE agents as orchestrators coordinating specialized sub-agents for sub-tasks such as localization, editing, and validation. The challenge lies in discovering effective hierarchies automatically: as the number of sub-agents grows, the search space becomes combinatorial, and it is difficult to attribute credit to individual sub-agents within a team. We address these challenges by formulating hierarchy discovery as a multi-armed bandit (MAB) problem, where each arm represents a candidate sub-agent and the reward measures its helpfulness when collaborating with others. This framework, termed Bandit Optimization for Agent Design (BOAD), enables efficient exploration of sub-agent designs under limited evaluation budgets. On SWE-bench-Verified, BOAD outperforms single-agent and manually designed multi-agent systems. On SWE-bench-Live, featuring more recent and out-of-distribution issues, our 36B system ranks second on the leaderboard at the time of evaluation, surpassing larger models such as GPT-4 and Claude. These results demonstrate that automatically discovered hierarchical multi-agent systems significantly improve generalization on challenging long-horizon SWE tasks. Code is available at https://github.com/iamxjy/BOAD-SWE-Agent.
format Preprint
id arxiv_https___arxiv_org_abs_2512_23631
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle BOAD: Discovering Hierarchical Software Engineering Agents via Bandit Optimization
Xu, Iris
Zeng, Guangtao
He, Zexue
Jin, Charles
Pareja, Aldo
Gutfreund, Dan
Gan, Chuang
Hong, Zhang-Wei
Machine Learning
Artificial Intelligence
Large language models (LLMs) have shown strong reasoning and coding capabilities, yet they struggle to generalize to real-world software engineering (SWE) problems that are long-horizon and out of distribution. Existing systems often rely on a single agent to handle the entire workflow-interpreting issues, navigating large codebases, and implementing fixes-within one reasoning chain. Such monolithic designs force the model to retain irrelevant context, leading to spurious correlations and poor generalization. Motivated by how human engineers decompose complex problems, we propose structuring SWE agents as orchestrators coordinating specialized sub-agents for sub-tasks such as localization, editing, and validation. The challenge lies in discovering effective hierarchies automatically: as the number of sub-agents grows, the search space becomes combinatorial, and it is difficult to attribute credit to individual sub-agents within a team. We address these challenges by formulating hierarchy discovery as a multi-armed bandit (MAB) problem, where each arm represents a candidate sub-agent and the reward measures its helpfulness when collaborating with others. This framework, termed Bandit Optimization for Agent Design (BOAD), enables efficient exploration of sub-agent designs under limited evaluation budgets. On SWE-bench-Verified, BOAD outperforms single-agent and manually designed multi-agent systems. On SWE-bench-Live, featuring more recent and out-of-distribution issues, our 36B system ranks second on the leaderboard at the time of evaluation, surpassing larger models such as GPT-4 and Claude. These results demonstrate that automatically discovered hierarchical multi-agent systems significantly improve generalization on challenging long-horizon SWE tasks. Code is available at https://github.com/iamxjy/BOAD-SWE-Agent.
title BOAD: Discovering Hierarchical Software Engineering Agents via Bandit Optimization
topic Machine Learning
Artificial Intelligence
url https://arxiv.org/abs/2512.23631