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

谷歌推出TPU视频扩散时空注意力加速方案

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

先了解这件事

AI 综述

谷歌开发团队针对视频扩散模型在长视频生成中的计算瓶颈,推出了基于TPU的加速方案。在81帧1440p视频生成场景下,自注意力机制耗时占比高达88.2%。针对这一瓶颈,团队通过JAX与Pallas Splash Attention内核,在TPU v6e上成功将稀疏时空注意力转化为硬件推理加速。该方案结合Sparse VideoGen对空间与时间注意力头进行划分,仅保留约38.87%的query-key对,并通过硬件瓦片跳过机制大幅节省了计算与带宽开销。

AI 根据报道生成 · 6 小时前更新

报道时间线

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

10月1日
  1. Google Developers · AI 筛选
    在 TPU 上加速视频扩散模型的时空注意力机制

    通过 JAX 与 Pallas Splash Attention 内核,开发团队在 TPU v6e 上成功将视频扩散模型的稀疏时空注意力转化为硬件推理加速。针对 81 帧 1440p 视频下自注意力耗时占比达 88.2% 的瓶颈,方案结合 Sparse VideoGen 划分空间与时间头,仅保留约 38.87% 的 query-key 对,并通过硬件瓦片跳过机制大幅节省计算与带宽开销。

本事件热度走势

当前热度 9·可比范围峰值 10(10月1日 01:00)·近 24 小时可比范围变化 –

02.557.51010月1日01:0010月1日03:0010月1日04:0010月1日06:00

趋势仅比较持续完整观测到的相同主体,范围可能小于当前热度统计。移动指针或点击图表查看每小时热度;键盘可用左右方向键切换。