提出高效估计语言模型有害输出概率的方法,解决罕见风险难以捕捉问题。
Estimating Tail Risks in Language Model Output Distributions

- 用不安全版本模型实现重要性采样,提升有害输出采样效率
- 仅需500样本即可估计10^-4量级的有害概率,效率提升10-20倍
- 可检测输入扰动敏感性,预判部署风险,适合安全评估者使用
语言模型正被大规模部署,其安全性愈发关键。尽管对齐技术降低了有害输出概率,但在每日数十亿次调用下,罕见最坏情况仍会发生。现有评估忽略模型的概率特性与尾部行为。本文提出一种高效估计任意输入下有害输出概率的方法:通过构建目标模型的不安全版本,使有害输出更易生成,从而实现重要性采样。在滥用和错位基准测试中,该方法仅需10-20倍更少样本,即可达到与暴力蒙特卡洛采样相当的估计精度。例如,仅用500样本即可准确估计10^-4量级的有害概率。此外,该估计能揭示模型对输入扰动的敏感性,并预测实际部署风险。结果表明,精准的稀有事件估计在安全评估中既必要又可行。代码已开源。
原文摘要 · Abstract (English)
Language models are increasingly capable and are being rapidly deployed on a population-level scale. As a result, the safety of these models is increasingly high-stakes. Fortunately, advances in alignment have significantly reduced the likelihood of harmful model outputs. However, when models are queried billions of times in a day, even rare worst-case behaviors will occur. Current safety evaluations focus on capturing the distribution of inputs that yield harmful outputs. These evaluations disregard the probabilistic nature of models and their tail output behavior. To measure this tail risk, we propose a method to efficiently estimate the probability of harmful outputs for any input query. Instead of naive brute-force sampling from the target model, where harmful outputs could be rare, we operationalize importance sampling by creating unsafe versions of the target model. These unsafe versions enable sample-efficient estimation by making harmful outputs more probable. On benchmarks measuring misuse and misalignment, these estimates match brute-force Monte Carlo estimates using 10-20x fewer samples. For example, we can estimate probability of harmful outputs on the order of 10^-4 with just 500 samples. Additionally, we find that these harmfulness estimates can reveal the sensitivity of models to perturbations in model input and predict deployment risks. Our work demonstrates that accurate rare-event estimation is both critical and feasible for safety evaluations. Code is available at https://github.com/rangell/LMTailRisk
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。