arXiv:2410.15021cs.CL2024-10ACL被引 10

用偏差-多样性分解解释大模型推理中的采样效应

Diversity Explains Inference Scaling Laws: Through a Case Study of Minimum Bayes Risk Decoding

  • 从偏差与多样性的角度重新分析MBR解码机制
  • 发现增加样本多样性能有效提升生成质量
  • 适合关注大模型推理优化的研究者参考

推理方法在激发大语言模型(LLMs)性能方面起着关键作用。当前的LLM推理方法依赖于生成多个样本,其中一种典型方法是最小贝叶斯风险(MBR)解码。已有研究通过实证分析揭示了MBR解码带来的生成性能提升,并报告了多种观察结果,但其理论基础仍不明确。为此,本文从偏差-多样性分解的角度提出新的理论解释:MBR解码的质量估计误差可分解为两个主要因素——偏差(衡量效用函数与人类评估的一致性)和多样性(衡量效用函数质量估计的变异性)。理论分析表明,同时优化偏差与多样性存在困难,验证了通过增加多样性来提升MBR解码性能的有效性。此外,我们发现多样性可解释推理缩放定律中样本数量增加导致性能提升的一个方面。在多个NLP任务上的实验结果均与这一理论特征一致。代码已公开于https://github.com/naist-nlp/mbr-bias-diversity。

原文摘要 · Abstract (English)

Inference methods play an important role in eliciting the performance of large language models (LLMs). Currently, LLMs use inference methods utilizing generated multiple samples, which can be derived from Minimum Bayes Risk (MBR) Decoding. Previous studies have conducted empirical analyses to clarify the improvements in generation performance achieved by MBR decoding and have reported various observations. However, the theoretical underpinnings of these findings remain uncertain. To address this, we offer a new theoretical interpretation of MBR decoding from the perspective of bias-diversity decomposition. In this interpretation, the error in the quality estimation of hypotheses by MBR decoding is decomposed into two main factors: bias, which considers the closeness between the utility function and human evaluation, and diversity, which represents the variability in the quality estimation of the utility function. The theoretical analysis reveals the difficulty of simultaneously improving bias and diversity, confirming the validity of enhancing MBR decoding performance by increasing diversity. Furthermore, we reveal that diversity can explain one aspect of inference scaling laws that describe performance improvement by increasing sample size. Moreover, experiments across multiple NLP tasks yielded results consistent with these theoretical characteristics. Our code is available at https://github.com/naist-nlp/mbr-bias-diversity.

大模型推理生成质量多样性

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