NVIDIA 推出 Jet-Nemotron:53 倍提速的混合架构语言模型实现 98% 的推理成本节约

近日,NVIDIA 的研究团队宣布发布 Jet-Nemotron,这是一系列全新的语言模型 (包含 2 亿和 4 亿参数的版本),其生成速度比当前最领先的全注意力语言模型高出 53.6 倍,并在准确性上达到了甚至超过了这些模型的水平。这一突破不是通过从头开始重新训练模型实现的,而是采用了一种名为 「后神经架构搜索」(PostNAS) 的新技术对现有预训练模型进行了改造。

随着现代语言模型的广泛应用,如 Qwen3、Llama3.2 和 Gemma3 等,这些模型虽然在准确性和灵活性上设立了新的标杆,但其 O(n²) 的自注意力机制造成了计算和内存的高昂成本,尤其是在处理长文本任务时,这使得大规模部署变得异常昂贵,也几乎不可能在边缘设备或内存受限的设备上运行。尽管有一些尝试用更高效的架构替代全注意力 Transformer(如 Mamba2、GLA、RWKV 等),但在准确性上却始终难以实现突破,直到现在。

PostNAS 作为 Jet-Nemotron 的核心创新,主要包括以下几个步骤: 首先,选择一个最先进的全注意力模型 (如 Qwen2.5),并冻结其多层感知器 (MLP) 层,以保护模型的学习能力并大幅降低训练成本; 接着,用新的硬件高效线性注意力模块 JetBlock 替换掉计算成本高的全注意力模块; 最后,通过超网络训练和束搜索,自动确定最优的全注意力层的位置,从而保持在特定任务上的准确性。

Jet-Nemotron 的性能指标令人瞩目: 其 2B 模型在各大基准测试中与 Qwen3-1.7B-Base 相当或更优,并且生成吞吐量提升了 47 倍。同时,在 256K 上下文长度下,解码速度的提升达到 53.6 倍,使得推理成本降低了 98%。这为边缘设备的部署带来了变革性的改变。

此外,Jet-Nemotron 的推出意味着企业能够以更低的成本实现更高的回报率。对于从业者而言,Jet-Nemotron 能够在不改变数据管道的情况下对现有模型进行改造,提升了实时 服务的能力。而对于研究人员而言,PostNAS 降低了语言模型架构创新的成本,加速了 的发展。

项目:https://github.com/NVlabs/Jet-Nemotron

划重点:

🌟Jet-Nemotron 实现生成速度比现有模型提升 53.6 倍,推理成本降低 98%。

💻PostNAS 技术允许对现有预训练模型进行高效改造,保持准确性。

📈新模型的推出使得企业和研究者能在成本和性能上获得双重收益。

声明:本站所有文章,如无特殊说明或标注,均为本站原创发布。任何个人或组织,在未征得本站同意时,禁止复制、盗用、采集、发布本站内容到任何网站、书籍等各类媒体平台。如若本站内容侵犯了原著者的合法权益,可联系我们进行处理。

给 TA 打赏
共 {{data.count}} 人
人已打赏
AI 资讯

IBM 与 AMD 联手冲刺量子计算,打造未来 AI 新生态

2025-8-28 1:20:32

AI 资讯

GPT-5 通关宝可梦,创下 9517 步新纪录,效率惊人!

2025-8-28 1:20:50

个人中心
购物车
优惠劵
今日签到
有新私信 私信列表
搜索