网易首页 > 网易号 > 正文 申请入驻

并行革命,32倍吞吐量跃升!英伟达Helix架构突破百万Token推理瓶颈

0
分享至


新智元报道

编辑:peter东 英智

【新智元导读】长期以来,大模型受限于有限的上下文窗口,在处理长任务或对话时不得不忘记早期信息,英伟达近日推出的Helix并行技术受DNA结构启发,分割显存和处理任务,显著提升大模型的上下文长度,并发能力,并可降低响应延迟。

想象一个使用大模型的任务,需要一次处理百万字符的文档,例如从百科全书中获取信息,或是分析数百页的法律卷宗,异或追踪持续数月的对话记录,都需要越来越长的上下文。

而大模型生成的每个词都需要扫描存储在所谓的KV缓存中存储的过去标记。

反复读取这个缓存会消耗GPU内存带宽。大模型还需要从内存中重新加载大量的前馈网络(FFN)权重来处理每个新词。

这个过程会减慢效应速度,从而导致用户与大模型对话时出现卡顿。

传统的解决方案,是使用张量并行(Tensor Parallelism, TP)将此负载分散到多个GPU上。但这仅能起到一定作用。

当规模超过一定限度后,GPU开始复制KV缓存,导致内存压力进一步增大。

而Helix这一英伟达针对其最新的Blackwall开发的并行策略,通过将模型Transformer层的注意力机制和前馈网络部分分开处理来解决卡顿问题。

Helix受DNA双螺旋结构的启发,Helix将KV、张量和专家等多个维度的并行性交织到一个统一的执行循环中。

每个阶段在其自身的瓶颈配置下运行,同时复用相同的GPU池。


论文链接:https://d1qx31qr3h6wln.cloudfront.net/publications/Helix_0.pdf

Helix是怎么做到百万上下文不卡顿

在注意力阶段,Helix使用一种名为KV并行(KVP)的新方法,将庞大的KV缓存分散到多个GPU上。

当TP超过KV头的数量时,张量并行会进行复制,从而增加了内存和带宽开销,如图1a到c描述的过程。

Helix通过将TP=2与KVP=2相结合,形成2D布局来避免内存和带宽开销的增加,对应图1d。


图1:传统的张量并行(TP)与Helix的不同注意力分片策略KVP的对比示意图

同时,由于KVP GPU持有与其本地KV头相关联的所有查询头,并冗余地计算QKV投影。

这使得每个KV分片能够进行完全本地的FlashAttention,确保了模型的推理精度。

之后KVP GPU之间沿着查询头维度进行单对单的全连接通信,通信的成本和KV缓存的大小无关,因此大模型的上下文长度即使扩展到百万token,也不会影响查询效率。

此外,Helix还通过重叠通信和计算,一旦计算出一个token的注意力输出,Helix就会启动该token的全对全交换,同时计算下一个token的注意力。

这种紧密的重叠将通信延迟隐藏在有用的工作之后,保持GPU利用率高,并进一步加速实时解码。

图2中上图的八个请求会同步执行注意力计算。随后进行顺序的全对全通信。

图2表底部对应使用HOP-B时,一个请求的通信与下一个请求的计算重叠,通过细粒度流水线减少了token间的延迟。


图2:Helix通过细粒度流水线技术加速大模型的响应

引入Helix带来的高并发和低延迟

根据英伟达官网给出的计算,使用DeepSeek-R1 671B模型,在给定延迟下,当并发的用户数增大时,Helix相比传统方法体现出优势。

而到了图中第一个箭头标注的点时,其单GPU产出的token数是传统方法的32倍,这意味着可以将并发用户数量提高高达32倍。


图3:使用100万上下文长度的DeepSeek-R1,评估使用经过最新NVIDIA GB200 NVL72(Blackwell)在固定延迟下的并发能力

在低并发设置下,Helix可以通过减token与token间的最低延迟时间,来提高用户交互体验,如图3右下方的对比所示。

该研究的参与者St-Maurice指出「Helix正在重塑我们处理LLM交互和设计的方式。」

他指出,Helix并行处理和优化的KV缓存分片正在为大模型提供可扩展的显存外挂,这与开发者改进旧处理器(如奔腾)的方式高度相似。

