给视频生成模型加个可插拔的记忆库,让它更懂物理常识。
Learning Plug-and-play Memory for Guiding Video Diffusion Models
- 用3D CNN和滤波器分离视觉与物理语义信息,构建可注入记忆
- 仅用150M参数和1万样本训练,就能显著提升视频真实感
- 训练后无需修改主模型,可直接用于各类扩散模型
基于DiT的视频生成模型虽在画质和时序连贯性上表现优异,但仍常违背基本物理规律和常识动态,暴露其缺乏显式世界知识。本文探索如何为它们引入一个可插拔的记忆模块以注入有用的世界知识。受基于Transformer的LLM中上下文记忆的启发,我们通过实证研究发现,对DiT的隐藏状态进行干预可有效引导模型行为;在嵌入空间中使用简单的低通与高通滤波器能自然分离低层外观特征与高层物理/语义线索,实现定向引导。基于此,我们提出可学习的记忆编码器DiT-Mem,由堆叠的3D CNN、低/高通滤波器和自注意力层构成。该编码器将参考视频映射为一组紧凑的记忆令牌,并在DiT自注意力层中拼接作为记忆。训练时冻结扩散主干,仅优化记忆编码器。该方法仅需150M参数和10,000个数据样本,训练高效,且推理时可即插即用。大量实验表明,该方法显著提升了模型对物理规则的遵循能力与视频保真度。代码与数据已公开:https://thrcle421.github.io/DiT-Mem-Web/
原文摘要 · Abstract (English)
Diffusion Transformer(DiT) based video generation models have recently achieved impressive visual quality and temporal coherence, but they still frequently violate basic physical laws and commonsense dynamics, revealing a lack of explicit world knowledge. In this work, we explore how to equip them with a plug-and-play memory that injects useful world knowledge. Motivated by in-context memory in Transformer-based LLMs, we conduct empirical studies to show that DiT can be steered via interventions on its hidden states, and simple low-pass and high-pass filters in the embedding space naturally disentangle low-level appearance and high-level physical/semantic cues, enabling targeted guidance. Building on these observations, we propose a learnable memory encoder DiT-Mem, composed of stacked 3D CNNs, low-/high-pass filters, and self-attention layers. The encoder maps reference videos into a compact set of memory tokens, which are concatenated as the memory within the DiT self-attention layers. During training, we keep the diffusion backbone frozen, and only optimize the memory encoder. It yields a rather efficient training process on few training parameters (150M) and 10K data samples, and enables plug-and-play usage at inference time. Extensive experiments on state-of-the-art models demonstrate the effectiveness of our method in improving physical rule following and video fidelity. Our code and data are publicly released here: https://thrcle421.github.io/DiT-Mem-Web/.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。