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

无损并行生成token,Uno让LLM推理最高提速3倍

0
分享至

大语言模型的推理速度,一直是落地应用的瓶颈。传统方法一次只能生成一个token,就像排队过闸机,效率上不去。现在,一篇新论文提出了一种名为Uno的扩散增强LLM,实现了无损的并行token生成,在多个评测场景下,速度最高提升到基础模型的3倍

这项工作的核心思路,是绕开现有加速方案的两难困境。此前流行的投机解码(speculative decoding)方法,通常需要一个额外的、更小的草稿模型来预测多个token,再由大模型验证。而扩散LLM(diffusion LLM)虽然能并行生成,但往往以牺牲生成质量或模型能力为代价。Uno的贡献在于,它同时避免了这两种代价。


Uno的工作原理:一套参数,两种用途

Uno被定义为一类扩散增强的大语言模型。它首先定义一个自回归(autoregressive)模型分布,然后利用扩散过程,从这同一个分布中并行抽取多个token。关键在于,模型的参数被分成了两部分:一部分是标准的自回归权重,按照常规的下一token预测任务进行训练;另一部分则是一组轻量级的扩散权重,通过一个简短的蒸馏阶段加入。

这个蒸馏阶段带来的额外开销很小,几乎可以忽略不计,却能无缝集成到现有的训练流程中。更重要的是,因为采样器直接从自回归分布本身进行采样,所以这种加速是无损的。这意味着,一个现有的开源权重自回归模型,可以直接升级为Uno,而无需从头重新训练。

实测性能:全面领先,最高3倍加速

论文给出的评测数据显示,Uno在每一个评估的批次大小下,都优于主流的投机解码方法,包括设备支持的最大批次。相比基础模型,Uno最高能实现3倍的推理加速。

在模型规模的对比上,一个8B参数的Uno模型,在智能体工具使用、代码生成和长上下文推理等任务上,表现超过了26B参数的DiffusionGemma,以及专有的Mercury 2模型。这组对比直观地展示了Uno在效率与性能平衡上的优势。

这项技术的意义:推理成本的重新计算

对于开发者而言,这项技术的价值在于,它提供了一条清晰的升级路径。不需要推翻现有的模型架构,不需要重新训练庞大的模型,只需在现有模型上增加一个轻量级的蒸馏步骤,就能获得显著的推理速度提升。这直接关系到推理成本的计算——同样的硬件,单位时间内能处理更多的请求,或者用更少的硬件支撑同样的业务量。

论文已经公开在arXiv上(编号:2609.04010),相关讨论也可以在dair.ai的论文社区中找到。对于正在为推理延迟和成本头疼的团队来说,Uno提供了一个值得关注的方案。

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

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.

相关推荐
热点推荐
5球大战,国米 3‑2 那不勒斯;国米主帅齐沃:“我们的勇气得到了回报”

5球大战,国米 3‑2 那不勒斯;国米主帅齐沃:“我们的勇气得到了回报”

福酱的小时光
2026-09-06 05:22:10
订婚宴上,老公用德语宣布婚礼当天新娘会另有其人,他以为我听不懂,当我用德语送上祝福他傻眼

订婚宴上,老公用德语宣布婚礼当天新娘会另有其人,他以为我听不懂,当我用德语送上祝福他傻眼

晓艾故事汇
2026-09-03 17:02:23
特朗普接班人已明朗?美国或出现历史上第1个,被中国制裁的总统

特朗普接班人已明朗?美国或出现历史上第1个,被中国制裁的总统

解锁世界风云
2026-09-05 18:01:03
上央视了!事发密云,目前已刑拘25人!

上央视了!事发密云,目前已刑拘25人!

在密云
2026-09-05 19:08:42
父亲用扁担送双胞胎姐妹上大学!俩人高考差3分,考入同一大学,住同一寝室

父亲用扁担送双胞胎姐妹上大学!俩人高考差3分,考入同一大学,住同一寝室

封面新闻
2026-09-04 23:39:05
事业没了、婚也离了,被封5年后赵薇再曝近况,瘦到脱相不敢认

事业没了、婚也离了,被封5年后赵薇再曝近况,瘦到脱相不敢认

阿讯说天下
2026-08-30 06:10:34
不出意外,CBA新赛季这四支球队成绩或将下滑

不出意外,CBA新赛季这四支球队成绩或将下滑

