Think Big, Search Small: Where Capacity Matters in Hierarchical Search Agents?

· Source: Takara TLDR - Daily AI Papers · Field: Technology & Digital — Artificial Intelligence & Machine Learning, Software Development & Engineering · Depth: Advanced, quick

Summary

Hierarchical search agents, which use large language models to decompose complex questions into sub-queries for parallel sub-agents, typically instantiate all roles with a single model of identical scale. This research investigates optimal model capacity distribution by factorizing hierarchical search into delegation, execution, and a fixed answer generation role. Experiments on five multi-hop QA benchmarks reveal that role factorization consistently outperforms single-agent baselines, improving exact match from 4.5 to 8.6 points. Crucially, capacity sensitivity is asymmetric; scaling the delegation backbone yields an ~11-point EM improvement, while scaling the execution sub-agent only provides ~2.6 points, identifying decomposition as the primary capability bottleneck. Furthermore, a 1.7B-parameter executor, trained via quality-filtered trajectory distillation, achieves accuracy comparable to a frontier sub-agent while consuming 37% fewer sub-agent tokens.

Key takeaway

For AI Engineers designing multi-agent systems for complex QA, you should prioritize allocating significant model capacity to the delegation component. This strategy, focusing on task decomposition, yields substantial performance gains, as scaling delegation improves exact match by ~11 points. Conversely, you can downsize your execution sub-agents, potentially using smaller models like a 1.7B-parameter executor, to reduce token consumption by 37% without sacrificing accuracy, optimizing resource allocation.

Key insights

Concentrating model capacity in the delegation role significantly improves hierarchical search agent performance, while execution can be downsized.

Principles

Method

Factorize hierarchical search into delegation, execution, and answer generation roles. Conduct controlled capacity sweeps on delegation and execution axes to identify bottlenecks.

In practice

Topics

Code references

Best for: Research Scientist, AI Architect, NLP Engineer, AI Scientist, Machine Learning Engineer, AI Engineer

Related on AIssential

Open in AIssential →

Editorial summary, takeaway, and curation by AIssential. Original article published by Takara TLDR - Daily AI Papers.