When2Think:面向高效混合推理模型的难度感知长度控制

HuggingFace Daily Papers(社区热门论文)·2026-09-17 08:00·1天前
AI 导读

When2Think 是一个混合推理后训练框架,通过实例级难度感知控制(IDAC)按题目难度动态分配计算量,让模型在简单题上直接作答、难题上保留长推理。在 AIME24 上 Pass@3 相对基座模型提升 10.0%、token 用量减少 27.9%;在 AIME25 上达到 40.0% Pass@3,优于压缩与纯路由基线。

HuggingFace Daily Papers(社区热门论文)
40AI 编辑部评分,满分 100

When2Think:面向高效混合推理模型的难度感知长度控制

2026-09-17 08:00· 1天前
AI 导读

When2Think 是一个混合推理后训练框架,通过实例级难度感知控制(IDAC)按题目难度动态分配计算量,让模型在简单题上直接作答、难题上保留长推理。在 AIME24 上 Pass@3 相对基座模型提升 10.0%、token 用量减少 27.9%;在 AIME25 上达到 40.0% Pass@3,优于压缩与纯路由基线。

Large Reasoning Models (LRMs) achieve strong performance on complex tasks but exhibit systematic inefficiency: they often overthink easy problems and underthink hard ones. Existing approaches based on uniform length penalties or rigid routing incur an efficiency tax, trading reduced computation on easy instances for accuracy loss on hard instances. We formulate efficient reasoning as an instance-adaptive computation allocation problem and propose When2Think, a post-training framework for hybrid reasoning that dynamically allocates computation based on problem difficulty.

Our method introduces Instance-level Difficulty-Aware Control (IDAC), a reward-shaping mechanism that leverages pre-computed reference statistics (accuracy and token usage) to regulate reasoning depth. Combined with verifier-based rewards and batch-wise standardized advantages, IDAC enables stable critic-free optimization without learned reward models or online reference-model queries. When2Think encourages direct answering on easy instances while preserving extended reasoning on hard instances, thereby learning when to use System 1 (NoThink) versus System 2 (Think).

Experiments on mathematical benchmarks demonstrate improved accuracy-efficiency trade-offs: on AIME24, Pass@3 increases by 10.0% while token usage is reduced by 27.9% relative to the base model, and on AIME25, When2Think achieves 40.0% Pass@3, outperforming compression and routing-only baselines.

来源:HuggingFace Daily Papers(社区热门论文)· arxiv.org