arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2403.05973cs.CLcs.AIcs.LG
收藏

仅利用大语言模型的生成内容对其进行校准

Calibrating Large Language Models Using Their Generations Only

Parameter Lab(参数实验室);IT University of Copenhagen(哥本哈根信息技术大学);Pioneer Centre for Artificial Intelligence(人工智能先锋中心);NAVER AI Lab(NAVER人工智能实验室);University of Tübingen(蒂宾根大学);Tübingen AI Center(蒂宾根人工智能中心)

Dennis Ulmer, Martin Gubri, Hwaran Lee, Sangdoo Yun, Seong Joon Oh

AI总结:

针对仅能通过生成文本交互的大语言模型校准难题,提出APRICOT方法训练额外模型基于文本输入输出预测置信度,在闭卷问答的白盒黑盒场景校准误差表现具竞争力,可检测错误答案。

摘要:

随着大语言模型(LLM)越来越多地部署在面向用户的应用中,通过准确量化模型对其预测的置信度来建立信任并保障安全性变得愈发重要。然而,找到有效的LLM校准方法——尤其是当仅能通过模型生成的文本与之交互时——仍是一项挑战。我们提出APRICOT(置信度目标辅助预测):这是一种设置置信度目标,并训练一个额外模型仅基于LLM的文本输入和输出来预测其置信度的方法。该方法具备多项优势:概念上简洁明了,除目标模型的输出外无需访问其内部结构,不会干扰语言生成过程,且拥有大量潜在应用场景,例如将预测的置信度进行言语化表达,或是基于置信度调整给出的答案。我们的实验表明,在闭卷问答任务中,针对白盒和黑盒LLM,该方法在校准误差方面表现具备竞争力,可用于检测LLM的错误答案。

英文摘要:

As large language models (LLMs) are increasingly deployed in user-facing applications, building trust and maintaining safety by accurately quantifying a model's confidence in its prediction becomes even more important. However, finding effective ways to calibrate LLMs - especially when the only interface to the models is their generated text - remains a challenge. We propose APRICOT (auxiliary prediction of confidence targets): A method to set confidence targets and train an additional model that predicts an LLM's confidence based on its textual input and output alone. This approach has several advantages: It is conceptually simple, does not require access to the target model beyond its output, does not interfere with the language generation, and has a multitude of potential usages, for instance by verbalizing the predicted confidence or adjusting the given answer based on the confidence. We show how our approach performs competitively in terms of calibration error for white-box and black-box LLMs on closed-book question-answering to detect incorrect LLM answers.

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

↑