AdaptR1: Reinforcement Learning Based Adaptive Interleaved Thinking in Multi-hop Question Answering
About
Large Language Models (LLMs) have achieved remarkable performance in complex reasoning tasks through Chain-of-Thought (CoT) prompting. However, this approach often leads to ``over-thinking,'' where models generate unnecessarily long reasoning traces for simple queries and incur avoidable inference cost. While recent work has explored adaptive reasoning, existing methods typically make a single query-level decision about whether to reason. This overlooks the dynamic nature of multi-step tasks, where the need for explicit reasoning varies across intermediate stages. To address this limitation, we introduce AdaptR1, a Reinforcement Learning (RL) based framework for adaptive interleaved thinking in multi-hop Question Answering (QA). Unlike previous approaches that require Supervised Fine-Tuning (SFT) for cold-start initialization, AdaptR1 uses a fully RL-based strategy with a quality-gated efficiency reward to dynamically allocate reasoning budgets at each step. Under the Graph-R1 setting, AdaptR1 reduces average think tokens by 69.71\%, with a 90.35\% reduction on HotpotQA, while maintaining performance comparable to or better than standard baselines. Furthermore, our analysis reveals that overthinking in multi-hop reasoning is not uniformly distributed but occurs predominantly during the initial planning stages, highlighting the effectiveness of step-wise adaptive budget allocation.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Multi-hop Question Answering | 2WikiMultiHopQA (test) | EM61.72 | 226 | |
| Multi-hop Question Answering | HotpotQA fixed (test) | EM57.81 | 16 | |
| Multi-hop Question Answering | Musique fixed split (test) | Exact Match (EM)39.84 | 16 | |
| Multi-hop Question Answering | Multi-hop QA Datasets calculated average (test) | EM51.43 | 16 | |
| Multi-hop Question Answering | Natural Questions (NQ) fixed split (test) | Exact Match (EM)35.94 | 16 | |
| Multi-hop Question Answering | PopQA fixed split (test) | Exact Match (EM)49.22 | 16 | |
| Multi-hop Question Answering | TriviaQA fixed split (test) | EM64.06 | 16 |