k8凯发天生赢家

新闻中心 新闻中心

全球初次单机征服万亿巨模DeepSeek-V4!RL后训练框架Orbit开源!

从数学、代码、复杂推理 ,到多

作者:谢博钧
颁布功夫:2026-06-02 11:55:37
阅读量:9236

全球初次单机征服万亿巨模DeepSeek-V4!RL后训练框架Orbit开源!

从数学、代码、复杂推理 ,到多轮工具挪用 ,大模型的好多能力的提升都离不开 RL 后训练 。但当模型规模进入 MoE 万亿参数级别之后 ,RL 不再只是一个算法问题 ,同时越发是一个系统问题 。

训练侧必要包容重大的模型权重、梯度和优化器状态;rollout 侧必要持续天生样本 ,并维持足够高的吞吐;reference policy 又会进一步放大显存和调度压力 。同时 ,好多 RL 系统在训练时使用较高精度模型 ,而真正 rollout 或部署时使用低精度 serving 模型 。这些精度差距 ,最终会体此刻部署成效与 RL 成效的不一致上 。

通过将 base model 固定在部署时使用的低精度暗示 ,并只更新 adapter ,Orbit 将 Kimi-K2.6、DeepSeek V4 级此外 1T 模型 RL 后训练压缩到单台 8×B200 上实现 。同时 ,训练和 rollout 使用统一条低精度 base + adapter 蹊径 ,从系统层面解除了训练模型与 rollout / 部署模型之间的精度不一致 。

Orbit 做到「让万亿模型进入单节点 RL 区间」这件事的意思在于:

预防了「训练精度」和「部署精度」不一致带来的误差 ,从而带来更不调换高效的 RL 后训练;单节点 RL 能够显著降低多节点训练时的通讯时延与故障率;在同样的 HBM 预算下 ,模型会获得更宽的训练空间 ,从前必要多卡能力训的模型 ,有机遇被压缩到单卡 。

官方博客:https://spherelab.ai/orbit/Github:https://github.com/Sphere-AI-Lab/orbit

Orbit:支持万亿参数模型 RL 微调的高效框架

显存节造:如下图 1 所示的估算中 ,单节点 8×B200 的 HBM 预算约为 1536GB 。对 1T 级模型而言 ,传统全参微调的 weight + grad 显存下界会远超单机预算;而 Orbit 蹊径由于冻结低精度 base ,只训练 adapter ,能够把 1T 级模型的 RL 后训练放进单节点预算内 。

图 1 分歧框架下大参数模型的单节点显存需要估算

训推精度对齐:在好多 RL 系统里 ,训练侧可能使用 BF16 或 FP8 等高精度  ,而推理侧使用 INT4、FP4 等低精度 。对于监督微调来说 ,这种差距有时能够被视作推理优化的一部门;但在 RL 中 ,policy log-prob 自身就是训练信号的一部门 ,训练侧和推理侧之间的误差 log-prob diff 会直接影响不变性 。

Orbit 将这一问题前置到了系统设计中:训练和推理使用一样的低精度 base  ,并在其上加载统一个 BF16 adapter ,从而维持训推精度一致 。

Adapter-first 的系统设计:Orbit 萦绕 adapter 对 RL 训练、推理、同步、reference policy 和低精度 MoE 做了一套整体设计 。base 始终冻结 ,每次训练更新后 ,只必要将 MB 级 adapter (不需将 GB 级的 base)从训练引擎推送到推理引擎 。这不仅削减了权重同步的体积 ,也预防了频仍重建推理引擎的开销 。

单节点 Kimi-K2.6 了局

在这组尝试中 ,模型运行在单台 8×B200 上 ,训练精度为 INT4 base + BF16 adapter ,rollout 精度使用一样的 INT4 base + BF16 adapter 。也就是说 ,训练和 rollout 走的是统一条低精度 base + adapter 蹊径 。

在约 200 step 的 RL 过程中 ,Orbit 观察到了几个同时成立的信号:

reward 上升;eval accuracy 上升;pass@k 上升;train-rollout log-prob diff 维持不变 。

图 2 Kimi-2.6 在 Orbit 下单机 RL 后训练信号

图 3 Kimi-2.6 在 Orbit 下单机 RL 后训练的显存纪录

图 2 显示 ,Kimi-K2.6 的 rollout raw reward、eval accuracy 和 pass@k 曲线随训练推动而不变上升 。同时 ,train-rollout log-prob diff 不变维持在一个区间 。

对于一个对 log-prob 差距极度敏感的训练范式来说 ,这些信号现实地证了然 Orbit 的 RL 后训练关环不仅在单机上把 1T 的模型上不变能跑 ,同时跑对了且在测试工作上有成果 。

