Authors: Zhehao Zhou, Tongrui Su, Shengyu Zhu, Da Li, Zeyu Liu, Jun Zhu, Xueqi Cheng
Organizations: Beijing Institute of Technology · Key Laboratory of AI Safety, Institute of Computing Technology, Chinese Academy of Sciences · Kuaishou Technology
Latent Chain-of-Thought (CoT) reasoning improves efficiency by shifting intermediate computation into continuous hidden states. However, fully latent methods often struggle on precision-critical tasks such as mathematical reasoning, while existing hybrid approaches retain substantial token-generation overhead. We introduce VITAL, an importance-guided hybrid reasoning framework that selectively verbalizes informative tokens as reasoning anchors while keeping most computation latent. A lightweight gating head dynamically selects between explicit token generation and direct hidden-state feedback. VITAL constructs hybrid trajectories from explicit CoT using token- importance estimation and trains the model through curriculum-based supervised training, followed by reinforcement learning with a reward that balances correctness and conciseness. Experiments show that VITAL consistently out- performs existing fully latent reasoning methods, achieving relative average accuracy gains of 9.6% on GPT-2 and 5.1% on Llama-3.2-1B at comparable reasoning lengths. Compared with hybrid baselines, VITAL achieves higher reasoning accuracy with substantially shorter reasoning trajectories on Qwen3-1.7B.
Figures & tables
Figure 1: Architecture of VITAL. A lightweight gating head dynamically selects between explicit token generation and latent computation at each reasoning step.
Figure 2: Curriculum-based supervised training for VITAL. We construct hybrid reasoning trajectories from explicit CoT by identifying high-importance tokens as the anchors.
Method
In-domain
Out-of-domain
Overall
GSM8K
GSM8K-Hard
Multi-Arith
SVAMP
Average
Acc.(%)
#Tokens
Acc.(%)
#Tokens
Acc.(%)
#Tokens
Acc.(%)
#Tokens
Acc.(%)
#Tokens
GPT-2
CoT-SFT
42.7
27.6
9.0
39.1
85.0
19.6
41.6
17.6
45.2
25.9
Coconut
34.1
8.0
6.9
8.0
80.7
8.0
32.0
8.0
38.4
8.0
CODI
24.6
8.0
5.0
8.0
45.0
8.0
26.9
8.0
25.4
8.0
Table 1: Main results on in-domain and out-of-domain benchmarks. Performance is measured by accuracy (Acc., %) and average reasoning length (# Tokens).
Method
GSM8K-NL
MATH
MATH500
ARC-C
Average
Acc.(%)
#Tokens
Acc.(%)
#Tokens
Acc.(%)
#Tokens
Acc.(%)
#Tokens
Acc.(%)
#Tokens
Qwen3-1.7B
CoT-SFT
69.5
81.8
35.6
321.1
35.0
302.7
57.6
541.7
49.4
311.8
HRPO
64.5
110.2
27.6
763.2
26.0
765.1
53.7
464.2
43.0
525.7
Token Assorted
60.1
73.9
26.9
183.4
27.6
253.4
66.0
88.3
45.2
149.8
VITAL (Ours)
53.5
32.5
23.4
136.6
26.2
120.1
63.2
67.8
41.6
89.3
Table 2: Results on Qwen3-1.7B. Acc.(%) denotes final-answer accuracy, and #Tokens denotes average reasoning length, including both explicit tokens and latent steps.
Figure 3: Token-category distributions of full CoT and VITAL on GSM8K-NL (Qwen3-1.7B).
Figure 4: Performance degradation under noise injection. We inject Gaussian noise N(0,σ2Id) to the hidden states during reasoning and report the accuracy result of GSM8K on Llama-3.2-1B.
Method
GSM8K-Aug
GSM8K-Hard
MultiArith
SVAMP
Avg.
SFT
69.1
36.5
98.9
77.5
71.3
Coconut
60.2
25.3
97.2
65.4
62.0
Lowest
59.2
25.1
97.8
66.1
62.1
Random
62.8
28.2
97.8
67.8
64.1
VITAL
64.1
30.1
98.3
69.2
65.4
Table 3: Accuracy (%) on four benchmarks. VITAL uses highest-importance tokens as reasoning anchors.
Figure 5: Ablation on the top- γ ratio. We vary the proportion of explicit anchors and evaluate its effect.
Appendix figures & tables6 assets
Supplementary material from the paper’s appendix.
Appendix
Training Dataset
Num. Data
Avg. CoT Tokens
GSM8K-Aug
385,620
20.3
Appendix
Table 4: Training data statistics.
Benchmark
Type
Data Size
GSM8K-Aug
In-domain
1,319
GSM8K-Hard
Out-of-domain
1,319
SVAMP
Out-of-domain
1,000
MultiArith
Out-of-domain
500
Appendix
Table 5: Evaluation benchmark statistics.
Hyperparameter
Value
Anchor ratio γ
0.2
Latent placeholders klatent
1
Gating threshold τ
0.5
Loss weight λ
0.1
Curriculum stages
3
Epochs per stage
3
Appendix
Table 6: Hyperparameters of VITAL.
Method
Model
LR
Epochs
Coconut
GPT-2
1×10−4
20
Llama-3.2-1B
8×10−4
20
Llama-3.2-3B
8×10−4
20
CODI
GPT-2
3×10−3
40
Llama-3.2-1B
8×10−4
10
Llama-3.2-3B
8×10−4
10
Appendix
Table 7: Configurations of the baselines. All models are trained for the specified number of epochs or 50 hours, whichever is reached first.
Figure 6: An example of the constructed hybrid CoT.
Method
In-domain
Out-of-domain
Overall
GSM8K-Aug
GSM8K-Hard
MultiArith
SVAMP
Average
Acc. (%)
# Tokens
Acc. (%)
# Tokens
Acc. (%)
# Tokens
Acc. (%)
# Tokens
Acc. (%)
# Tokens
CoT-SFT
55.4
28.5
12.7
35.7
96.7
17.6
61.0
14.5
56.5
24.0
Coconut
36.7
8.0
8.7
8.0
90.5
8.0
37.8
8.0
43.4
8.0
CODI
47.2
8.0
11.2
8.0
93.2
8.0
56.5
8.0
52.0
8.0
CoLaR
45.4
13.5
10.1
40.4
96.3
7.3
51.8
6.0
51.1
16.8
Appendix
Table 8: Experimental results on Llama-3.2-3B. Accuracy and average reasoning chain length are reported across in-domain and out-of-domain benchmarks.
Explicit chain-of-thought (CoT) reasoning substantially improves the reasoning ability of large language models (LLMs), but incurs high inference cost due to lengthy autoregressive traces. Existing latent reasoning methods offer a promis- ing alternative, yet they often treat reasoning as uniformly compressible, causing precision-critical intermediate steps to be overly compressed and thereby degrading reasoning accuracy. In this work, we propose Selective Latent Thinking (SLT), a reasoning paradigm that selectively compresses redundant reasoning spans into latent representations while preserving precision-critical spans as explicit CoT within the same reasoning trajectory. Our primary instantiation uses a lightweight decoder to anticipate a short upcoming reasoning span and applies confidence- based gating to determine the longest span that can be reliably compressed; the resulting selective compression policy is further optimized with trajectory-level reinforcement learning to balance answer correctness against reasoning cost. Ex- periments across four mathematical reasoning benchmarks show that which spans are compressed is critical: with the same compressor, compressing arbitrary spans reduces the GSM-Hard accuracy of Qwen3-4B from 54.7% to 17.3%, whereas keeping numerical tokens explicit retains 51.8%. SLT reduces explicit reasoning length by 58.4% with only a 2.8-point accuracy drop compared to explicit CoT on Llama-3.2-1B. Notably, the principle extends beyond this design: instantiated within existing latent reasoning models via a learned compression gate, it improves accuracy at equal reasoning length (+2.6 points over Latent-SFT) and at shorter length (+5.6 points over Latent-GRPO with 13% fewer tokens).
Hui Xie, Jie Liu, Ziyue Qiao +1
Eindhoven University of Technology, Netherlands · School of Computing and Information Technology, Great Bay University, China
Chain-of-thought (CoT) prompting improves reasoning in large language models (LLMs) by externalizing intermediate computation as discrete text tokens, but this textual interface also introduces redundancy and inference overhead. Latent reasoning offers a promising alternative by carrying part of the computation in continuous representations. However, existing methods typically predefine when latent computation is invoked and how it is allocated during decoding, leaving a key problem unresolved: when to invoke latent computation, what type of computation to perform, and how much budget to allocate. We propose Typed Latent Reasoning (Tyler), a typed and budget-aware framework for latent reasoning during autoregressive decoding. Tyler learns a policy that, at each decoding step, chooses between emitting a text token and switching to a latent computation module specialized for a particular reasoning function. Once invoked, an operator maps the current reasoning state into latent tokens that support global planning, local state updates, or reusable procedural abstraction. Across extensive experiments on three backbone LLMs, Tyler improves accuracy by up to 14.49 points over CoT and by up to 4.30 points over the strongest competing baseline. It further generalizes across diverse reasoning domains and achieves the best final-stage performance with the lowest forgetting.
While long, explicit chains-of-thought (CoT) have proven effective on complex reasoning tasks, they are costly to generate during inference. Non-verbal reasoning methods have emerged with shorter generation lengths by leveraging continuous representations, yet their performance lags behind verbalized CoT. We propose Abstract Chain-of-Thought, a discrete latent reasoning post-training mechanism in which the language model produces a short sequence of tokens from a reserved vocabulary in lieu of a natural language CoT, before generating a response. To make previously unseen ''abstract'' tokens useful, we introduce a policy iteration-style warm-up loop that alternates between (i.) bottlenecking from a verbal CoT via masking and performing supervised fine-tuning, and (ii.) self-distillation by training the model to generate abstract tokens from the prompt alone via constrained decoding with the codebook. After warm-up, we optimize the generation of abstract sequences with warm-started reinforcement learning under constrained decoding. Abstract-CoT achieves up to 11.6× fewer reasoning tokens while demonstrating comparable performance across mathematical reasoning, instruction-following, and multi-hop reasoning, and generalizes across language model families. We also find an emergent power law distribution over the abstract vocabulary, akin to those seen in natural language, that evolves across the training phases. Our findings highlight the potential for post-training latent reasoning mechanisms that enable efficient inference through a learned abstract reasoning language.