男足的小球童
2026-09-05 15:21:07
68.42米!严子怡夺得钻石联赛总决赛女子标枪冠军

68.42米!严子怡夺得钻石联赛总决赛女子标枪冠军

观察者网
2026-09-06 06:34:11
2000美元起售、折叠痕却没消掉!iPhone Ultra被嘲「终极身份象征」:苹果努力了这么久,还是败给了物理规律

2000美元起售、折叠痕却没消掉!iPhone Ultra被嘲「终极身份象征」:苹果努力了这么久,还是败给了物理规律

泡泡网
2026-09-03 10:59:09
事情闹大了,首尔紧急发声,宁得罪自己人,也不能落得高市的下场

事情闹大了,首尔紧急发声,宁得罪自己人,也不能落得高市的下场

正经的烧杯1
2026-09-06 05:38:04
我妈出轨20年,我爸忍耐了20年,我妈50岁生日时我才知道他有多狠

我妈出轨20年,我爸忍耐了20年,我妈50岁生日时我才知道他有多狠

千秋文化
2026-08-12 20:13:55
顾不上伊朗了?特朗普警告:美国若不做这件事,中国就要赢了!

顾不上伊朗了?特朗普警告:美国若不做这件事,中国就要赢了!

乌克兰小静
2026-09-05 08:36:07
心理学:从吃饭看家境,穷女人吃饭时往往暴露这两个习惯细节

心理学:从吃饭看家境,穷女人吃饭时往往暴露这两个习惯细节

心理观察局
2026-09-06 07:24:10
麒麟儿,海魂衫,还有不知所终的少年心

麒麟儿,海魂衫,还有不知所终的少年心

澎湃新闻
2026-09-04 20:46:09
中国空军太丢脸:预警机被骑脸,歼20团灭,为何却让14亿人叫好?

中国空军太丢脸:预警机被骑脸,歼20团灭,为何却让14亿人叫好?

蜉蝣说
2026-08-16 21:59:54
三星新机突然曝光:9月6日,这配置把我整不会了

三星新机突然曝光:9月6日,这配置把我整不会了

手机讲坛
2026-09-06 00:09:18
再过8天,一场席卷全国的大“变天”就要来了9月7号白露正式上岗

再过8天,一场席卷全国的大“变天”就要来了9月7号白露正式上岗

周哥一影视
2026-09-05 06:42:33
心理学:对待已经产生隔阂的人,最好的做法,不是主动示好,而是这两个字

心理学:对待已经产生隔阂的人,最好的做法,不是主动示好,而是这两个字

心理观察局
2026-09-05 06:52:08
3-2!2-1!意甲神仙夜!国米让2追3补时绝杀,罗马93分钟绝杀登顶

3-2!2-1!意甲神仙夜!国米让2追3补时绝杀,罗马93分钟绝杀登顶

董老师看竞技
2026-09-06 07:02:09
“不要跟我抢饭碗”走红!一位新老师的群消息,戳中万千家长,网友:遇上这样的老师,是家长的福气,学生的幸运,学校的骄傲

“不要跟我抢饭碗”走红!一位新老师的群消息,戳中万千家长,网友:遇上这样的老师,是家长的福气,学生的幸运,学校的骄傲

火山詩话
2026-09-04 13:53:30
2026-09-06 08:31:00
字节漫游指南
字节漫游指南
有态度网友ytd
215文章数 111关注度
往期回顾 全部

科技要闻

华为何庭波,再次更新韬定律论文

头条要闻

决胜盘一度0比5落后 郑钦文连赢七局上演"史诗级逆转"

头条要闻

决胜盘一度0比5落后 郑钦文连赢七局上演"史诗级逆转"

体育要闻

本西蒙斯加盟国王,不管怎样,回来就好

娱乐要闻

阿Sa蔡卓妍首度求婚细节,感动落泪

财经要闻

被曝试药后死亡,药企董事竟怒怼记者

汽车要闻

带升降立标MPV 岚图梦想家9预售价42.99万起

态度原创

游戏
房产
手机
本地
公开课

发售时间迎来新线索!《龙珠》新作或明年初就能玩到

房产要闻

突发重磅!海口出台楼市新政!

手机要闻

华为Pura 90系列未加入涨价潮:价格最稳的华为影像旗舰

本地新闻

扒完小作文,富豪们私藏的度假胜地有多绝

公开课

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

无障碍浏览 进入关怀版