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

Invariant Reasoning Directions in Latent Trajectories of Language Models

About

Latent reasoning models perform multi-step inference directly in hidden-state space, yet the structure of these latent reasoning trajectories remains poorly understood. We show that contrastive refinement signals between stronger and weaker reasoning trajectories exhibit a highly concentrated low-rank structure, while unconstrained latent updates remain sensitive to paraphrases, checkpoint choice, and trajectory perturbations. These observations suggest that latent reasoning trajectories contain stable invariant directions mixed with unstable instance-specific variation. We introduce \textbf{Trajectory-Invariant Latent Refinement (TILR)}, a training-free intervention framework for identifying and manipulating stable reasoning directions in latent space. TILR first learns a low-rank invariant subspace from contrastive trajectory differences across inputs, then constrains latent interventions to this subspace while suppressing poorly aligned updates through an adaptive alignment gate. Across six reasoning benchmarks, we find that a small number of latent directions explain most variation between strong and weak reasoning trajectories. Interventions on these directions causally improve reasoning consistency and reduce trajectory instability under paraphrases and perturbations. TILR improves answer consistency under paraphrase by ~10% and reduces latent trajectory variance by up to $50\%$ while preserving reasoning accuracy. These results support a geometric view of latent reasoning in which transferable reasoning behavior emerges from stable low-dimensional structure within hidden-state trajectories.

Arun Vignesh Malarkkan, Manan Roy Choudhury, Utkarsh Byahut, Yash Ravindra Charde, Vivek Gupta, Yanjie Fu• 2026

Related benchmarks

TaskDatasetResultRank
Mathematical ReasoningAQUA-RAT
Accuracy32.3
183
Mathematical ReasoningGSM-PLUS--
162
ReasoningGSM8K--
111
ReasoningStrategyQA
Accuracy65.5
58
Mathematical ReasoningGSM8K
Accuracy0.605
22
Mathematical ReasoningMathQA
Exact Match42.1
12
Multi-step ReasoningSVAMP
Accuracy33.9
11
Mathematical ReasoningMathQA
Exact Match Accuracy60.1
6
Mathematical ReasoningAQUA-RAT
Accuracy (Exact Match)47.6
6
Mathematical ReasoningSVAMP
Exact-match Accuracy55.9
6
Showing 10 of 11 rows

Other info

Follow for update