PRISM: A Dual View of LLM Reasoning through Semantic Flow and Latent Computation
Organizations: Rice University · Stony Brook University
Abstract
Large language models (LLMs) solve complex problems by generating multi-step reasoning traces. Yet these traces are typically analyzed from only one of two perspectives: the sequence of tokens across different reasoning steps in the generated text, or the hidden-state vectors across model layers within one step. We introduce PRISM (Probabilistic Reasoning Inspection through Semantic and Implicit Modeling), a framework and diagnostic tool for jointly analyzing both levels, providing a unified view of how reasoning evolves across steps and layers. Across multiple reasoning models and benchmarks, PRISM uncovers systematic patterns in the reasoning process, showing that failed trajectories are more likely to become trapped in unproductive verification loops and further diverge into distinct modes such as overthinking and premature commitment, which behave differently once a candidate answer is reached. It further reveals how prompting reshapes reasoning behavior beyond aggregate accuracy by altering both semantic transitions and internal computational patterns. By modeling reasoning trajectories as structured processes, PRISM makes these behaviors observable and analyzable rather than relying solely on final-task accuracy. Taken together, these insights position PRISM as a practical tool for analyzing and diagnosing reasoning processes in LLMs.
Figures & tables
| From / To | FA | SR | AC | UV |
|---|---|---|---|---|
| FA | 0.466 | 0.076 | 0.131 | 0.326 |
| SR | 0.038 | 0.419 | 0.445 | 0.098 |
| AC | 0.060 | 0.079 | 0.715 | 0.147 |
| UV | 0.124 | 0.119 | 0.320 | 0.438 |
| Start Distribution | 0.000 | 0.982 | 0.007 | 0.011 |
| Stationary Dist. | 0.128 | 0.133 | 0.501 | 0.238 |
| From SR | FA | SR | AC | UV | From AC | FA | SR | AC | UV |
|---|---|---|---|---|---|---|---|---|---|
| SR R0 | 5% | 35% | 46% | 14% | AC R0 | 0% | 0% | 87% | 13% |
| SR R1 | 0% | 80% | 9% | 11% | AC R1 | 0% | 0% | 99% | 1% |
| SR R2 | 0% | 0% | 2% | 98% | AC R2 | 0% | 0% | 0% | 100% |
| SR R3 | 4% | 69% | 25% | 3% | AC R3 | 5% | 12% | 61% | 22% |
| SR R4 | 6% | 22% | 62% | 9% | AC R4 | 23% | 13% | 46% | 18% |
| SR R5 | 0% | 99% | 0% | 0% | AC R5 | 4% | 8% | 68% | 19% |
| Correct Incorrect | Correct Long Fail | Correct Short Fail | ||||||||||
|---|---|---|---|---|---|---|---|---|---|---|---|---|
| From / To | FA | SR | AC | UV | FA | SR | AC | UV | FA | SR | AC | UV |
| FA | 3.4% | +2.2% | 0.0% | +1.2% | 1.2% | +2.3% | 2.2% | +1.1% | 0.7% | 2.6% | +3.0% | +0.2% |
| SR | 0.9% | 0.9% | +5.1% | 3.3% | 0.6% | +0.6% | +3.4% | 3.5% | 1.7% | 5.3% | +11.4% | 4.3% |
| AC | +0.4% | 1.5% | +5.1% | 4.1% | +1.4% | 1.6% | +4.3% | 4.0% | 3.7% | 2.4% | +14.0% | 8.0% |
| UV | 0.9% | 0.5% | +5.9% | 4.5% | 0.6% | 0.6% | +5.6% | 4.4% | +1.7% | 0.7% | +5.3% | 6.3% |
| Stratos | OpenThinker | Qwen | Nemotron | ||||||||||||||
| MATH500 | GPQA-D | AIME24 | WebInstruct | MATH500 | GPQA-D | AIME24 | WebInstruct | MATH500 | GPQA-D | AIME24 | WebInstruct | MATH500 | GPQA-D | WebInstruct | All | ||
| Implicit Stage divergence (long fail vs. short fail) | |||||||||||||||||
| FA | 2.36 | 0.91 | 1.31 | 2.05 | 4.63 | 2.53 | 3.30 | 5.25 | 1.52 | 0.65 | 2.40 | 1.04 | 3.15 | 1.47 | 1.46 | 2.27 | |
| SR | 1.19 | 2.06 | 1.56 | 1.30 | 2.31 | 0.74 | 1.14 | 1.74 | 1.37 | 0.93 | 2.21 | 1.14 | 1.99 | 1.38 | 1.75 | 1.52 | |
| AC | 0.44 | 1.34 | 1.10 | 0.65 | 0.61 | 1.45 | 0.89 | 0.75 | 0.92 | 0.73 | 2.16 | 0.49 | 1.09 | 1.71 | 0.73 | 1.00 | |
| UV | 1.41 | 1.10 | 1.51 | 1.55 | 1.35 | 2.34 | 1.63 | 1.14 | 1.13 | 0.53 | 1.03 | 1.01 | 1.92 | 0.55 | 0.90 | 1.27 | |
| 1st-order TransMat | Implicit vs. P1 | ||||||||||||||
| Category | P1 (Baseline) | P2 (Concise) | P3 (Explore) | Category | P2 | P3 | |||||||||
| FA | SR | AC | UV | FA | SR | AC | UV | FA | SR | AC | UV | ||||
| FA | 0.444 | 0.068 | 0.142 | 0.345 | 0.464 | 0.069 | 0.169 | 0.299 | 0.411 | 0.085 | 0.184 | 0.320 | FA | 0.265 | 0.272 |
| SR | 0.036 | 0.383 | 0.434 | 0.147 | 0.043 | 0.372 | 0.465 | 0.120 | 0.038 | 0.400 | 0.424 | 0.137 | SR | 0.173 | 0.218 |
| AC | 0.055 | 0.104 | 0.636 | 0.205 | 0.052 | 0.115 | 0.549 | 0.284 | 0.042 | 0.091 | 0.536 | 0.331 | AC | 0.193 | 0.483 |
| UV | 0.050 | 0.141 | 0.270 | 0.539 | 0.051 | 0.150 | 0.284 | 0.515 | 0.045 | 0.109 | 0.373 | 0.473 | UV | 0.292 | 0.339 |
Appendix figures & tables18 assets
Supplementary material from the paper’s appendix.
Appendix
| Dataset | Split | Size | Max Tokens | License |
|---|---|---|---|---|
| MATH-500 ( Lightman et al., 2023 ) | test | 500 | 32768 | MIT |
| GPQA-Diamond ( Rein et al., 2024 ) | test | 198 | 32768 | Apache 2.0 |
| WebInstruct-Verified ( Ma et al., 2025 ) | test | 288 | 32768 | Apache 2.0 |
| AIME-2024 ( HuggingFaceH4, 2024 ) | train | 30 | 32768 | Apache 2.0 |
| Model | Dataset | Acc | AvgTok | AvgStep | LongFail | ShortFail | FA n | SR n | AC n | UV n | UNK n | |
|---|---|---|---|---|---|---|---|---|---|---|---|---|
| Stratos | AIME24 | 30 | ||||||||||
| Stratos | GPQA Diamond | 198 | ||||||||||
| Stratos | MATH500 | 500 | ||||||||||
| Stratos | WebInstruct | 288 | ||||||||||
| OpenThinker | AIME24 | 30 | ||||||||||
| OpenThinker | GPQA Diamond | 198 |
| Context | FA | SR | AC | UV |
|---|---|---|---|---|
| FA FA | 0.770 0.199 | 0.041 0.133 | 0.025 0.057 | 0.163 0.108 |
| FA SR | 0.244 0.117 | 0.515 0.210 | 0.180 0.211 | 0.060 0.056 |
| FA AC | 0.227 0.101 | 0.077 0.046 | 0.510 0.127 | 0.186 0.066 |
| FA UV | 0.506 0.234 | 0.067 0.036 | 0.178 0.101 | 0.249 0.142 |
| SR FA | 0.558 0.212 | 0.100 0.052 | 0.171 0.141 | 0.171 0.114 |
| SR SR | 0.015 0.008 | 0.534 0.061 | 0.353 0.059 | 0.098 0.029 |
| From To | FA | SR | AC | UV |
|---|---|---|---|---|
| FA | ||||
| SR | ||||
| AC | ||||
| UV |
| From To | FA | SR | AC | UV |
|---|---|---|---|---|
| FA | ||||
| SR | ||||
| AC | ||||
| UV |
| From To | FA | SR | AC | UV |
|---|---|---|---|---|
| FA | ||||
| SR | ||||
| AC | ||||
| UV |
| From To | FA | SR | AC | UV |
|---|---|---|---|---|
| FA | ||||
| SR | ||||
| AC | ||||
| UV |
| Metric | Category | Value |
|---|---|---|
| Avg Path Length | Correct | |
| Incorrect | ||
| Stationary Distribution ( ) | FA | |
| SR | ||
| AC | ||
| UV |
| Stratos | OpenThinker | Qwen | Nemotron | ||||||||||||||
| MATH500 | GPQA-D | AIME24 | WebInstruct | MATH500 | GPQA-D | AIME24 | WebInstruct | MATH500 | GPQA-D | AIME24 | WebInstruct | MATH500 | GPQA-D | WebInstruct | All | ||
| Regime divergence (long fail vs. short fail, pooled std) | |||||||||||||||||
| AC | 1.00 | ||||||||||||||||
| UV | 1.27 | ||||||||||||||||
| FA | 2.27 | ||||||||||||||||
| SR | 1.52 | ||||||||||||||||
| Source: FA | Source: SR | Source: AC | Source: UV | |||||||||||||
| Regime | FA | SR | AC | UV | FA | SR | AC | UV | FA | SR | AC | UV | FA | SR | AC | UV |
| R0 | 43 | 8 | 18 | 32 | 5 | 35 | 46 | 14 | 0 | 0 | 87 | 13 | 0 | 0 | 0 | 100 |
| R1 | 0 | 0 | 0 | 0 | 0 | 80 | 9 | 11 | 0 | 0 | 99 | 1 | 12 | 49 | 17 | 23 |
| R2 | 71 | 6 | 13 | 9 | 0 | 0 | 0 | 0 | 0 | 0 | 0 | 100 | 7 | 15 | 28 | 50 |
| R3 | 0 | 0 | 0 | 0 | 4 | 69 | 25 | 3 | 5 | 12 | 61 | 22 | 4 | 14 | 36 | 47 |
| R4 | 38 | 7 | 21 | 34 | 6 | 22 | 62 | 9 | 23 | 13 | 46 | 18 | 0 | 14 | 25 | 62 |
LLM Reasoning
Large language model reasoning studies how models solve multi-step problems, verify intermediate steps, and use computation or tools reliably.
See all 902 papers