arXiv:2410.03523cs.LGcs.AI2024-10ICLR被引 22

用概率方法评估大模型,更准识别模型真实能力。

A Probabilistic Perspective on Unlearning and Alignment for Large Language Models

  • 提出首个概率化评估框架,基于输出分布而非单一预测
  • 实验证明确定性评估会误判模型的遗忘与对齐效果
  • 适合关注模型可靠性、安全性的研究者与开发者

大语言模型(LLM)的全面评估仍是开放问题。现有评估依赖贪婪解码生成的确定性点估计,但我们发现这类方法无法捕捉模型输出的整体分布,导致能力评估失真,尤其在遗忘与对齐等关键场景中尤为严重。为此,我们提出首个形式化的概率评估框架,引入具有高概率保证的新度量指标。这些指标不依赖具体应用,使从业者能在部署前更可靠地估计模型能力。实验表明,确定性评估会错误显示成功遗忘与对齐,而我们的概率评估更能准确反映模型真实表现。通过一个遗忘案例研究,我们提出两种解决方案:(1) 基于熵优化的新型损失函数;(2) 自适应温度缩放。结果表明,该方法在近期基准上显著提升了概率设置下的遗忘性能。整体而言,从点估计转向输出分布的概率评估,是实现全面评估的重要一步。代码见 https://www.cs.cit.tum.de/daml/probabilistic-unlearning/。

原文摘要 · Abstract (English)

Comprehensive evaluation of Large Language Models (LLMs) is an open research problem. Existing evaluations rely on deterministic point estimates generated via greedy decoding. However, we find that deterministic evaluations fail to capture the whole output distribution of a model, yielding inaccurate estimations of model capabilities. This is particularly problematic in critical contexts such as unlearning and alignment, where precise model evaluations are crucial. To remedy this, we introduce the first formal probabilistic evaluation framework for LLMs. Namely, we propose novel metrics with high probability guarantees concerning the output distribution of a model. Our metrics are application-independent and allow practitioners to make more reliable estimates about model capabilities before deployment. Our experimental analysis reveals that deterministic evaluations falsely indicate successful unlearning and alignment, whereas our probabilistic evaluations better capture model capabilities. We show how to overcome challenges associated with probabilistic outputs in a case study on unlearning by introducing (1) a novel loss based on entropy optimization, and (2) adaptive temperature scaling. We demonstrate that our approach significantly enhances unlearning in probabilistic settings on recent benchmarks. Overall, our proposed shift from point estimates to probabilistic evaluations of output distributions represents an important step toward comprehensive evaluations of LLMs. Code available at https://www.cs.cit.tum.de/daml/probabilistic-unlearning/.

大模型评估概率建模模型遗忘对齐

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。