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

提示风险控制:大语言模型负责任部署的严格框架

Prompt Risk Control: A Rigorous Framework for Responsible Deployment of Large Language Models

Columbia University(哥伦比亚大学);Princeton University(普林斯顿大学)

Thomas P. Zollo, Todd Morrill, Zhun Deng, Jake C. Snell, Toniann Pitassi, Richard Zemel

AI总结:

本文提出提示风险控制框架,通过为最坏情况响应和用户间质量差异等风险度量提供严格上界来选择提示,并支持部署分布偏移,实验表明其能降低最坏结果风险,促进大语言模型负责任部署。

摘要:

大语言模型能力的近期爆发引发了人们对如何最好地提示模型执行给定任务的广泛关注。尽管基于验证集上的平均性能来简单地选择一个提示可能很诱人,但这可能导致部署中出现意外生成的糟糕响应,尤其是对处境最差的用户而言。为了缓解这一前景,我们提出了提示风险控制(Prompt Risk Control),这是一个轻量级框架,用于基于信息性风险度量族的严格上界来选择提示。我们提供了为多种度量生成上界的方法,包括衡量最坏情况响应和用户群体中生成质量差异的量。此外,我们扩展了底层的统计定界技术,以适应部署中可能出现的分布偏移。在开放式聊天、医疗问题摘要和代码生成等应用上的实验凸显了这一框架如何通过降低最坏结果的风险来促进负责任的部署。

英文摘要:

The recent explosion in the capabilities of large language models has led to a wave of interest in how best to prompt a model to perform a given task. While it may be tempting to simply choose a prompt based on average performance on a validation set, this can lead to a deployment where unexpectedly poor responses are generated, especially for the worst-off users. To mitigate this prospect, we propose Prompt Risk Control, a lightweight framework for selecting a prompt based on rigorous upper bounds on families of informative risk measures. We offer methods for producing bounds on a diverse set of metrics, including quantities that measure worst-case responses and disparities in generation quality across the population of users. In addition, we extend the underlying statistical bounding techniques to accommodate the possibility of distribution shifts in deployment. Experiments on applications such as open-ended chat, medical question summarization, and code generation highlight how such a framework can foster responsible deployment by reducing the risk of the worst outcomes.

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

↑