Recurrent Autoregressive Diffusion: Global Memory Meets Local Attention
TLDR
Proposes RAD, a recurrent autoregressive diffusion framework with LSTM for long video generation, improving memory and consistency.
Reasoning
The paper introduces a novel integration of RNN (LSTM) into diffusion transformers for long video generation, addressing forgetting and spatiotemporal inconsistencies. Strengths include a consistent training-inference procedure and improved memory retention. Weaknesses are the limited evaluation on game datasets (Memory Maze, Minecraft) and lack of comparison to broader world model benchmarks.
Read-first score
Read-first score 38, weighted from topical fit, citation, graph, method, reproducibility, and recency signals. Original total remains 37.
Field roles
Rank sensitivity
Stability: volatile; rank range: 199.
Keyword Scores
Deep Analysis
Innovations
- Introduces an LSTM-based recurrent neural network into the diffusion transformer framework for long-term memory retention in video generation.
- Proposes the Recurrent Autoregressive Diffusion (RAD) framework with frame-wise autoregressive memory update and retrieval, eliminating training-inference gap and lack of overlap across windows.
- Demonstrates that LSTM with attention matches state-of-the-art RNN blocks like TTT and Mamba2 in diffusion models.
Methodology
The method integrates an LSTM into a diffusion transformer to form the RAD framework, which performs frame-wise autoregressive memory updates and retrievals consistently during training and inference. Experiments are conducted on long video generation using the Memory Maze and Minecraft datasets.
Key Results
RAD achieves superior long video generation performance on Memory Maze and Minecraft, outperforming existing diffusion-RNN approaches and confirming the efficiency of LSTM for sequence modeling.