ReLF 用整流流做并行文本生成,少步数下速度超自回归模型

论文AI 评分 72/100arXiv cs.AI
AI 聚合

本条为 AI 依据下方公开信源自动整理生成的摘要,不构成转载,可能存在偏差,请以原文为准。整理者:贝果科技 AI 资讯助手。

一篇 arXiv 论文提出 Rectified Language Flow(ReLF),用整流流把噪声搬运到 token 嵌入上生成文本,以替代自回归解码。方法组合了 Hellinger-SVD 构造的固定 token 嵌入、用交叉熵训练的自条件引导,以及面向少步生成的正则化 reflow;并提出 singular logits,把解析项与有界神经残差相加。在 OpenWebText 上,三种测试规模下,64 次函数评估时生成困惑度低于自回归基线且多样性对齐;两轮 reflow 后,16 次函数评估仍优于自回归基线,batch size 1 下生成 1024 token 序列快 15 到 19 倍。

全文梳理

AI 摘要依据下方信源原文自动整理,非原文转载

方法的核心设计

ReLF 属于连续状态流与扩散模型路线,目标是用并行生成替代逐 token 的自回归解码。它把噪声搬运到 token 嵌入空间完成文本生成,并叠加三项设计:用 Hellinger-SVD 构造固定的 token 嵌入;自条件引导以交叉熵为训练目标;正则化 reflow 用于压缩采样步数。论文还提出 singular logits,由两部分相加构成,一部分是解析计算项,在噪声消退时推动结果落到合法的 token 嵌入上,另一部分是有界神经残差,用来学习上下文相关的 token 偏好。

评测结果与速度

实验在 OpenWebText 上进行,覆盖三种模型规模。在 64 次函数评估的预算下,ReLF 的生成困惑度低于自回归基线,同时保持多样性对齐,并且在该设定下优于此前的连续流方法。经过两轮 reflow 后,即使把函数评估次数压到 16 次,它仍然优于自回归基线;在 batch size 1 的条件下生成 1024 token 序列,速度是后者的 15 到 19 倍。这些数字说明少步采样并未以明显质量损失为代价。

适用场景与限制

材料给出的对比集中在困惑度与生成速度两项,评测语料为 OpenWebText,序列长度为 1024 token,速度数据取自 batch size 1。论文未在摘要中说明更大批量、更长序列或其他语料上的表现,也未给出与自回归模型在具体下游任务上的对照。对关注推理成本与并行解码的读者,函数评估次数与加速倍数的对应关系是判断可用性的关键信息。

为什么值得看

连续流语言模型在质量上追平自回归模型的证据,附具体加速倍数与评估预算,值得关注并行生成路线的人细看。

对齐论文

信源1 家

  1. [1]arXiv cs.AI一手信源Rectified Language Flow

关键事实

  • 论文提出 ReLF,用整流流将噪声传输到 token 嵌入来生成文本[1]

  • 方法包含 Hellinger-SVD 固定 token 嵌入、交叉熵训练的自条件引导与正则化 reflow[1]

  • 在 OpenWebText 上,64 次函数评估时三种模型规模的生成困惑度均低于自回归基线[1]

  • 两轮 reflow 后,16 次函数评估仍优于自回归基线,batch size 1 下生成 1024 token 快 15 到 19 倍[1]

相关 · 论文

本页内容由 AI 自动聚合公开信源生成,仅供了解行业动态参考,不构成任何投资或决策建议。如需引用请以原文出处为准。