跳到正文
arXiv cs.CL· Mathias Ollu, Nikos Komodakis·· 13 小时前AI 评分32

H-CDLM:面向语言建模的分层连续扩散框架

Denoising Hierarchical Representations: Joint Continuous Diffusion for Language Modeling

AI 导读

研究者提出分层连续扩散语言模型 H-CDLM,通过并行扩散 token 本身与预训练 token 嵌入聚类得到的粗粒度簇两种模态,并允许各模态使用独立采样器与调度。应用于 CoBit 得到 H-CoBit,在 LM1B 和 OWT 上生成困惑度分别为 49.4 和 50.4,较基线改善 24.2 和 20.7 点,GSM8K 准确率达 27.4%。该框架同样适用于 FLM 流匹配模型,代码将公开。

正文 · AI 翻译

查看 PDF HTML(实验性)

摘要:扩散语言模型(DLMs)有望实现与顺序无关的并行文本生成。最近,连续扩散和流匹配模型取得了显著进展,这得益于精心设计的 token 表示以及扩散/流空间。在本工作中,我们提出了分层连续扩散语言模型(H-CDLMs),这是一个简单的框架,以极小的计算和参数开销进一步改进连续 DLMs。借鉴离散 DLM 和连续图像扩散文献中关于联合扩散的研究,我们并行地对多种模态进行扩散。这些模态表示不同语义粒度上的 token:在我们的实例化中,即 token 本身以及通过对预训练 token 嵌入进行聚类得到的更粗粒度簇。我们提出了一个通用设置,允许为每种模态配置采样器和调度,以增强模态之间的相互作用。将其应用于 CoBit 得到 H-CoBit,在各项基准测试中带来了大幅度的实证提升。在数据集熵下,H-CoBit 提升了 MAUVE,并在 LM1B 上达到 49.4、在 OWT 上达到 50.4 的生成困惑度(GenPPL),分别比基线提升 24.2 和 20.7 个点,甚至超越了同等规模的离散 DLMs。在 GSM8K 上,它达到 27.4% 的准确率,优于先前的连续扩散和基于流的模型。我们进一步将 H-CDLM 应用于流匹配模型 FLM,得到 H-FLM 并取得了一致的提升,表明该框架可泛化到不同的连续生成范式。我们的代码将在 此 https URL 公开。
评论: 27 页,10 张图
主题: 计算与语言(cs.CL);机器学习(cs.LG)
引用为: arXiv:2610.08738 [cs.CL]
  (或 arXiv:2610.08738v2 [cs.CL] 查看此版本)
  https://doi.org/10.48550/arXiv.2610.08738

由 DataCite 颁发的 arXiv DOI

提交历史

来自:Mathias Ollu [查看邮箱]
[v1] 2026 年 10 月 6 日星期二 17:38:19 UTC(101 KB)
[v2] 2026 年 10 月 27 日星期二 10:43:42 UTC(101 KB)

来源:arXiv cs.CL · arxiv.org