让AI生成更符合语义结构的回复,提升对话和指令任务质量
Structure-Conditional Minimum Bayes Risk Decoding
- 设计新型效用函数,使MBR decoding更关注生成结果的潜在结构差异
- 在真实指令任务中,生成质量最高提升13.7个百分点
- 适合需要精准控制输出结构的对话与指令生成场景
最小贝叶斯风险(MBR)解码近年来重新受到关注,作为传统生成策略的替代方案。尽管其在机器翻译中表现良好,但在开放性任务如对话或指令遵循中可能面临挑战。我们假设:在这些场景下,使用标准相似性效用函数的MBR可能选出的是模型分布的普遍代表,而非特定潜在结构下的最优响应。为此,我们提出三种轻量级效用函数改进,增强MBR对生成空间中结构变异性敏感度。我们构建了一个包含对话意图、情感和响应结构(如句子、段落或列表)三类典型潜在结构的数据集,并提出两个评估结构最优性的指标。分析表明,常见相似性效用函数在此指标下表现不足。而我们的方法显著提升了结构最优性。最终在AlpacaEval和MT-Bench基准上验证,结构敏感性提升使胜率最高提高13.7个百分点。
原文摘要 · Abstract (English)
Minimum Bayes Risk (MBR) decoding has seen renewed interest as an alternative to traditional generation strategies. While MBR has proven effective in machine translation, where the variability of a language model's outcome space is naturally constrained, it may face challenges in more open-ended tasks such as dialogue or instruction-following. We hypothesise that in such settings, applying MBR with standard similarity-based utility functions may result in selecting responses that are broadly representative of the model's distribution, yet sub-optimal with respect to any particular grouping of generations that share an underlying latent structure. In this work, we introduce three lightweight adaptations to the utility function, designed to make MBR more sensitive to structural variability in the outcome space. To test our hypothesis, we curate a dataset capturing three representative types of latent structure: dialogue act, emotion, and response structure (e.g., a sentence, a paragraph, or a list). We further propose two metrics to evaluate the structural optimality of MBR. Our analysis demonstrates that common similarity-based utility functions fall short by these metrics. In contrast, our proposed adaptations considerably improve structural optimality. Finally, we evaluate our approaches on real-world instruction-following benchmarks, AlpacaEval and MT-Bench, and show that increased structural sensitivity improves generation quality by up to 13.7 percentage points in win rate.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。