mlsanityData to 7 Oct 2026

Distributional Energy-Based Models for Uncertainty-Aware Structured LLM Reasoning

arXiv 2605.18871 ↗First posted 15 May 2026cs.LG cs.AIcited by 2

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

  1. 2025Learning to Rank Chain-of-Thought: Using a Small Model21 May
  2. 2025Evolving Deeper LLM Thinking17 Jan
  3. 2023Self-Refine: Iterative Refinement with Self-Feedback30 Mar

Other references · 35

  1. 2018BERT: Pre-training of Deep Bidirectional Transformers for Language Understandingcited by 40,204
  2. 2023GPT-4 Technical Reportcited by 18,132
  3. 2022Training language models to follow instructions with human feedbackcited by 15,712
  4. 2024The Llama 3 Herd of Modelscited by 14,192
  5. 2021LoRA: Low-Rank Adaptation of Large Language Modelscited by 14,084
  6. 2021Training Verifiers to Solve Math Word Problemscited by 9,743
  7. 2021Evaluating Large Language Models Trained on Codecited by 8,053
  8. 2023Direct Preference Optimization: Your Language Model is Secretly a Reward Modelcited by 7,916
  9. 2025DeepSeek-R1: Incentivizing Reasoning Capability in LLMs via Reinforcement Learningcited by 6,146
  10. 2016Layer Normalizationcited by 5,579
  11. 2021Measuring Mathematical Problem Solving With the MATH Datasetcited by 5,203
  12. 2016SGDR: Stochastic Gradient Descent with Warm Restartscited by 4,993
23 more
  1. 2024Qwen2.5 Technical Reportcited by 4,713
  2. 2022Large Language Models are Zero-Shot Reasonerscited by 4,323
  3. 2019Parameter-Efficient Transfer Learning for NLPcited by 4,195
  4. 2016Simple and Scalable Predictive Uncertainty Estimation using Deep Ensemblescited by 3,888
  5. 2023Let's Verify Step by Stepcited by 3,598
  6. 2019The Curious Case of Neural Text Degenerationcited by 2,881
  7. 2017Decoupled Weight Decay Regularizationcited by 1,722
  8. 2021DeBERTaV3: Improving DeBERTa using ELECTRA-Style Pre-Training with Gradient-Disentangled Embedding Sharingcited by 1,219
  9. 2024Large Language Monkeys: Scaling Inference Compute with Repeated Samplingcited by 946
  10. 2022Solving math word problems with process- and outcome-based feedbackcited by 871
  11. 2019Deep Ensembles: A Loss Landscape Perspectivecited by 438
  12. 2022Last Layer Re-Training is Sufficient for Robustness to Spurious Correlationscited by 379
  13. 2024Skywork-Reward: Bag of Tricks for Reward Modeling in LLMscited by 322
  14. 2023Proving Test Set Contamination in Black Box Language Modelscited by 227
  15. 2023Rethinking Benchmark and Contamination for Language Models with Rephrased Samplescited by 192
  16. 2023TACO: Topics in Algorithmic COde generation datasetcited by 150
  17. 2025Process Reward Models That Thinkcited by 118
  18. 2020Residual Energy-Based Models for Text Generationcited by 95
  19. 2022Uncertainty Estimation for Language Reward Modelscited by 35
  20. 2025Autoregressive Language Models are Secretly Energy-Based Models: Insights into the Lookahead Capabilities of Next-Token Predictioncited by 7
  21. 2024Hitchhiker's guide on the relation of Energy-Based Models with other generative models, sampling and statistical physics: a comprehensive reviewcited by 5
  22. 2025Think Consistently, Reason Efficiently: Energy-Based Calibration for Implicit Chain-of-Thoughtcited by 3
  23. 2025A Theoretical Lens for RL-Tuned Language Models via Energy-Based Modelscited by 1

Built on by

Cited by year (2026 to 7 Oct)

Citing papers per year: 2026: 2.

2 papers.

Similar, not cited

Close in wording, no citation either way

  1. 2025Less is More: Improving LLM Reasoning with Minimal Test-Time Intervention
  2. 2025Sample Smart, Not Hard: Correctness-First Decoding for Better Reasoning in LLMs
  3. 2026Think Twice Before You Write -- an Entropy-based Decoding Strategy to Enhance LLM Reasoning
  4. 2026Silent Failures in Quantized LLM Reasoning: A Taxonomy-Based Analysis of Hollow Convergence and Failure Mode Shifts
  5. 2025Probabilistic Soundness Guarantees in LLM Reasoning Chains
  6. 2026Beyond Confidence: Stability-Aware Test-Time Adaptation for LLM Reasoning
  7. 2025Dist2ill: Distributional Distillation for One-Pass Uncertainty Estimation in Large Language Models
  8. 2026Evaluating LLM Uncertainty in Long-Form Generation Using Deterministic Ground Truth