AInspiro
EN

NVIDIA搞了种新架构:大模型不再一次蹦一个词,速度翻2.4倍

技术趋势
🤖 本文由 AI 生成,内容仅供参考。

你见过的大模型,都是一次吐一个词

GPT、ClaudeGemini——不管哪个模型,生成文本时都是一个token接一个token往外蹦。这是"自回归"架构,从2018年GPT-1开始就是这个套路。

7月2日,NVIDIA开源了 Nemotron-Labs-TwoTower,一个用扩散方式生成文本的语言模型。简单说:它不再是逐词生成,而是一次性处理整个输出序列。

结果是:速度翻了2.42倍,质量保持了98.7%。

双塔架构:冻结一半,加速一半

NVIDIA的工程师没有从头训练一个新模型。他们拿Nemotron-3-Nano-30B(一个30B的自回归模型),把它拆成了两个"塔":

  • 冻结上下文塔:原来的30B参数原封不动,负责理解输入,不参与生成
  • 去噪器塔:新增30B参数,专门负责扩散式生成文本

总参数60B,但计算量没有翻倍。因为上下文塔是冻结的,推理时不用更新。

这个设计的聪明之处在于:没有抛弃自回归模型的能力,而是在此基础上叠加扩散能力。你既可以用原来的自回归模式,也可以切到扩散模式。两种都行。

具体快在哪

自回归模型的瓶颈是"等待"——每生成一个token,都要等上一个token算完。扩散模型是并行生成整个序列,所以没有这个延迟。

NVIDIA的实测数据:用1个block生成256个token时,速度提升约1.8倍。用32个block生成8K个token时,速度提升超过4倍。

越长的文本,加速效果越明显。

三种解码模式都能用

  • 纯扩散:最快,但可能有小噪声
  • 自回归+扩散混合:扩散形成骨架,自回归精调
  • 投机解码:自回归快速生成草稿,扩散验证

还没到完美

98.7%的质量保留率听起来不错,但剩下的1.3%是什么?简单任务几乎看不出区别,但高度结构化的文本——代码、数学推导、格式严格的文档——偶尔会出小偏差。

另外,扩散模式目前只支持批量生成,交互式对话场景里你还得用自回归。但NVIDIA已经在研究把扩散做到流式输出。


为什么这事重要

如果扩散LLM这条路走通,整个推理成本结构都要改写。

现在跑一个大模型生成回复,成本主要花在"一个token接一个token解码"上。如果扩散模式能大规模化,同样的硬件能服务更多用户,或者同样的用户体验只花一半成本。

而且NVIDIA选择开源这个模型——Apache 2.0许可——意味着其他实验室可以直接基于TwoTower做二次开发。扩散LLM的门槛被拉低了。

这不是取代自回归,而是给LLM架构增加了一个新选项。对需要高吞吐量的场景——批量文本生成、代码补全、内容翻译——这个选项很值钱。