arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2402.06544cs.CLcs.AIcs.LG
收藏

校准大型语言模型的长文本生成

Calibrating Long-form Generations from Large Language Models

Duke University(杜克大学);Yale University(耶鲁大学)

Yukun Huang, Yixin Liu, Raghuveer Thirukovalluru, Arman Cohan, Bhuwan Dhingra

AI总结:

针对长文本生成只能部分正确、二元校准失效的问题,提出将正确性与置信度视为分数分布的统一框架、三项指标及两种置信度诱导方法,并验证模型级联等改进策略。

摘要:

为提升大型语言模型(LLM)的可靠性,校准至关重要——模型给出的置信度分数应与其回答正确的实际可能性相符。然而,当前的置信度诱导方法和校准指标通常依赖于对回答正确性的二元真/假评估。这种方法不适用于答案可能部分正确的长文本生成。为弥补这一空白,我们提出一个统一校准框架,在该框架中,LLM回答的正确性及其相关置信度都被视为跨越一系列分数的分布。在此框架内,我们开发了三个指标以精确评估LLM校准,并进一步提出两种基于自洽性和自评估的置信度诱导方法。我们的实验涵盖长文本问答和摘要任务,结果表明:更大的模型并不一定保证更好的校准;校准性能被发现依赖于具体指标;自洽性方法在事实类数据集上表现优异。我们还发现,可以通过微调、整合相关源文档、调节温度以及将自洽性与自评估相结合等技术来增强校准。最后,我们展示了该系统的一个实际应用:在有限API预算下,选择并级联开源模型与ChatGPT,以优化正确性。本研究不仅挑战了关于LLM校准的既有观念,也为提升长文本生成的可信性提供了实用方法。

英文摘要:

To enhance Large Language Models' (LLMs) reliability, calibration is essential -- the model's assessed confidence scores should align with the actual likelihood of its responses being correct. However, current confidence elicitation methods and calibration metrics typically rely on a binary true/false assessment of response correctness. This approach does not apply to long-form generation, where an answer can be partially correct. Addressing this gap, we introduce a unified calibration framework, in which both the correctness of the LLMs' responses and their associated confidence levels are treated as distributions across a range of scores. Within this framework, we develop three metrics to precisely evaluate LLM calibration and further propose two confidence elicitation methods based on self-consistency and self-evaluation. Our experiments, which include long-form QA and summarization tasks, demonstrate that larger models don't necessarily guarantee better calibration, that calibration performance is found to be metric-dependent, and that self-consistency methods excel in factoid datasets. We also find that calibration can be enhanced through techniques such as fine-tuning, integrating relevant source documents, scaling the temperature, and combining self-consistency with self-evaluation. Lastly, we showcase a practical application of our system: selecting and cascading open-source models and ChatGPT to optimize correctness given a limited API budget. This research not only challenges existing notions of LLM calibration but also offers practical methodologies for improving trustworthiness in long-form generation.

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

↑