arXiv:2502.21265cs.CL2025-02被引 4

让词汇不同的模型也能在生成时联合预测,提升翻译效果。

Token-level Ensembling of Models with Different Vocabularies

  • 通过表面形式对齐实现不同词表模型的推理期集成
  • 在机器翻译任务中优于单一模型表现
  • 无需训练参数或修改模型,适合异构模型组合

模型集成能提升文本生成的鲁棒性和性能,通常通过加权合并多个模型的输出分布实现。然而,传统方法要求模型共享相同子词词表,限制了其应用范围,尤其在开源模型词表各异或实验中调整词表大小时。本文提出一种仅在推理阶段生效的算法,使词表不同的模型可进行逐标记集成,无需额外学习参数或修改原模型。该方法确保集成后生成的标记在表面形式上保持一致。我们在编码器-解码器模型与仅解码器大模型的组合上进行了实验,评估其在机器翻译任务上的表现。结果表明,该方法不仅扩展了可集成模型对的范围,还普遍提升了翻译性能,优于任一单独模型。

原文摘要 · Abstract (English)

Model ensembling is a technique to combine the predicted distributions of two or more models, often leading to improved robustness and performance. For ensembling in text generation, the next token's probability distribution is derived from a weighted sum of the distributions of each individual model. This requires the underlying models to share the same subword vocabulary, limiting the applicability of ensembling, since many open-sourced models have distinct vocabularies. In research settings, experimentation or upgrades to vocabularies may introduce multiple vocabulary sizes. This paper proposes an inference-time only algorithm that allows for ensembling models with different vocabularies, without the need to learn additional parameters or alter the underlying models. Instead, the algorithm ensures that tokens generated by the ensembled models \textit{agree} in their surface form. We apply this technique to combinations of traditional encoder-decoder models and decoder-only LLMs and evaluate on machine translation. In addition to expanding to model pairs that were previously incapable of token-level ensembling, our algorithm frequently improves translation performance over either model individually.

模型集成词表对齐机器翻译

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