Distributional Energy-Based Models for Uncertainty-Aware Structured LLM Reasoning
On the map: LLM reasoning → Test-time compute and efficient thinking
Abstract (arXiv)
When Large Language Models produce structured outputs such as travel plans, code solutions, or multi-step proofs, individual reasoning steps may appear correct while the output as a whole violates budgets, fails test cases, or contradicts earlier deductions. We propose a decomposed energy function that combines a learned quality scorer with deterministic analytical constraint penalties for verifying structured LLM outputs. The quality scorer is a heterogeneous ensemble of low-rank adapters on a single frozen encoder (3% trainable parameters); the ensemble mean ranks candidates while the standard deviation quantifies epistemic uncertainty, driving a two-pass inference loop that triggers targeted regeneration or abstention. Across five benchmarks (GSM8K, MuSR, TravelPlanner, TACO, Knights & Knaves), our 149M-parameter verifier orchestrating a pool of 7-26B open generators outperforms single-shot Qwen-72B on every benchmark, matches Claude Sonnet 4.6 on MuSR (67.7% vs. 68.0%), and reduces constraint violations by 53% relative to Opus 4.6 on TravelPlanner (oracle 0.028, random 0.231). The two routes are complementary: structural verification wins when constraints are checkable (the verifier captures signal frontier models cannot self-detect), while pretraining-scale priors win where they are not (narrative inference, code semantics). A cross-dataset confounding analysis confirms genuine quality discrimination on four reasoning tasks and identifies a model-identity shortcut on code, mitigated via last-layer retraining. Scorers trained on difficult data transfer zero-shot: a MuSR-trained scorer achieves 93.9% on GSM8K without seeing a math problem.
Key facts. Cited by 2 papers in the map (data through 7 Oct 2026). Placed in Test-time compute and efficient thinking, one of 17 directions in LLM reasoning. Builds on 38 of its 65 references that are in the map. First posted on arXiv 15 May 2026.
Builds on
38 of 65 references.
Closest ancestors · 3
Same direction, newest first
- 2025Learning to Rank Chain-of-Thought: Using a Small Model21 May
- 2025Evolving Deeper LLM Thinking17 Jan
- 2023Self-Refine: Iterative Refinement with Self-Feedback30 Mar
Other references · 35
- 2018BERT: Pre-training of Deep Bidirectional Transformers for Language Understandingcited by 40,204
- 2023GPT-4 Technical Reportcited by 18,132
- 2022Training language models to follow instructions with human feedbackcited by 15,712
- 2024The Llama 3 Herd of Modelscited by 14,192
- 2021LoRA: Low-Rank Adaptation of Large Language Modelscited by 14,084
- 2021Training Verifiers to Solve Math Word Problemscited by 9,743
- 2021Evaluating Large Language Models Trained on Codecited by 8,053
- 2023Direct Preference Optimization: Your Language Model is Secretly a Reward Modelcited by 7,916
- 2025DeepSeek-R1: Incentivizing Reasoning Capability in LLMs via Reinforcement Learningcited by 6,146
- 2016Layer Normalizationcited by 5,579
- 2021Measuring Mathematical Problem Solving With the MATH Datasetcited by 5,203
- 2016SGDR: Stochastic Gradient Descent with Warm Restartscited by 4,993
23 more
- 2024Qwen2.5 Technical Reportcited by 4,713
- 2022Large Language Models are Zero-Shot Reasonerscited by 4,323
- 2019Parameter-Efficient Transfer Learning for NLPcited by 4,195
- 2016Simple and Scalable Predictive Uncertainty Estimation using Deep Ensemblescited by 3,888
- 2023Let's Verify Step by Stepcited by 3,598
- 2019The Curious Case of Neural Text Degenerationcited by 2,881
- 2017Decoupled Weight Decay Regularizationcited by 1,722
- 2021DeBERTaV3: Improving DeBERTa using ELECTRA-Style Pre-Training with Gradient-Disentangled Embedding Sharingcited by 1,219
- 2024Large Language Monkeys: Scaling Inference Compute with Repeated Samplingcited by 946
- 2022Solving math word problems with process- and outcome-based feedbackcited by 871
- 2019Deep Ensembles: A Loss Landscape Perspectivecited by 438
- 2022Last Layer Re-Training is Sufficient for Robustness to Spurious Correlationscited by 379
- 2024Skywork-Reward: Bag of Tricks for Reward Modeling in LLMscited by 322
- 2023Proving Test Set Contamination in Black Box Language Modelscited by 227
- 2023Rethinking Benchmark and Contamination for Language Models with Rephrased Samplescited by 192
- 2023TACO: Topics in Algorithmic COde generation datasetcited by 150
- 2025Process Reward Models That Thinkcited by 118
- 2020Residual Energy-Based Models for Text Generationcited by 95
- 2022Uncertainty Estimation for Language Reward Modelscited by 35
- 2025Autoregressive Language Models are Secretly Energy-Based Models: Insights into the Lookahead Capabilities of Next-Token Predictioncited by 7
- 2024Hitchhiker's guide on the relation of Energy-Based Models with other generative models, sampling and statistical physics: a comprehensive reviewcited by 5
- 2025Think Consistently, Reason Efficiently: Energy-Based Calibration for Implicit Chain-of-Thoughtcited by 3
- 2025A Theoretical Lens for RL-Tuned Language Models via Energy-Based Modelscited by 1
Built on by
Citing papers per year: 2026: 2.
2 papers.
Similar, not cited
Close in wording, no citation either way
- 2025Less is More: Improving LLM Reasoning with Minimal Test-Time Intervention
- 2025Sample Smart, Not Hard: Correctness-First Decoding for Better Reasoning in LLMs
- 2026Think Twice Before You Write -- an Entropy-based Decoding Strategy to Enhance LLM Reasoning
- 2026Silent Failures in Quantized LLM Reasoning: A Taxonomy-Based Analysis of Hollow Convergence and Failure Mode Shifts
- 2025Probabilistic Soundness Guarantees in LLM Reasoning Chains
- 2026Beyond Confidence: Stability-Aware Test-Time Adaptation for LLM Reasoning
- 2025Dist2ill: Distributional Distillation for One-Pass Uncertainty Estimation in Large Language Models
- 2026Evaluating LLM Uncertainty in Long-Form Generation Using Deterministic Ground Truth