跳到正文
热点事件持续更新

Split-KV:扩散草稿模型训练的上下文并行方法

1 篇报道1 个报道来源9 小时前更新

先了解这件事

AI 综述

Geng Zhang 等作者发布论文提出 Split-KV,一种面向扩散草稿模型训练的精确上下文并行(CP)方法。该方法将 key/value 分片保留在本地,把每层注意力通信限制在 query 规模的张量上,使通信量与上下文长度无关。 作者报告,在 CP 规模为 64 时,Split-KV 保持 85–91% 的弱扩展效率,相比最快的 CP 基线最高实现 3.67 倍加速;其基于全局 query 视图的 LPT 负载均衡策略将不均衡度控制在 1% 以下,规划速度比最佳基线快三个数量级。

AI 根据报道生成 · 54 分钟前更新

报道时间线

沿着报道,了解事件的不同侧面。

11月5日
  1. arXiv cs.AI
    Split-KV:面向扩散草稿模型训练的高效上下文并行方法

    Split-KV 是一种精确的上下文并行(CP)方法,将 key/value 分片保留在本地,把每层注意力通信限制在 query 规模的张量上,使通信量与上下文长度无关。在 CP 规模为 64 时,它保持 85–91% 的弱扩展效率,相比最快的 CP 基线最高实现 3.67 倍加速。其基于全局 query 视图的 LPT 负载均衡策略将不均衡度控制在 1% 以下,规划速度比最佳基线快三个数量级。

本事件热度走势

还没有足够的连续观测数据,暂不绘制趋势。