arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2603.03752cs.CLcs.AI
收藏

置信度校准的小-大语言模型协作用于成本有效的推理

Confidence-Calibrated Small-Large Language Model Collaboration for Cost-Efficient Reasoning

Amazon Web Services(亚马逊网络服务);Tsinghua University(清华大学)

Chuang Zhang, Zizhen Zhu, Yihao Wei, Bing Tian, Junyi Liu, Henan Wang, Xavier Wang, Yaxiao Liu

AI总结:

COREA通过结合小型和大型语言模型,利用置信度校准提升推理效率,降低21.5%-16.8%成本,同时保持高准确率。

摘要:

大型语言模型(LLMs)相比小型语言模型(SLMs)在推理能力上表现出色,但成本显著更高。我们提出了COllaborative REAsoner(COREA)系统,通过将SLM与LLM串联,以在复杂推理任务中实现准确性和成本之间的平衡。COREA首先尝试使用SLM回答问题,其输出包括答案和口头化的置信度评分。置信度低于预定义阈值的问题会转交LLM进行更准确的解决。我们引入了一种基于强化学习的训练算法,通过额外的置信度校准奖励对SLM的置信度进行对齐。广泛的实验表明,我们的方法在多种数据集和模型架构上共同提升了SLM的推理能力和置信度校准。与单独使用LLM相比,COREA在域外数学和非数学数据集上分别将成本降低了21.5%和16.8%,仅在pass@1上绝对下降不超过2%。

英文摘要:

Large language models (LLMs) demonstrate superior reasoning capabilities compared to small language models (SLMs), but incur substantially higher costs. We propose COllaborative REAsoner (COREA), a system that cascades an SLM with an LLM to achieve a balance between accuracy and cost in complex reasoning tasks. COREA first attempts to answer questions using the SLM, which outputs both an answer and a verbalized confidence score. Questions with confidence below a predefined threshold are deferred to the LLM for more accurate resolution. We introduce a reinforcement learning-based training algorithm that aligns the SLM's confidence through an additional confidence calibration reward. Extensive experiments demonstrate that our method jointly improves the SLM's reasoning ability and confidence calibration across diverse datasets and model backbones. Compared to using the LLM alone, COREA reduces cost by 21.5% and 16.8% on out-of-domain math and non-math datasets, respectively, with only an absolute pass@1 drop within 2%.

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

↑