中文

通过多提示改进最小贝叶斯风险解码

计算与语言 2024-10-07 v2

摘要

虽然经过指令微调的大语言模型(LLM)作为文本生成器效果良好,但对提示构建敏感,在实际应用中表现不稳定且次优。仅依赖单一的“最佳”提示无法捕捉到对生成问题不同方法的覆盖。基于这一观察,我们提出了多提示解码(multi-prompt decoding)方法,即在推理阶段从提示库中生成多个候选答案,再通过最小贝叶斯风险(Minimum Bayes Risk, MBR)解码进行ensemble。我们表明,多提示方法在全面的条件生成任务上均优于单一提示的 MBR 解码,这源于其估计的候选空间更具多样性和更高质量。进一步实验表明,多提示方法在不同任务、模型和指标下均能提升生成效果。

关键词

引用

@article{arxiv.2407.15343,
  title  = {Improving Minimum Bayes Risk Decoding with Multi-Prompt},
  author = {David Heineman and Yao Dou and Wei Xu},
  journal= {arXiv preprint arXiv:2407.15343},
  year   = {2024}
}

备注

Accepted to EMNLP 2024