arXiv:2604.01170cs.LGcs.AI2026-04被引 3

让大模型在推理时自动校准,提升准确率并节省计算资源。

Online Reasoning Calibration: Test-Time Training Enables Generalizable Conformal LLM Reasoning

  • 引入在线校准机制,动态调整采样过程以适应不同推理阶段。
  • 在分布外任务中实现67%的计算节省,误差率仍很低。
  • 适合追求高效率与泛化能力的推理系统开发者使用。

尽管测试时扩展已使大语言模型能够解决极难任务,但当前最优结果伴随高昂的计算成本。这些低效源于后训练模型的校准不足以及主流采样技术缺乏校准。本文提出在线推理校准(ORCA),结合置信预测与测试时训练,通过元学习为每个输入动态更新校准模块。该方法在分布漂移下提供有效置信度估计,例如推理过程中不同阶段的思想模式变化或开发与部署间提示分布差异。ORCA不仅具备理论保证的置信风险控制,还在实证上展现出更高效率和泛化性。在δ=0.1的风险水平下,Qwen2.5-32B在分布内任务中使用监督标签可节省47.5%计算量,自洽标签下节省40.7%;在零样本分布外设置中,对MATH-500的计算节省从静态校准基线的24.8%提升至67.0%,且该趋势在多个模型族和下游基准上一致。代码已开源:https://github.com/wzekai99/ORCA。

原文摘要 · Abstract (English)

While test-time scaling has enabled large language models to solve highly difficult tasks, state-of-the-art results come at exorbitant compute costs. These inefficiencies can be attributed to the miscalibration of post-trained language models, and the lack of calibration in popular sampling techniques. Here, we present Online Reasoning Calibration (ORCA), a framework for calibrating the sampling process that draws upon conformal prediction and test-time training. Specifically, we introduce a meta-learning procedure that updates the calibration module for each input. This allows us to provide valid confidence estimates under distributional shift, e.g. in thought patterns that occur across different stages of reasoning, or in prompt distributions between model development and deployment. ORCA not only provides theoretical guarantees on conformal risks, but also empirically shows higher efficiency and generalization across different reasoning tasks. At risk level $δ=0.1$, ORCA improves Qwen2.5-32B efficiency on in-distribution tasks with savings up to 47.5% with supervised labels and 40.7% with self-consistency labels. Under zero-shot out-of-domain settings, it improves MATH-500 savings from 24.8% of the static calibration baseline to 67.0% while maintaining a low empirical error rate, and the same trend holds across model families and downstream benchmarks. Our code is publicly available at https://github.com/wzekai99/ORCA.

大模型推理置信校准测试时训练

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。