Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

DiffAdapt: Difficulty-Adaptive Reasoning for Token-Efficient LLM Inference

About

Recent reasoning Large Language Models (LLMs) demonstrate remarkable problem-solving abilities but often generate long thinking traces whose utility is unclear. Our work aims to improve their efficiency, enabling them to reach high performance without overthinking. First, we analyze the entropy of token probabilities in reasoning traces. Across three models, we observe a consistent U-shaped entropy pattern: high entropy on easy problems despite high accuracy, low entropy on problems with medium difficulty, and high entropy on hard problems reflecting uncertainty. Specifically, we notice 22--25\% entropy reduction from easy to medium difficulty regions, suggesting an {overthinking} phenomenon on easy instances. Building on these insights, we introduce \textbf{DiffAdapt}, a lightweight framework that selects Easy/Normal/Hard inference strategies per question based on their difficulty and reasoning trace entropy. Each inference strategy consists of a fixed prompt, temperature and maximum token length. In contrast to existing efficiency optimization methods, our approach does not fine-tune base LLM but a small probe that classifies LLM's final hidden state, allowing inexpensive adaptation. We comprehensively evaluate our method on five models and eight benchmarks. Our method achieves comparable or improved accuracy while reducing token usage by up to 22.4\%, establishing a practical path toward compute-efficient reasoning.

Xiang Liu, Xuming Hu, Xiaowen Chu, Eunsol Choi• 2025

Related benchmarks

TaskDatasetResultRank
General ReasoningMMLU-Pro
Accuracy39.9
201
Mathematical ReasoningMATH500
Accuracy68.3
86
Multi-step Narrative ReasoningMuSR
Accuracy65.86
22
Out-of-Domain Reasoning AggregationOOD Average
Accuracy54.2
22
Scientific Question AnsweringGPQA Diamond
Accuracy (ACC)48.48
22
Logical reasoningLSAT-AR
Accuracy48.26
22
Mathematical ReasoningGSM8K
Accuracy78.6
8
Mathematical Reasoning8 reasoning benchmarks (including GSM8K, MATH500, AIME 24, AIME 25, and OlympiadBench) (test)
Token Savings22.4
5
Mathematical ReasoningOlympiadBench (first 40 problems)
Time (minutes)10
3
Showing 9 of 9 rows

Other info

Follow for update