arXiv:2511.12940cs.CV2025-11被引 8

用循环神经网络增强扩散模型的长期记忆,生成超长视频更连贯。

Recurrent Autoregressive Diffusion: Global Memory Meets Local Attention

  • 在扩散 Transformer 中加入 LSTM 层,实现全局记忆更新与检索。
  • 在 Memory Maze 与 Minecraft 数据集上显著减少遗忘和时空不一致。
  • 无训练推理差距,支持重叠滑窗全注意力,适合长视频生成任务。

视频生成近期从短视频的双向模型转向超长视频的自回归生成。以往模型多采用滑动窗口注意力以控制计算成本,但缺乏对窗口外长期信息的有效压缩与检索,导致遗忘和时空不一致问题。为在固定内存预算下提升历史信息保留能力,我们在扩散 Transformer(DiT)中引入时间循环神经网络(RNN)层。实验发现,每个 DiT 层后接的 LSTM 层即可达到与 Test-Time Training(TTT)和 Mamba2 等先进 RNN 块相当的性能。现有扩散-RNN 方法常因训练-推理差异或窗口间重叠不足而性能下降。为此,我们提出新型 Recurrent Autoregressive Diffusion(RAD)框架:利用循环模块进行记忆更新与检索,并通过重叠滑动窗口的全注意力保持局部细节,且无训练-推理差距。在 Memory Maze 与 Minecraft 数据集上的实验表明,该框架在长视频生成中具有明显优势。

原文摘要 · Abstract (English)

Recent advancements in video generation has shifted from bidirectional models for short videos to autoregressive ones for ultra long video generation. Previous models, which usually use sliding window attention to restrict inference cost, lack effective memory compression and retrieval for long-term generation beyond the window size, leading to issues of forgetting and spatiotemporal inconsistencies. To enhance the retention of historical information with a fixed memory budget, we additionally incorporate temporal recurrent neural network (RNN) layers into the diffusion transformer (DiT) model. Specifically, we found that a LSTM layer after attention at each DiT layer achieves comparable performance to other state-of-the-art RNN blocks, such as Test-Time Training (TTT) and Mamba2. Moreover, existing diffusion-RNN approaches often suffer from performance degradation due to training-inference gap or the lack of overlap across windows. To address these limitations, we propose a novel Recurrent Autoregressive Diffusion (RAD) framework, which leverages recurrent blocks for memory update and retrieval and preserves local details by full attention on overlapping sliding windows, with no training and inference gap. Experiments on Memory Maze and Minecraft datasets demonstrate the superiority for long video generation by our framework with global memory and local attention.

视频生成扩散模型长序列RNN

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