Route, Communicate, and Reason: Gated Routing and Adaptive Depth for Efficient Multi-Agent Reasoning
Summary
GRADE (Gated Routing and Adaptive Depth for Efficient Reasoning) is a hierarchical multi-agent system designed to optimize multi-agent ensembling by addressing challenges of agent selection, hierarchy depth, and inter-agent communication costs. It employs four lightweight learned gates to govern these aspects, along with branch pruning. The system is trained using CoGRPO (Collaborative Group-Relative Policy Optimization), a novel critic-free recipe adapted for multi-agent hierarchies, which assigns a shared advantage signal to participating gates and agents. GRADE utilizes a hot-swappable Expert Registry, allowing expert models to be replaced at inference time without retraining via per-agent calibration maps. With approximately 17B average active parameters, GRADE surpasses all baselines on GSM8K, MMLUPro, and GPQA, outperforming the strongest baseline by 4.8 points on MMLUPro at half the active compute. It also remains competitive on AIME-2025. Ablation studies indicate the hierarchy and masked cross-attention are key accuracy contributors, and per-agent calibration is crucial for safe hot-swapping.
Key takeaway
For Machine Learning Engineers designing efficient multi-agent systems, GRADE offers a blueprint to significantly reduce inference costs while boosting performance. You should consider implementing gated routing and adaptive depth mechanisms to dynamically manage agent consultation and communication. Utilizing a hot-swappable expert registry with per-agent calibration can enhance system flexibility, enabling expert model updates without full retraining. This approach could yield substantial gains in compute efficiency and accuracy for complex reasoning tasks.
Key insights
GRADE optimizes multi-agent reasoning via gated routing and adaptive depth, reducing compute while improving performance.
Principles
- Gated routing can dynamically select agents.
- Adaptive depth controls query traversal efficiency.
- Shared advantage signals train multi-agent gates.
Method
GRADE uses four learned gates for agent selection, hierarchy depth, communication, and pruning, trained with CoGRPO, a critic-free multi-agent policy optimization.
In practice
- Implement dynamic agent selection with gates.
- Use per-agent calibration for flexible expert models.
- Apply masked cross-attention for accuracy gains.
Topics
- Multi-Agent Systems
- Gated Routing
- Adaptive Depth
- Expert Registry
- Inference Efficiency
- Collaborative Policy Optimization
Best for: Research Scientist, AI Engineer, AI Scientist, Machine Learning Engineer, NLP Engineer
Related on AIssential
See Counsel's argued verdicts on the open AI decisions leaders are weighing →
Editorial summary, takeaway, and curation by AIssential. Original article published by Artificial Intelligence.