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

Efficiently Scaling LLM Reasoning with Certaindex

About

Test-time reasoning algorithms such as chain-of-thought, self-consistency, and MCTS enhance LLM problem-solving but can wastefully generate many tokens without improving accuracy. At the same time, we observe that these algorithms exhibit answer stabilization: their intermediate solutions often cease to change after a certain point, and further investment of compute does not change their final answer. To quantify this phenomenon, we introduce Certaindex, an algorithm-agnostic metric measuring this evolving stability, signaling when further computation is unlikely to alter the final result. Certaindex is lightweight, can accelerate reasoning program inference via early exit, and further enables dynamic token allocation, gang scheduling, and many opportunities when integrated with real-world LLM serving systems. To quantify real-world benefits, we built Certaindex as a scheduler into Dynasor, our reasoning-aware LLM serving system, and demonstrate up to 50% compute savings and 3.3x higher throughput in real workloads with no accuracy drop. Our code is available at https://github.com/hao-ai-lab/Dynasor.git

Yichao Fu, Junda Chen, Siqi Zhu, Zheyu Fu, Zhongdongming Dai, Yonghao Zhuang, Yian Ma, Aurick Qiao, Tajana Rosing, Ion Stoica, Hao Zhang• 2024

Related benchmarks

TaskDatasetResultRank
Code GenerationHumanEval--
1036
Mathematical ReasoningMATH500
Accuracy88.3
50
Mathematical ReasoningAIME 25
Accuracy87.6
45
Mathematical ReasoningMATH 500
Accuracy88.1
37
Scientific ReasoningGPQA
Accuracy51.8
28
Mathematical ReasoningMATH 500
Pass@1 Rate92
26
General ReasoningOverall MATH-500 AIME25 HumanEval GPQA
Accuracy79.9
24
Mathematical ReasoningAIME 25
Pass@1 Accuracy41
18
Mathematical ReasoningAIME 22-24
Pass@145.1
12
Mathematical ReasoningAMC 23
Pass@175
12
Showing 10 of 11 rows

Other info

Follow for update