arXiv:2607.22696cs.CVcs.AI2026-07被引 1

让1050亿参数视频模型在单张显卡上高效微调。

MegaSlide-DiT: Memory-Centric Adaptation and Deformable Local Attention for Efficient Video Diffusion

  • 把模型权重全放主机内存,只按需加载部分到显卡。
  • 用动态局部注意力替代全局注意力,内存和计算量线性增长。
  • 适合想在个人工作站上微调大模型的研究者使用。

基于扩散Transformer(DiT)的高分辨率视频生成模型虽保真度高,但迅速耗尽单台工作站的内存预算。一个超过1000亿参数的DiT模型通常需要超过1TB的持久状态,而朴素的时空自注意力机制随序列长度呈二次增长。参数内存与激活内存这两道瓶颈阻碍了研究者在无大型GPU集群的情况下对大规模生成模型进行微调。本文从系统角度重新审视该问题,提出MegaSlide-DiT原型,证明可在仅配备1.5TB主机内存的单张H200 GPU上对预训练的1050亿参数DiT进行完整参数微调。核心思想是:模型的持久权重、主权重和优化器动量全部保留在主机内存中,仅临时分片按需流式传输至显卡。同时,以3D可变形滑动注意力(3D-DSA)替代二次方全局注意力,实现内存与计算复杂度在序列长度上线性化。通过详细的内存统计、执行轨迹和评估结果验证设计有效性。MegaSlide-DiT并非在单卡上从零训练1050亿模型,也未突破带宽限制,而是为高阶工作站上的大规模视频扩散模型提供一条切实可行的全参数微调路径。

原文摘要 · Abstract (English)

High-resolution video diffusion models built on Diffusion Transformers (DiTs) deliver strong fidelity but quickly exhaust the memory budget of a single workstation. A 100 billion-plus parameter DiT easily requires over a terabyte of persistent state, while naive spatiotemporal self-attention grows quadratically in sequence length. These two walls -- parameter memory and activation memory -- prevent researchers from adapting massive generative models without large GPU clusters. We revisit this problem from a systems perspective and introduce MegaSlide-DiT, a prototype that demonstrates how a pre-trained 105B DiT can be adapted on a single H200 GPU with 1.5 TB of host RAM. Our key insight is that the GPU need not own the model state: all persistent weights, master weights and optimizer moments remain in host memory, while only transient shards are streamed to the GPU on demand. Simultaneously, we replace quadratic global attention with 3D Deformable Slide Attention (3D-DSA), a motion-adaptive local attention operator that reduces both memory and computational complexity to linear in the sequence length. We report detailed memory accounting, execution traces and evaluation results to substantiate our design. MegaSlide-DiT does not claim to train a 105B model from scratch on a single GPU, nor does it magically solve bandwidth limits; rather, it offers a pragmatic path for full-parameter adaptation of massive video diffusion models on high-end workstations.

视频生成扩散模型高效微调内存优化

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。