别再说推理优化只能靠堆卡了。英伟达的工程师们刚刚扔出一个叫TwoTower的模型,它用一种近乎“寄生”的方式,在自家一个已有的、巨大的自回归模型骨架上,嫁接了一个轻巧的扩散解码头。结果?在几乎不损失生成质量的体感下,吞吐量硬生生拉高了2.4倍。这不是在实验室里的数据游戏,而是一个开源、可商用的真家伙,目标直指那些被海量批量推理任务卡住脖子的团队。
一个“非主流”的造物:双塔如何分工?
传统的自回归(AR)模型像一位一丝不苟的书法家,必须一笔一画、从左到右地完成整篇作品,速度受限于每一笔的“思考”时间。而扩散模型更像一位印象派画家,可以同时在画布的多个区域打底、晕染、逐步聚焦细节,理论上能实现并行化。TwoTower的野心,就是把后者的并行优势,嫁接到前者的强大基础能力之上。
冻结的骨架:价值25万亿个token的沉淀
模型的第一部分,是那个被称为“上下文塔”(Context Tower)的庞然大物。它基于NVIDIA自家的Nemotron-3-Nano-30B-A3B,一个用惊人的25万亿(25T)token预训练出来的自回归骨干。关键决策在于:将它完全冻结。这意味着工程师们锁住了这个模型近300亿参数所蕴含的、对语言、逻辑和世界知识的全部理解。这个骨架是“无价”的,因为它承载了海量算力换来的认知,任何不当的改动都可能带来不可预知的损伤。
流动的造物:专注“从噪声中生成”的降噪器
与冻结骨架相对的是“降噪器塔”(Denoiser Tower),它是唯一被训练的部分。其使命单一而纯粹:学习如何从一团随机的“噪声”中,逐步恢复出合理、连贯的文本序列。这个塔的参数规模更小,活跃参数约30亿。它的训练数据是约2.1万亿token,远少于骨干的训练量。这种设计精妙地实现了“分工”:让最贵、最强的部件负责理解上下文和提供“语义蓝图”,让新训练的、更轻量的部件专职高效执行“生成动作”。
近乎无损的质量:98.7%的秘密
最令人惊讶的数字不是加速倍率,而是质量保留率:98.7%。这意味着,用新方法生成的文本,在各种评估指标上,与原版昂贵的自回归模型输出相比,差异微乎其微。这打破了过去很多尝试中“加速必损质”的魔咒。
层对齐与状态播种:连接两个世界的桥梁
实现这一奇迹的关键技术是“层对齐交叉注意力”和“状态播种”。你可以把它想象成一个精密的翻译器。冻结的AR模型在处理输入时,每一层都会产生一种内部“状态”或“理解”。TwoTower没有试图让降噪器去重新学习这些理解,而是通过交叉注意力机制,像架设了一座座桥梁,将AR模型每一层的实时状态,精准地“播种”到降噪器的对应位置。降噪器不是在凭空创作,而是在AR模型时刻不断的“指导”和“修正”下进行扩散去噪。这确保了生成的内容始终在骨干模型所定义的“合理语义空间”内,从而严控了质量流失。
质量的悖论:为什么减少迭代可能更好?
自回归模型因为其逐词生成的天性,有时候会陷入一种“过度思考”的循环,导致输出变得冗长或重复。而扩散模型的去噪过程,在拥有强大语义引导的情况下,反而可能因为“一步到位”的并行生成,产出更凝练、更直接的结果。TwoTower保留了98.7%的质量,在某些评估维度上,可能不是一种损失,而是一种特性——它过滤掉了AR模型偶尔产生的“思维毛刺”,提供了另一种风格但同样高质量的输出。
吞吐量飞升:并行解码的胜利
2.42倍的吞吐量提升,是扩散模型架构优势最直接的体现。在典型的批量推理场景下,这几乎等同于将算力成本直接砍半。
块状生成与投机并行
TwoTower支持一种名为“块大小S=16”的生成模式。这意味着模型不再是一个token一个token地生成,而是以16个token为一个“块”进行并行推测和验证。同时,它结合了类似“投机解码”(γ=0.8)的技术,让小型的降噪器塔快速生成多个候选“草稿”,再由强大的AR骨干进行高效验证。这种“粗生成-快验证”的流水线模式,彻底打破了自回归的序列依赖瓶颈,让GPU的并行计算单元得以满负荷运转。
三种模式的灵活切换
这个模型并非只有“扩散”一种玩法。它还支持“模拟AR模式”和纯粹的“AR模式”。模拟AR模式可以在扩散框架下模拟自回归的生成逻辑,用于兼容现有基于AR的某些应用或评估流程。这种设计极具工程智慧:它允许用户在追求极致吞吐(扩散模式)和确保与旧系统行为一致(模拟AR/纯AR模式)之间灵活选择,降低了新技术的采纳门槛。
工程哲学与开源价值
TwoTower的发布,透露出NVIDIA在生成式AI领域一种愈发清晰的工程哲学:不追求从零开始的炫技,而是专注于对现有成熟资产的“增效”与“加速”。
复用,而非重建
在动辄需要万亿token和天量算力预训练大模型的今天,TwoTower展示了一条更经济的路径。它复用了已有的、经过海量数据验证的Nemotron-3-Nano-30B-A3B作为认知基座,只投入相对少量的算力训练一个专用的“加速器插件”。这种“骨干不变,插件增强”的模式,对于拥有强大基座模型但苦于推理成本的公司来说,吸引力巨大。它意味着提升服务响应和降低成本,不一定需要等待下一代基座模型,可以通过工程创新立刻实现。
开源生态的关键拼图
该模型的开源与商用许可,是其影响力的放大器。它不仅仅是发了一篇论文或一个Demo,而是给出了一个可直接部署、验证和集成的生产级工具。对于无数面临推理算力压力的团队,这相当于英伟达递过来一把经过自家工厂测试的“瑞士军刀”。它将扩散语言模型从一个前沿研究话题,直接推到了工程优化的主战场。当主流的路径在卷模型规模和数据量时,TwoTower在另一条赛道——推理效率上,树立了一个新的、触手可及的标杆。这或许预示着,大模型竞争的下半场,重点将从“谁更强”转向“谁更会省”。

