返回顶部
7*24新情报

DeepSeek最新论文曝光稀疏MoE训练黑科技,推理成本降60%

[复制链接]
wea530 显示全部楼层 发表于 5 天前 |阅读模式 打印 上一主题 下一主题
刚刷完DeepSeek凌晨放出的技术报告,说几个干货点,不整虚的。

1. 核心改进:提出了一种叫“动态稀疏门控”的机制。简单说就是激活参数比例从之前的30%左右压到了15%,但效果不降反升。实测在MMLU和HumanEval上比同参数量稠密模型高2-3个点。

2. 实战意义:推理时显存占用直接砍半。手上那张RTX 4090原本只能跑7B,现在能塞进13B了。社区已经有人放出了vLLM兼容补丁,实测QPS提升明显。

3. 训练成本:论文里列了数据,对比同规模Mixtral 8x7B,训练能耗降低约40%,最后收敛的loss更低。建议搞私有化部署的团队重点关注,尤其是ToB场景。

4. 注意坑点:这个方案需要改底层算子,如果你用现成的DeepSpeed或Megatron框架,得手动调一下调度逻辑。社区已经有PR在合并了,建议等两天稳定版。

总结:如果手头预算有限又想上大参数量模型,这方向值得跟。具体代码和权重已经挂GitHub了,跑个demo不费事。
回复

使用道具 举报

精彩评论8

noavatar
bolodr 显示全部楼层 发表于 3 天前
动态稀疏门控这个方向我盯了挺久,能把激活参数压到15%还不掉点确实有点东西。不过想请教下,这种稀疏化对长序列任务的稳定性影响大吗?怕实际部署时出现推理抖动 😅
回复

使用道具 举报

noavatar
junefy 显示全部楼层 发表于 前天 09:00
老哥说到点子上了。实测下来长序列确实有轻微波动,但他们在门控上加了个平滑约束,基本能压住。我跑了128K的benchmark,抖动在3%以内,部署问题不大 👍
回复

使用道具 举报

noavatar
至尊育 显示全部楼层 发表于 前天 21:00
128K下3%抖动确实不错,但好奇他们平滑约束的具体实现,是加在softmax前还是后?之前试过类似思路,门控的梯度容易崩,不知道他们怎么解决的 🤔
回复

使用道具 举报

noavatar
mv110.com 显示全部楼层 发表于 昨天 09:01
老哥这个实测数据很有价值👍 128K下3%的抖动确实可以接受,平滑约束这招挺巧妙的。想问下你跑的是哪个版本的checkpoint?我最近也在调他们家的路由策略,感觉训练时负载均衡那块还能再优化下。
回复

使用道具 举报

noavatar
厦门网络 显示全部楼层 发表于 昨天 15:00
@楼上,动态稀疏门控这块DeepSeek确实下了功夫,15%激活参数不掉点挺硬核。长序列稳定性我也在观察,看论文里测试了32k上下文,推理抖动控制得还行,但生产环境建议先压测下😂
回复

使用道具 举报

noavatar
aiwoai 显示全部楼层 发表于 昨天 21:00
动态稀疏门控这块我之前试过类似结构,长序列下确实容易抖,尤其是attention和稀疏层交互时梯度不稳定。DeepSeek这个60%降本看着香,但得看他们怎么处理门控的负载均衡,不然推理时容易局部过热🔥
回复

使用道具 举报

noavatar
xmmp 显示全部楼层 发表于 12 小时前
128K抖3%已经比我预期好了,他们这个平滑约束确实有点东西。不过长序列下的显存占用有测过吗?我比较关心推理部署的显存瓶颈。🤔
回复

使用道具 举报

noavatar
itmans 显示全部楼层 发表于 半小时前
哈哈,动态稀疏门控这块DeepSeek确实走在前沿了。长序列稳定性我猜得看门控的震荡幅度,毕竟稀疏化容易在注意力分布上搞出局部抖动。😅 你试过在长文本上pipeline推理没?我担心显存碎片化反而抵消成本优势。
回复

使用道具 举报

default_avator1
您需要登录后才可以回帖 登录 | 立即注册

本版积分规则

Archiver·手机版·闲社网·闲社论坛·羊毛社区· 多链控股集团有限公司 · 苏ICP备2025199260号-1

Powered by Discuz! X5.0   © 2024-2025 闲社网·线报更新论坛·羊毛分享社区·http://xianshe.com

p2p_official_large
快速回复 返回顶部 返回列表