该技术能允许大模型应用扩展其用户规模的同时,保证其快速响应。

对于虚拟助手、法律机器人以及AI Copolit等应用,Helix的引入可以做到既处理大量工作负载,同时还保持低延迟响应能力。

Helix是否为画靶射箭的争论

对于这项技术突破,西北人工智能咨询公司的首席执行官兼联合创始人Wyatt Mayham表示:「英伟达的数百万个token的上下文窗口是一项令人印象深刻的工程里程碑,但对于大多数公司来说,它是一个寻找问题的解决方案,它解决了现有模型如长上下文推理和二次扩展等真实限制,但技术可能性和实际实用性之间存在差距。」

Mayham承认Helix在特定领域中很有用,例如需要完整文档保真度的合规性强的行业,或医疗系统一次性分析患者终身病史。

但这只是部分特例,大多数组织最好是构建更智能的流水线,而不是购买helix所需的Blackwell架构下的GB200机架。

且通常情况下,检索增强生成(RAG)系统能够在百万个token的范围内,表现的比将上下文长度提升到100k更好。

而Info-Tech研究集团技术顾问Justin St-Maurice则指出:在当今世界,为人类生成百科全书大小的回答并不是胜利。

相反,关键在于使大模型的输出对其他人工智能相关且可用。

这种能力可能成为未来智能体进步的推手。

有了当大模型的输出能具有对应的认知框架,智能体可以保持更丰富的内部状态,参与更复杂、更长时间的聊天,并执行更深入文档分析。

St-Maurice指出:Helix带来的长上下文窗口,能够支持context engineer(上下文工程)在庞大的上下文窗口中管理和优化信息,以最大限度地提高智能体的有效性和可靠性。

凭借在扩展的上下文窗口中处理和交换更大数据量的能力,AI智能体可以以以前不切实际的方式沟通和协作,从而改变多智能体应用的设计框架。

参考资料:

https://research.nvidia.com/publication/2025-07_helix-parallelism-rethinking-sharding-strategies-interactive-multi-million

https://www.computerworld.com/article/4019170/new-nvidia-technology-provides-instant-answers-to-encyclopedic-length-questions.html

https://d1qx31qr3h6wln.cloudfront.net/publications/Helix_0.pdf

https://interestingengineering.com/innovation/nvidia-helix-breakthrough-long-context-ai?utm_source=chatgpt.com

https://developer.nvidia.com/blog/asking-an-encyclopedia-sized-question-how-to-make-the-world-smarter-with-multi-million-token-real-time-inference/?utm_source=chatgpt.com


特别声明:以上内容(如有图片或视频亦包括在内)为自媒体平台“网易号”用户上传并发布,本平台仅提供信息存储服务。

Notice: The content above (including the pictures and videos if any) is uploaded and posted by a user of NetEase Hao, which is a social media platform and only provides information storage services.

相关推荐
热点推荐
向太爆知名女星被骗婚!假豪门老公身家就一辆车,婚礼酒席还逃单

向太爆知名女星被骗婚!假豪门老公身家就一辆车,婚礼酒席还逃单

君笙的拂兮
2025-09-13 16:13:31
5-3,36岁国米旧将佩里西奇发威,率队掀翻领头羊,一夜升至第一

5-3,36岁国米旧将佩里西奇发威,率队掀翻领头羊,一夜升至第一

侧身凌空斩
2025-09-14 02:44:52
“牡丹花下死,做鬼也风流”!如今,谁也救不了44岁的宋小宝

“牡丹花下死,做鬼也风流”!如今,谁也救不了44岁的宋小宝

山河月明史
2025-07-08 14:24:38
湖南37岁女医生佟佳雨突然去世,前后仅3天,死因曝光,结婚7个月

湖南37岁女医生佟佳雨突然去世,前后仅3天,死因曝光,结婚7个月

鋭娱之乐
2025-09-14 22:22:04
1932年,一男子搂着两名原住民美女拍照。两位美女笑得非常自然

1932年,一男子搂着两名原住民美女拍照。两位美女笑得非常自然

百态人间
2025-08-27 11:24:18
斯洛特:我情绪波动很大 5分钟补时不可思议 换科尔克兹是因黄牌

斯洛特:我情绪波动很大 5分钟补时不可思议 换科尔克兹是因黄牌

