arXiv:2410.02902cs.CLcs.AI2024-10ICLR被引 22

用大模型评判器优化指令遵循,提升生成质量且降低推理开销。

Better Instruction-Following Through Minimum Bayes Risk

  • 用参考答案评估器进行最小贝叶斯风险解码,选优输出
  • 相比贪心解码等方法,性能显著提升,70B模型也有效
  • 自训练后贪心解码即可媲美原模型的复杂解码,适合部署

通用大模型评判器可实现人类级评估,不仅可用于评测指令遵循型大模型,还能用于监督和改进其表现。本文探索使用最小贝叶斯风险(MBR)解码,通过基于参考答案的评判器从候选输出中选择高质量结果。实验表明,在AlpacaEval和MT-Bench上,该方法显著优于贪心解码、无参考的Best-of-N解码及基于词法或嵌入的度量方式,且在高达700亿参数的模型上均有效,说明小模型评判器可监督大模型。为进一步降低测试时开销,我们采用基于MBR解码输出的迭代自训练,发现使用直接偏好优化(DPO)的自训练带来显著性能提升,使仅用贪心解码的模型通常达到甚至超过原始模型在MBR下的表现。

原文摘要 · Abstract (English)

General-purpose LLM judges capable of human-level evaluation provide not only a scalable and accurate way of evaluating instruction-following LLMs but also new avenues for supervising and improving their performance. One promising way of leveraging LLM judges for supervision is through Minimum Bayes Risk (MBR) decoding, which uses a reference-based evaluator to select a high-quality output from amongst a set of candidate outputs. In the first part of this work, we explore using MBR decoding as a method for improving the test-time performance of instruction-following LLMs. We find that MBR decoding with reference-based LLM judges substantially improves over greedy decoding, best-of-N decoding with reference-free judges and MBR decoding with lexical and embedding-based metrics on AlpacaEval and MT-Bench. These gains are consistent across LLMs with up to 70B parameters, demonstrating that smaller LLM judges can be used to supervise much larger LLMs. Then, seeking to retain the improvements from MBR decoding while mitigating additional test-time costs, we explore iterative self-training on MBR-decoded outputs. We find that self-training using Direct Preference Optimisation leads to significant performance gains, such that the self-trained models with greedy decoding generally match and sometimes exceed the performance of their base models with MBR decoding.

大模型评估MBR解码自训练指令遵循

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