arXivDaily arXiv每日学术速递 周一至周五更新
​通知:联邦节假日调休:哥伦布日 / 原住民日(Columbus Day / Indigenous Peoples' Day),今日(13日)无更新。
arXiv 2605.07776cs.LGcs.AIcs.CL

追踪语言模型推理中的不确定性

Tracing Uncertainty in Language Model "Reasoning"

Data Science Section, IT University of Copenhagen(丹麦哥本哈根技术大学数据科学部门);MaiNLP, Center for Information and Language Processing, LMU Munich(慕尼黑大学信息与语言处理中心MaiNLP);Department of Applied Mathematics and Computer Science, Technical University of Denmark(丹麦技术大学应用数学与计算机科学系);Pioneer Centre for Artificial Intelligence(先锋人工智能中心);Munich Center for Machine Learning (MCML)(慕尼黑机器学习中心(MCML))

Nils Grünefeld, Bertram Højer, Philipp Mondorf, Barbara Plank, Anna Rogers, Christian Hardmeier, Stefan Heinrich, Jes Frellsen

AI总结:

研究通过不确定性量化分析语言模型推理轨迹,发现其不确定性特征可预测最终答案正确性,且早期即可检测错误。

摘要:

语言模型(LM)的'推理',通常描述为链式思维或测试时扩展,常提升基准性能,但其动态机制尚不明确。本文通过将推理轨迹视为演化的模型状态,利用不确定性轨迹特征(如斜率和线性度)预测最终答案正确性,五种模型在GSM8K和ProntoQA上的AUROC达0.807,且仅用前几百年token即可检测错误,正确轨迹的不确定性下降更陡且非线性,表明基于不确定性决策的方法为研究LM推理过程提供了原理性视角。

英文摘要:

Language model (LM) "reasoning", commonly described as Chain-of-Thought or test-time scaling, often improves benchmark performance, but the dynamics underlying this process remain poorly understood. We study these dynamics through the lens of uncertainty quantification by treating the "reasoning" traces, the intermediate token sequences generated by LMs, as evolving model states. We summarize each trace by an uncertainty trace profile: a small set of features describing the shape of the uncertainty signal over its trace, such as its slope and linearity. We find that across five LMs evaluated on GSM8K and ProntoQA, these profiles predict whether a trace yields a correct final answer with AUROC up to 0.807, improving markedly on recent related work. We reach AUROC 0.801 using only the first few hundred tokens of full traces, suggesting that errors can be detected early in the generation. A detailed comparison of correct and incorrect traces further reveals qualitatively distinct uncertainty profiles, with correct traces showing a steeper and less linear decline in uncertainty. Together, the results suggest that our method, grounded in decision-making under uncertainty, provides a principled lens for studying the generative process underlying LM "reasoning".

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