動画生成の疎な注意計算をGPUごとに割り当てるTessera
Decoupling Logical Masks from GPU Execution for Dynamic Block-Sparse Attention
この論文をやさしく読む
ひとことで言うと
動画生成モデルの疎な注意計算を、マスクの意味を保ったまま各GPU向けに効率よく実行する仕組み。
何に役立つ？
動画拡散モデルでBSAを使う際の実行時間短縮に役立つ可能性がある。報告された最大6.79倍は評価対象のBSA要求に対する値。
この研究の面白いところ
注意すべき相手を決める論理マスクと、GPUが処理するタイル配置を分離し、事前計測に基づいて実行方式を選ぶ。
どこまで分かった？
評価はNVIDIAの4世代のGPU、2,315マスク、対象動画モデルに基づく。動画生成全体の速度や他のGPUで同じ倍率になるとは要旨に示されていない。
v1のアブストラクトに基づくAI解説。日本語訳とは別に、用途の解釈を含みます。
アブストラクトの日本語訳
反復的なノイズ除去で動画を生成する動画拡散Transformerでは、注意計算が推論の負荷を高める。ブロック疎注意（BSA）は論理マスクで選んだブロックだけを計算して負荷を下げるが、論理ブロックの形状と実行方法を結び付けると、マスクやGPUの違いに適応しにくい。また、実行時にカーネルを特化させる準備の負荷が、計算時間の節約を上回ることがある。本研究は、指定された注意の相互作用を維持しながら、論理マスクとGPU上の実行を切り離す動的BSA用ランタイムTesseraを提案する。物理マッピング層は論理ブロックを保持、結合、分割してマスク形状やGPUに合うタイルへ変換する。タスク編成層はタイルをまとめて割り当て、データ再利用、並列化、転送と計算の重ね合わせを図る。さらに、事前のプロファイリングで作った検索表により、準備負荷を抑えて実行方式を選ぶ。NVIDIAの4世代のGPUに対応する専用CUDAカーネルとして実装し、実際の注意マスク2,315個と産業用動画拡散モデルで評価した結果、対象モデルのBSA要求で従来システムに対し最大6.79倍の高速化を得た。
v1の要旨から自動生成。本文の精読・人による確認は未実施。
- 初稿
- 2026-09-22（UTC）
- 最新改訂
- 2026-09-22 · v1
- 査読・掲載
- 査読状況未確認
更新履歴
- v1 2026-09-22 この版を読む
取得できた版を表示。版の更新は査読済みを意味しません。過去版の本文差分は未解析です。
原文の要旨
Attention computation makes inference expensive in video diffusion transformers (vDiTs), which generate videos through iterative denoising. Block-sparse attention (BSA) reduces this cost by computing only blocks selected by a logical mask, which specifies attention interactions to compute. However, coupling logical block geometry to execution choices limits adaptation to varying masks and graphics processing units (GPUs), while runtime kernel specialization can incur preparation overhead that outweighs execution time savings. We present Tessera, a specialized runtime for dynamic BSA that decouples logical masks from GPU execution while preserving specified attention interactions. Its physical mapping layer retains, combines, or subdivides logical attention blocks into physical tiles suited to different attention mask shapes and GPU architectures. Its task organization layer groups and schedules tiles within GPU tasks to reuse data, expose parallelism, and overlap data movement with computation. Finally, profile-guided regime selection enables low- overhead execution plan selection through a lookup table constructed from offline profiling. We implement Tessera with specialized CUDA kernels supporting four NVIDIA GPU generations. Evaluated on 2,315 real attention masks and industrial video diffusion models, Tessera achieves up to 6.79x BSA request speedup over baseline systems in the evaluated video diffusion models.
著者のコメント
15 pages, 10 figures, 9 tables. Preprint
arXiv ID: 2609.25869 ／ 要約の誤りについて