Adaptive Multi-Expert Reasoning via Difficulty-Aware Routing and Uncertainty-Guided Aggregation
Organizations: Faculty of Computer Science, October University for Modern Science & Arts, Giza, Egypt
Abstract
Large language models (LLMs) demonstrate strong performance in math reasoning benchmarks, but their performance varies inconsistently across problems with varying levels of difficulty. This paper describes Adaptive Multi-Expert Reasoning (AMR), a framework that focuses on problem complexity by reasoning with dynamically adapted strategies. An agile routing system that focuses on problem text predicts problems' difficulty and uncertainty and guides a reconfigurable sampling mechanism to manage the breadth of generation. Three specialized experts create candidate responses, which are modified during multiple correction and finalization phases. A neural verifier assesses the correctness of responses, while a clustering-based aggregation technique identifies the final candidate answer based on a combination of consensus and answer quality. When evaluated on the GSM8K dataset, AMR achieved 75.28% accuracy while only using the original training data. This result outperformed the majority of comparable 7B models that were trained on synthetic data. This showcases that models using difficulty-based routing and uncertainty-driven aggregation are efficient and effective in improving math reasoning models' robustness.
Figures & tables
| Difficulty | Correct | Total | Accuracy |
| Easy (predicted) | 657 | 795 | 82.6% |
| Hard (predicted) | 336 | 524 | 64.1% |
| Easy (gold) | 636 | 742 | 85.7% |
| Hard (gold) | 357 | 577 | 61.9% |
| Overall | 993 | 1319 | 75.28% |
| Model | Base Model | Size | GSM8K (%) |
|---|---|---|---|
| LLaMA-2 [ 13 ] | – | 7B | 14.6 |
| LLaMA-2 | – | 13B | 28.7 |
| LLaMA-2 | – | 34B | 42.2 |
| LLaMA-2 | – | 70B | 56.8 |
| Mistral [ 5 ] | – | 7B | 52.2 |
| Mistral | – | 8 7B | 58.4 |
| Model | Base Model | Size | Synthetic Data | GSM8K(%) |
|---|---|---|---|---|
| MetaMath [ 17 ] | LLaMA-2 | 7B | MetaMathQA (395K) | 66.7 |
| MetaMath | LLaMA-2 | 13B | MetaMathQA (395K) | 70.8 |
| MetaMath | LLaMA-2 | 70B | MetaMathQA (395K) | 82.1 |
| MetaMath-Mistral | Mistral | 7B | MetaMathQA (395K) | 77.8 |
| WizardMath [ 9 ] | LLaMA-2 | 7B | Evol‑Instruct | 54.9 |
| WizardMath | LLaMA-2 | 13B | Evol‑Instruct | 63.9 |
LLM Mathematical Reasoning
Large language model mathematical reasoning studies how AI systems solve, explain, verify, and generalize across mathematical problems and proofs.
See all 270 papers