跳到正文
arXiv cs.CL· Yang Liu, Zhiqiang Yuan, Shu-Dong Huang, Jiancheng Lv, Liwei Yang, Qiwei Yan, Kailin Lv, Jinchao Zhang·· 12 小时前AI 评分31

ULME:让嵌入与语言建模相互增强的预测耦合框架

ULME: Predictive Coupling for Mutually Reinforcing Embedding and Language Modeling

AI 导读

ULME 是一个后训练框架,在因果 decoder-only LLM 上通过预测耦合统一嵌入与语言建模,让生成反向塑造嵌入所编码的内容。

正文 · AI 翻译

查看 PDF HTML(实验性)

摘要:文本嵌入和文本生成是现代大语言模型最有用的两项能力,但它们通常由分别训练的专用模型(一个嵌入器和一个生成器)提供,这些模型使用互不兼容的目标进行训练,并在检索增强生成和智能体系统中并排部署。近期诸如 GritLM 之类的统一嵌入-生成模型在两种能力之间共享参数,但它们要么在模式之间切换注意力机制,要么将嵌入和语言建模作为彼此从不监督的并行目标进行优化。我们认为这些设计只是部分统一:两种能力共存于同一组权重上,但并未相互塑造。我们提出 ULME,一个后训练框架,通过预测耦合在仅解码器的因果 LLM 上统一嵌入和语言建模。ULME 在查询和答案之间插入一串可学习槽位,从两个指定槽位读取语义嵌入和推理嵌入,并在共享的因果主干上训练三个目标:一个将语义槽位锚定到文档表示的对比损失、一个监督答案生成的语言建模损失,以及一个将推理槽位与答案的池化摘要对齐的停止梯度预测损失,从而使生成主动塑造嵌入所编码的内容。在两个模型系列(Qwen2.5、Qwen3)和涵盖 0.5B-8B 参数的八个规模上,单个 ULME 检查点将 MTEB 较最强的开源基线提升 3.3-4.1 分,同时收紧 BRIGHT 和 AdvBench-IR,并在标准准确率基准(MMLU、GSM8K、MMLU-Pro、HumanEval)上与仅生成微调持平,表明真正的统一产生的是相互增强而非仅仅共存的能力。
评论: AAAI 2027 已提交
主题: 计算与语言(cs.CL);人工智能(cs.AI)
引用为: arXiv:2610.30705 [cs.CL]
  (或 arXiv:2610.30705v1 [cs.CL] 用于此版本)
  https://doi.org/10.48550/arXiv.2610.30705

由 DataCite 颁发的 arXiv DOI(待注册)

提交历史

来自:Zhiqiang Yuan [查看电子邮件]
[v1] 2026年10月22日星期四 04:23:07 UTC(168 KB)

来源:arXiv cs.CL · arxiv.org