The LLM Already Knows: Estimating LLM-Perceived Question Difficulty via Hidden Representations
By: Yubo Zhu , Dongrui Liu , Zecheng Lin and more
Potential Business Impact:
Helps computers know how hard questions are.
Estimating the difficulty of input questions as perceived by large language models (LLMs) is essential for accurate performance evaluation and adaptive inference. Existing methods typically rely on repeated response sampling, auxiliary models, or fine-tuning the target model itself, which may incur substantial computational costs or compromise generality. In this paper, we propose a novel approach for difficulty estimation that leverages only the hidden representations produced by the target LLM. We model the token-level generation process as a Markov chain and define a value function to estimate the expected output quality given any hidden state. This allows for efficient and accurate difficulty estimation based solely on the initial hidden state, without generating any output tokens. Extensive experiments across both textual and multimodal tasks demonstrate that our method consistently outperforms existing baselines in difficulty estimation. Moreover, we apply our difficulty estimates to guide adaptive reasoning strategies, including Self-Consistency, Best-of-N, and Self-Refine, achieving higher inference efficiency with fewer generated tokens.
Similar Papers
Probing the Difficulty Perception Mechanism of Large Language Models
Computation and Language
Computers learn to tell if math problems are hard.
LLMs Encode How Difficult Problems Are
Computation and Language
Makes computers understand hard problems better.
Estimating problem difficulty without ground truth using Large Language Model comparisons
Machine Learning (CS)
Helps AI learn harder problems by guessing difficulty.