arXiv:2603.05432cs.CLcs.AI2026-03被引 2

用蒙特卡洛方法统一融合多个语言模型,提升生成质量。

Ensembling Language Models with Sequential Monte Carlo

  • 提出基于SMC的字节级采样框架,兼容不同词汇表的模型
  • 多种聚合策略比传统概率平均更优,显著提升生成效果
  • 适合需要高精度文本生成的研究者与工程实践

实践中可获取大量语言模型和提示策略,但性能对选择高度敏感。经典集成方法虽能提升表现,但在解码阶段应用困难:直接合并下一个词的概率会得到局部归一化、有偏差的近似分布。本文提出统一框架,将K个语言模型组合成f-ensemble分布(适用于广泛的函数f: R≥0^K → R≥0)。为从这些分布采样,提出字节级顺序蒙特卡洛(SMC)算法,在共享字符空间中运行,支持词汇不匹配模型的一致采样。在多种结构化文本生成任务上评估了f-ensemble家族,验证了替代聚合策略优于传统概率平均,并表明更优后验近似可带来更好集成性能。

原文摘要 · Abstract (English)

Practitioners have access to an abundance of language models and prompting strategies for solving many language modeling tasks; yet prior work shows that modeling performance is highly sensitive to both choices. Classical machine learning ensembling techniques offer a principled approach: aggregate predictions from multiple sources to achieve better performance than any single one. However, applying ensembling to language models during decoding is challenging: naively aggregating next-token probabilities yields samples from a locally normalized, biased approximation of the generally intractable ensemble distribution over strings. In this work, we introduce a unified framework for composing $K$ language models into $f$-ensemble distributions for a wide range of functions $f\colon\mathbb{R}_{\geq 0}^{K}\to\mathbb{R}_{\geq 0}$. To sample from these distributions, we propose a byte-level sequential Monte Carlo (SMC) algorithm that operates in a shared character space, enabling ensembles of models with mismatching vocabularies and consistent sampling in the limit. We evaluate a family of $f$-ensembles across prompt and model combinations for various structured text generation tasks, highlighting the benefits of alternative aggregation strategies over traditional probability averaging, and showing that better posterior approximations can yield better ensemble performance.

语言模型集成学习生成质量SMC

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