[控场AI]
概念Split-KV CP

Split-KV

一种精确的上下文并行(CP)方法,将 key/value 分片保留在各设备本地,每层注意力仅通信与查询规模相当的张量,使通信量与上下文长度解耦,适用于查询极短、上下文极长的非对称训练场景。

时间轴 (近 90 天)

11月5日

Split-KV 是一种精确的上下文并行(CP)方法,在 CP 规模为 64 时保持 85–91% 的弱扩展效率。

待验证50%
11月5日

Split-KV 相比现有最快的 CP 基线最高实现 3.67 倍加速。

待验证50%
11月5日

Split-KV 的核心机制是将 key/value 分片保留在各设备本地,每层注意力只通信与查询规模相当的张量,从而使通信量与上下文长度解耦。

待验证50%
11月5日

Split-KV 采用代价感知的最长处理时间优先(LPT)策略分配上下文 KV 块,负载不均衡率低于 1%。

待验证50%
11月5日

Split-KV 的 LPT 调度规划速度比最好的负载均衡基线快三个数量级。

待验证50%
11月5日

Split-KV 的代码已在 GitHub 的 verl 仓库开源,地址为 https://github.com/zxgx/verl。

待验证50%
11月5日

Split-KV 是精确(exact)方法,其计算结果与不分片时严格一致,不牺牲数值正确性。

待验证50%

全部知识事实 (7)

来源文章