首页 > 教程攻略 > ai资讯 >英伟达开源 TwoTower AI 模型:保留 98.7% 质量,Token 生成提速 2.42 倍

英伟达开源 TwoTower AI 模型:保留 98.7% 质量,Token 生成提速 2.42 倍

来源:互联网 时间:2026-07-04 13:03:22

7月3日,英伟达正式发布了一款名为Nemotron-Labs-TwoTower的新模型,并同步在Huggingface上开放权重。这次开源背后的核心思路其实很直接:

要解决大模型推理时Token生成速度的瓶颈问题。

英伟达开源 TwoTower AI 模型:保留 98.7% 质量,Token 生成提速 2.42 倍

先看几个关键参数。模型总参数量60B,采用独特的双塔(TwoTower)架构——一个30B的自回归模型/上下文塔(context tower),外加一个30B的扩散/降噪塔(denoising tower)。每个塔激活3B参数,搭配128个可路由专家。听起来有点复杂,但本质上是将传统扩散语言模型中混杂的网络任务拆开,让两个独立的神经网络各司其职。

具体来说,上下文塔保持冻结状态,专门维护文本的自回归上下文;去噪器塔则负责训练,对噪声块进行去噪。两个塔通过逐层交叉注意力连接协同工作。这个设计的最大亮点在于:分工明确,互不干扰,从而在不牺牲太多质量的前提下,大幅提升生成速度。

那么实际效果如何?英伟达给出的基准测试数据显示,双塔架构保留了98.7%的综合质量,但实际运行时的吞吐量提升了2.42倍。附上对比结果:

任务Nemotron-3-Nano-30B-A3B (AR)Nemotron-Labs-TwoTower (diffusion)
MMLU (5-shot, acc)78.5678.24
MMLU-Pro (5-shot, CoT EM)62.5960.93
ARC-Challenge (25-shot, acc_norm)91.7292.66
WinoGrande (5-shot, acc)76.0976.09
RACE (0-shot, acc)88.9088.90
HumanEval (0-shot)79.2775.58
MBPP-Sanitized (3-shot)74.7174.28
GSM8K (8-shot, acc)92.4990.14
MATH-500 (4-shot)84.4080.60
MMLU Global Lite (5-shot)73.9773.94
MGSM (8-shot, a vg acc)80.8080.40
Quality retained100%98.7%
Generation throughput (× AR)1.0×2.42×

从表格能清晰看到,在大多数任务上TwoTower模型与纯自回归模型持平或略微下降,但生成吞吐量直接翻了一倍多。对于需要高频推理的生产环境来说,这2.42倍的加速意味着成本和时间上的显著优势。授权协议方面,采用NVIDIA Nemotron Open Model License,开发者可以直接在Huggingface上下载权重进行二次开发。