Diagnosing Multi-step Reasoning Failures in Black-box LLMs via Stepwise Confidence Attribution
On the map: LLM reasoning → Test-time compute and efficient thinking
Abstract (arXiv)
Large Language Models have achieved strong performance on reasoning tasks with objective answers by generating step-by-step solutions, but diagnosing where a multi-step reasoning trace might fail remains difficult. Confidence estimation offers a diagnostic signal, yet existing methods are restricted to final answers or require internal model access. In this paper, we introduce Stepwise Confidence Attribution (SCA), a framework for closed-source LLMs that assigns step-level confidence based only on generated reasoning traces. SCA applies the Information Bottleneck principle: steps aligning with consensus structures across correct solutions receive high confidence, while deviations are flagged as potentially erroneous. We propose two complementary methods: (1) NIBS, a non-parametric IB approach measuring consistency without graph structures, and (2) GIBS, a graph-based IB model that learns subgraphs through a differentiable mask to capture logical variability. Extensive experiments on mathematical reasoning and multi-hop question answering show that SCA reliably identifies low-confidence steps strongly correlated with reasoning errors. Moreover, using step-level confidence to guide self-correction improves the correction success rate by up to 13.5% over answer-level feedback.
Key facts. Cited by 2 papers in the map (data through 14 Oct 2026). Placed in Test-time compute and efficient thinking, one of 17 directions in LLM reasoning. Builds on 47 of its 55 references that are in the map. First posted on arXiv 19 May 2026.
Builds on
47 of 55 references.
Closest ancestors · 5
Same direction, newest first
- 2025Trustworthy Reasoning: Evaluating and Enhancing Factual Accuracy in LLM Intermediate Thought Processes25 Jul
- 2025Uncertainty Quantification and Confidence Calibration in Large Language Models: A Survey20 Mar
- 2024Learning From Correctness Without Prompting Makes LLM Efficient Reasoner28 Mar
- 2024On the Self-Verification Limitations of Large Language Models on Reasoning and Planning Tasks12 Feb
- 2022Language Models (Mostly) Know What They Know11 Jul
Other references · 42
- 2024The Llama 3 Herd of Modelscited by 14,715
- 2022Chain-of-Thought Prompting Elicits Reasoning in Large Language Modelscited by 12,459
- 2021Training Verifiers to Solve Math Word Problemscited by 10,109
- 2022Self-Consistency Improves Chain of Thought Reasoning in Language Modelscited by 5,708
- 2021Measuring Mathematical Problem Solving With the MATH Datasetcited by 5,371
- 2023Let's Verify Step by Stepcited by 3,789
- 2023G-Eval: NLG Evaluation using GPT-4 with Better Human Alignmentcited by 2,276
- 2020DeBERTa: Decoding-enhanced BERT with Disentangled Attentioncited by 2,181
- 2023Teaching Large Language Models to Self-Debugcited by 988
- 2023Math-Shepherd: Verify and Reinforce LLMs Step-by-step without Human Annotationscited by 928
- 2022Solving math word problems with process- and outcome-based feedbackcited by 901
- 2023Graph of Thoughts: Solving Elaborate Problems with Large Language Modelscited by 886
30 more
- 2017Selective Classification for Deep Neural Networkscited by 852
- 2023Semantic Uncertainty: Linguistic Invariances for Uncertainty Estimation in Natural Language Generationcited by 819
- 2019MathQA: Towards Interpretable Math Word Problem Solving with Operation-Based Formalismscited by 662
- 2024Generative Verifiers: Reward Modeling as Next-Token Predictioncited by 480
- 2024LLMs-as-Judges: A Comprehensive Survey on LLM-based Evaluation Methodscited by 465
- 2022Large Language Models are Better Reasoners with Self-Verificationcited by 307
- 2023Generating with Confidence: Uncertainty Quantification for Black-box Large Language Modelscited by 293
- 2024ProcessBench: Identifying Process Errors in Mathematical Reasoningcited by 292
- 2024Rewarding Progress: Scaling Automated Process Verifiers for LLM Reasoningcited by 279
- 2022ROSCOE: A Suite of Metrics for Scoring Step-by-Step Reasoningcited by 177
- 2025Phi-4-reasoning Technical Reportcited by 151
- 2023Factuality Challenges in the Era of Large Language Modelscited by 126
- 2024Limitations of the LLM-as-a-Judge Approach for Evaluating LLM Outputs in Expert Knowledge Taskscited by 106
- 2024LLM-based NLG Evaluation: Current Status and Challengescited by 73
- 2023Selectively Answering Ambiguous Questionscited by 66
- 2024A Chain-of-Thought Is as Strong as Its Weakest Link: A Benchmark for Verifiers of Reasoning Chainscited by 65
- 2024Auto-Arena: Automating LLM Evaluations with Agent Peer Battles and Committee Discussionscited by 42
- 2024MoreHopQA: More Than Multi-hop Reasoningcited by 37
- 2025Understanding the Uncertainty of LLM Explanations: A Perspective Based on Reasoning Topologycited by 25
- 2024LLM Uncertainty Quantification through Directional Entailment Graph and Claim Level Response Augmentationcited by 23
- 2025Premise-Augmented Reasoning Chains Improve Error Identification in Math reasoning with LLMscited by 22
- 2025Uncertainty-Aware Step-wise Verification with Generative Reward Modelscited by 22
- 2023GraphReason: Enhancing Reasoning Capabilities of Large Language Models through A Graph-Based Verification Approachcited by 21
- 2025Uncertainty Quantification of Large Language Models through Multi-Dimensional Responsescited by 18
- 2024Contextualized Sequence Likelihood: Enhanced Confidence Scores for Natural Language Generationcited by 16
- 2025Adaptive Graph of Thoughts: Test-Time Adaptive Reasoning Unifying Chain, Tree, and Graph Structurescited by 16
- 2025Graph of Verification: Structured Verification of LLM Reasoning with Directed Acyclic Graphscited by 8
- 2026Every Response Counts: Quantifying Uncertainty of LLM-based Multi-Agent Systems through Tensor Decompositioncited by 7
- 2025Mind the Generation Process: Fine-Grained Confidence Estimation During LLM Generationcited by 6
- 2026Are LLM Uncertainty and Correctness Encoded by the Same Features? A Functional Dissociation via Sparse Autoencoderscited by 6
Built on by
Citing papers per year: 2026: 2.
2 papers.
Similar, not cited
Close in wording, no citation either way
- 2026Unsupervised Confidence Calibration for Reasoning LLMs from a Single Generation
- 2025All Roads Lead to Rome: Graph-Based Confidence Estimation for Large Language Model Reasoning
- 2025CER: Confidence Enhanced Reasoning in LLMs
- 2026EvalQReason: A Framework for Step-Level Reasoning Evaluation in Large Language Models
- 2026Confidence over Time: Confidence Calibration with Temporal Logic for Large Language Model Reasoning
- 2025Confidence Improves Self-Consistency in LLMs
- 2026Probability is Not Enough: Exploring and Counting Divergent Tokens for Reasoning Uncertainty Quantification in LLMs
- 2025Sample Smart, Not Hard: Correctness-First Decoding for Better Reasoning in LLMs