单节点 DeepSeek V4 Flash 了局

在这组尝试中 ,DeepSeek V4 Flash 同样运行在单台 8×B200 上 。训练精度为 FP4 base + BF16 adapter ,rollout 精度也使用一样的 FP4 base + BF16 adapter 。

图 4 DeepSeek V4 Flash 在 Orbit 下单机 RL 后训练信号

图 5 DeepSeek V4 Flash 在 Orbit 下单机 RL 后训练的显存纪录

从了局看 ,DeepSeek V4 Flash 在 100 step 以上的 RL 过程中同样维持不变:reward、eval、pass@k 整体上升 ,train-rollout log-prob diff 维持在不变区间 。这些趋向跟在 Kimi-K2.6 上的尝试了局类似 。

单节点 1.6T DeepSeek V4 Pro 初步验证

除了 Kimi-K2.6 和 DeepSeek V4 Flash 两组不变有效的训练了局 ,Orbit 还在 DeepSeek V4 Pro 1.6T 上实现初步验证 。

由于 DeepSeek V4 Pro base model 自身很强 ,尝试中用的 RL 训练数据不能让它涨点 ,因而该尝试更多是证明 Orbit 的系统蹊径能够扩大到更大的 1.6T 级 MoE 模型 。

图 6 DeepSeek V4 Pro 在 Orbit 下单机 RL 后训练信号和显存纪录

在 1.6T DeepSeek V4 Pro 上 ,Orbit 实现了单节点 8×B200 的尝试 ,展示了不变的 train-rollout log-prob diff 和可控不变的 GPU 显存 。

这组了局证明Orbit 的系统上限可在单节点 8×B200 达到 1.6T 级别 ,展示了其设计有机遇覆盖更大的 MoE 模型区间 。

从单节点万亿模型 ,到单卡更大模型

单节点跑通万亿模型 RL 反过来也说了然同样的硬件预算就能够覆盖更大的模型区间 。

对万亿模型来说 ,这意味着正本可能必要多机协同的 RL 后训练 ,能够被压缩到单节点实现 。对中幼模型来说在 Orbit 的 adapter-first 框架下 ,单卡也有机遇 RL 微调从前必要多卡能力支持的模型 ,或者在一样模型规模下支持更长 response、更大 batch、更高 rollout throughput 和更频仍的更新 。

因而 ,Orbit 的价值并不只在于「让大模型变得可训练」 ,也在于让幼模型的 RL 后训练变得更容易 。

技术细节

Active-expert-chunked dequantization:对于 MoE 模型来说 ,每个词元只会激活部门 experts 。Orbit 动态地将 router 选中的 experts 分组成固定大幼的 batch ,一时反量化后执行 grouped GEMM ,并在推算实现后开释高精度权重 。这样既能利用 grouped matrix multiplication 的吞吐 ,又能将一时显存峰值限度在较幼 chunk 内 ,预防大规模低精度 MoE 训练中的 OOM 。

Adapter-native async with double-buffered rollout:系统会为 adapter 守护版本号 ,并将新版本 adapter 流式写入 inactive slot;当前 active slot 持续服务 in-flight 要求 ,待新版本筹备好后再原子切换 。这样能够削减 rollout bubble 。在 Qwen3-4B + OFT、8×B200、TP=2 设置下 ,该设计带来了 1.42 倍的单步功夫优化和 44% 更高的 rollout throughput ,同时 eval accuracy 维持不变 。

DeepSeek V4 有关优化:Orbit 支持 Full-CUDA graph decoding、DeepGEMM、DeepEP V2 ,并使用 tilelang / Triton / CUDA 实现高效 attention backward 和 fusion kernels 。凭据 adapter 训练的特点 ,Orbit 还设计了 bypass-base-weight-grad 的高效 GEMM backward 算子 ,预防为冻结 base 推算不用要的梯度 。

从前 ,大模型 RL 后训练往往意味着更复杂的多机系统:更多节点、更重的权重同步和更复杂的系统协同 。

Orbit 提供了另一条蹊径:冻结低精度 base ,只训 adapter ,让训练、rollout 和部署对齐 ,并把整模同步换成 adapter 同步 。这让万亿模型能够进入单节点训练区间 ,更幼模型也能在单卡或更有限的硬件上跑得更远 。

从 Kimi-K2.6 到 DeepSeek V4 Flash ,再到 DeepSeek V4 Pro 1.6T ,Orbit 展示和提供了一套面向大模型后训练的高效框架 。

 

文章点评

未查问到任何数据!

颁发评论

◎迎接参加会商 ,请在这里颁发您的见解、互换您的概想 。

最新文章

热点文章

随机推荐

【网站地图】