直播吧
2025-09-15 00:07:27
男子100米再提速非洲决赛八席占一半 亚洲陷入后苏炳添低迷时代

男子100米再提速非洲决赛八席占一半 亚洲陷入后苏炳添低迷时代

劲爆体坛
2025-09-14 22:39:02
5分险胜!26分大胜!德国决赛对手不简单,能否夺冠关键看三分

5分险胜!26分大胜!德国决赛对手不简单,能否夺冠关键看三分

世界体育圈
2025-09-14 12:31:18
一度仅剩200只!2004年,广西岜盆发现1只怀孕的,被家狗咬死

一度仅剩200只!2004年,广西岜盆发现1只怀孕的,被家狗咬死

万象硬核本尊
2025-09-14 18:33:54
43岁我才明白:为什么中国人都用手机支付,在发达国家却无人问津

43岁我才明白:为什么中国人都用手机支付,在发达国家却无人问津

球场指挥家
2025-08-25 18:59:05
女人在发生关系时,最想听到的“下流话”你知道吗?

女人在发生关系时,最想听到的“下流话”你知道吗?

思絮
2025-08-29 08:23:08
女子哭诉停车5小时被收300元,网友惊叹上海物价,官方称收费合理

女子哭诉停车5小时被收300元,网友惊叹上海物价,官方称收费合理

小彭聊社会
2025-09-15 00:14:43
那些被“骗”到威海,花5W买海景房的人!后悔了么?

那些被“骗”到威海,花5W买海景房的人!后悔了么?

背包旅行
2025-09-14 15:03:54
农村惊现一日丧!有邻居去世,两兄弟连挖坑带埋花5000,不欠人情

农村惊现一日丧!有邻居去世,两兄弟连挖坑带埋花5000,不欠人情

火山诗话
2025-09-06 11:18:51
西贝创始人贾国龙道歉

西贝创始人贾国龙道歉

澎湃新闻
2025-09-14 22:59:09
孙怡人设崩塌!吃相难看,知三当三,难怪王京花看不上她

孙怡人设崩塌!吃相难看,知三当三,难怪王京花看不上她

查尔菲的笔记
2025-08-16 15:07:57
为何货币总量已经达到331万亿,超越欧美总和,大家却感觉没钱?

为何货币总量已经达到331万亿,超越欧美总和,大家却感觉没钱?

前沿天地
2025-09-14 20:58:07
糖尿病到最后都是咋去世的?医生叹息:去世前,一般都有2个症状

糖尿病到最后都是咋去世的?医生叹息:去世前,一般都有2个症状

宝哥精彩赛事
2025-09-10 16:48:27
徐文龙,被查

徐文龙,被查

鲁中晨报
2025-09-14 10:07:09
父亲下马后妻子要离婚,我转身接受组织调任省长,她后悔我却畅快

父亲下马后妻子要离婚,我转身接受组织调任省长,她后悔我却畅快

红豆讲堂
2025-09-10 08:35:06
2025-09-15 02:47:00
新智元 incentive-icons
新智元
AI产业主平台领航智能+时代
13445文章数 66149关注度
往期回顾 全部

科技要闻

L3级车型要来了!辅助驾驶迎重大利好

头条要闻

罗永浩向贾国龙重新开战:我看你智力有问题 不可救药

头条要闻

罗永浩向贾国龙重新开战:我看你智力有问题 不可救药

体育要闻

利物浦1-0绝杀十人伯恩利 萨拉赫95分钟点射

娱乐要闻

花泽香菜官宣离婚 结束与老公5年婚姻

财经要闻

西贝贾国龙,“错”得离谱

汽车要闻

混动狂潮 835马力V12 阿斯顿·马丁的最后浪漫

态度原创

数码
游戏
艺术
本地
公开课

数码要闻

上市十天:Intel锐炫Pro B50火速登顶畅销榜首!16GB大显存2999元

《隐形守护者》导演新影游demo15日上线!有金发美女

艺术要闻

故宫珍藏的墨迹《十七帖》,比拓本更精良,这才是地道的魏晋写法

本地新闻

云游中国 | 草原驭秋风 祁连山邂逅黑河源头

公开课

李玫瑾:为什么性格比能力更重要?

无障碍浏览 进入关怀版