Can Uncertainty Metrics Guide Search? Evaluating Search-Time Decision-Making in Large Language Models
Preprint
- 6 June 2025
- preprint
- Published by MDPI AG in Preprints.org
Abstract
Sampling-based search methods for Large Language Models (LLMs), such as Chain of Thought (CoT) and Tree of Thought (ToT), improve accuracy by exploring multiple reasoning paths. These approaches can be enhanced by search algorithms like Monte Carlo Tree Search (MCTS) and Bandit strategies, which rely on effective uncertainty estimation. While many uncertainty metrics have been proposed for LLMs, we argue that most of them are primarily descriptive, as they capture model confidence rather than prescriptive tools for guiding search toward correct answers. In this study, we evaluate whether existing uncertainty metrics can support search-time decision-making through two diagnostic tasks: Answer Selection and Answer Refinement. Our results show that while current metrics help flag risky or error-prone responses, they fall short in guiding search toward correctness. This highlights the need for optimization-aware uncertainty metrics explicitly designed to support correctness-driven decisions. Code is available at link.Keywords
All Related Versions
This publication has 13 references indexed in Scilit:
- Differentiable Good Arm Identification2025
- LinearAPT: An Adaptive Algorithm for the Fixed-Budget Thresholding Linear Bandit Problem2025
- Enhance Modality Robustness in Text-Centric Multimodal Alignment with Adversarial PromptingProceedings of the AAAI Conference on Artificial Intelligence, 2025
- RTLFixer: Automatically Fixing RTL Syntax Errors with Large Language Model2024
- Automatically Correcting Large Language Models: Surveying the Landscape of Diverse Automated Correction StrategiesTransactions of the Association for Computational Linguistics, 2024
- Automatic Engineering of Long Prompts2024
- Large Language Models are Better Reasoners with Self-Verification2023
- STaR: Bootstrapping Reasoning with Reasoning2022
- TruthfulQA: Measuring How Models Mimic Human Falsehoods2022
- CommonsenseQA: A Question Answering Challenge Targeting Commonsense Knowledge2019