CVPR 2024 · 深度论文精读

让扩散补绘跨越视频长度

AVID 将图像补绘模型改造成 16 帧视频模型,再以 Temporal MultiDiffusion 在每个去噪步骤融合重叠窗口,并用全视频中间帧约束跨窗口身份。它同时揭示一个关键边界:接缝平滑、CLIP 时序相似和真实运动一致性属于不同问题。

Zhixing Zhang 等CVPR 2024 · 7162–7172arXiv:2312.03816 v3Rutgers University · Meta GenAI训练:16 帧 · 512×512 · 6 FPS
Reading map

阅读路线

01 · Research motivation

三项困难需要三种控制机制

输入包含视频、首帧掩码和文本提示。系统需要把掩码传播到全部帧,在区域内部生成符合文本的新内容,同时精确保留区域外的原始像素语义,并支持补画、替换、重纹理等不同结构约束。

时间一致性

逐帧图像补绘会造成形状、颜色和纹理闪烁。AVID 在冻结图像生成能力的前提下训练 temporal self-attention motion modules。

结构强度

重纹理需要跟随原对象轮廓;对象替换需要放松轮廓;扩画不需要目标区域结构。单一结构条件强度难以覆盖三类需求。

可变长度

固定长度训练的时序注意力在更长序列上质量下降。整段处理的时空注意力成本也随帧数平方增长。

Paper verified论文明确验证三类核心任务:uncropping、object swap、re-texturing;附录另展示 content removal、environment swap 和顺序处理多个区域。
Author claim作者把该工作描述为 text-guided video inpainting 的早期统一框架。该定位成立于“显式首帧掩码 + 文本 + 多类型补绘 + 任意长度推理”的组合,不能理解为传统视频内容移除的首次研究。
02 · Problem formulation

变量、条件与优化目标

输入

原视频 \(v_0=\{x_i^0\}_{i=1}^{N'}\),首帧二值掩码 \(m^0\),编辑文本 \(y\)。XMem 将 \(m^0\) 传播为 \(m=\{m_i\}_{i=1}^{N'}\);扩画任务直接在各帧使用同一待填区域。

输出

编辑视频 \(\hat v_0\)。掩码内部响应文本和可选 HED 结构条件,掩码外部保持源视频信息。论文用 BP 衡量区域外 L1 变化。

固定长度与任意长度

\(N=16\) 是正文训练片段长度;\(N'\) 是测试视频长度。窗口步幅为 \(o\),重叠窗口数量为 \(n\)。附录的 24 帧实验用于长度外推消融。

可学习部分

阶段一仅更新 motion modules;阶段二冻结主 UNet,更新由 UNet encoder 初始化的结构模块 \(s_\theta\)。推理期间不微调视频。

扩散过程

\[ q(x_t\mid x_{t-1})=\mathcal{N}\!\left(\sqrt{1-\beta_t}\,x_{t-1},\beta_t\mathbf{I}\right) \]

时间步 \(t\in\{1,\ldots,T\}\),\(\beta_t\) 控制逐步加噪。令 \(\alpha_t=1-\beta_t\),\(\bar\alpha_t=\prod_{j=1}^{t}\alpha_j\),论文以省略随机项的简化更新式说明从噪声预测恢复上一时间步;去噪器预测噪声后写成:

\[ x_{t-1}=\frac{1}{\sqrt{\alpha_t}}\left(x_t-\frac{1-\alpha_t}{\sqrt{1-\bar\alpha_t}}\,\epsilon_\theta(x_t,t,c)\right) \]

图像补绘条件为 \(c=(x_m,m,\tau_\theta(y))\),其中 \(x_m=x_0\odot(1-m)\),\(\tau_\theta\) 是文本编码器。基础噪声预测损失为:

\[ \mathcal{L}_{\mathrm{img}}=\mathbb{E}_{\epsilon\sim\mathcal{N}(0,\mathbf{I})}\left[\left\lVert\epsilon-\epsilon_\theta(x_t,t,c)\right\rVert_2^2\right] \]
公式在系统中的位置:前向过程提供带噪训练样本;UNet 学会条件噪声预测;DDIM、PNDM 与 classifier-free guidance 只是论文列举的可用采样技术。论文没有报告最终实现采用的采样器、采样步数和 CFG 数值。
03 · Model and training

训练两组适配器,保留图像补绘先验

01 · Mask首帧掩码由 Grounding-DINO + SAM 获得,XMem 传播至视频。
02 · Inflate二维图像 LDM 逐帧执行,层间加入 pixel-wise temporal self-attention。
03 · Motion16 帧随机合成掩码视频训练,只更新 motion modules。
04 · Structure复制 UNet encoder 参数,输入区域内 HED,训练 ControlNet 风格模块。
05 · Windows长视频切为重叠 16 帧窗口,每个去噪步骤分别计算。
06 · Fuse同一帧的多窗口预测求均值,中间帧注意力抑制身份漂移。
AVID 官方方法总览:motion module 训练、structure guidance 训练和重叠窗口推理
Figure 2(官方项目页方法图):左侧训练 motion modules;中间冻结主 UNet 后训练结构模块;右侧在同一去噪步骤聚合重叠时间窗口。该图由作者项目页提供。

阶段一:视频噪声预测

每帧独立执行前向加噪,网络通过 temporal self-attention 交换时间信息。保留区域视频为 \(v_m=\{x_i^0\odot(1-m_i)\}_{i=1}^{N}\),条件为 \(c=(v_m,m,\tau_\theta(y))\)。

\[ \mathcal{L}_{\mathrm{motion}}=\mathbb{E}_{\epsilon\sim\mathcal{N}(0,\mathbf{I})}\left[\left\lVert\epsilon-\epsilon_\theta(v_t,t,c)\right\rVert_2^2\right] \]

只训练 motion modules 的目的,是避免小规模视频数据破坏预训练图像 LDM 的文本与视觉生成能力。该策略把空间生成先验和时间适配分开。

阶段二:可调结构引导

逐帧结构 \(s^i=S(x_i^0)\) 由 HED 提取,训练时只保留合成掩码区域内的 HED。结构模块输出 13 张、跨 4 个分辨率的特征 \(\{h_j\}_{j=1}^{13}\),注入 UNet 各 skip connection 和 middle block。

\[ c_s=s_\theta(v_t,t,c,s),\qquad \mathcal{L}_{\mathrm{structure}}=\mathbb{E}\left[\left\lVert\epsilon-\epsilon_\theta(v_t,t,c,c_s)\right\rVert_2^2\right] \]

推理时使用 \(\epsilon_\theta(v_t,t,c,\omega_s c_s)\)。\(\omega_s\) 越大,结果越贴近源视频结构。正文设置:对象替换和扩画取 \(\omega_s=0\),重纹理取 \(\omega_s=1\)。

04 · Any-length inference

每个去噪步骤都在重叠窗口之间求共识

Temporal MultiDiffusion 沿时间轴应用 MultiDiffusion。它在每一步去噪时让重叠窗口共同预测同一帧,再对这些预测求平均,从而避免已完成的独立短片产生突变接缝。

对 \(N'\) 帧视频,以长度 \(N\) 和步幅 \(o\) 建立窗口 \(v_t^i\)。论文给出的窗口数如下。当 \(N'-N\) 不能被 \(o\) 整除时,该式本身不会覆盖尾部;论文没有公开 tail window、padding 或末窗对齐策略:

\[ n=\left\lfloor\frac{N'-N}{o}\right\rfloor+1 \]

令 \(S_k\) 为包含全局第 \(k\) 帧的窗口索引集合;窗口 \(i\) 中与其对应的局部帧记为 \(v_{t-1}^i[j]\)。融合规则为:

\[ v_{t-1}[k]=\frac{1}{\lvert S_k\rvert}\sum_{i\in S_k}v_{t-1}^i[j] \]

同一帧接受多个相邻上下文的预测,平均操作在整个反向扩散轨迹中反复执行。它直接缓解窗口接缝,但长距离身份仍可能缓慢漂移。

中间帧注意力引导

每个 spatial self-attention 层缓存全视频中间帧的 \(K^{N'/2}\) 与 \(V^{N'/2}\)。第 \(i\) 帧保留自己的常规注意力,同时加入对该参考帧的辅助注意力:

\[ \operatorname{Attention}(i)= (1-\omega_a)\operatorname{softmax}\!\left(\frac{Q^i(K^i)^{\mathsf T}}{\sqrt d}\right)V^i +\omega_a\operatorname{softmax}\!\left(\frac{Q^i(K^{N'/2})^{\mathsf T}}{\sqrt d}\right)V^{N'/2} \]

作者选择中间帧,因为它到其他帧的总体时间距离更小。正文长视频设置 \(\omega_a=0.3\)、\(o=4\)。过强引导会让参考帧键值支配生成,\(\omega_a=1\) 的汽车案例出现车轮伪影。

计算复杂度

整段直接推理

附录将基础模型复杂度写为 \(O((HW)^2N'^2)\):空间 attention 为 \(O((HW)^2)\),temporal attention 随 \(N'^2\) 增长。

滑动窗口

窗口阶段约为 \(O((HW)^2N^2n)\);加入中间帧 attention 后作者写为 \(O(2(HW)^2N^2n)\)。当 \(N'\gg N\) 且步幅固定时,对总长度近似线性。

Our interpretation“任意长度”指推理算法可以覆盖任意 \(N'\),不等价于对任意长程运动都维持真实动力学。共同中间帧提升身份锚定,窗口平均提升局部连接;两者没有显式世界状态或物体轨迹记忆。
05 · Reproducibility

论文公开了系统骨架,训练配方仍不完整

项目论文披露复现含义
基础模型大规模预训练 inpainting Latent Diffusion Model,引用 Rombach et al.未给出精确 checkpoint 或版本
训练数据去水印 Shutterstock 视频私有来源;未报告视频数量、筛选与划分
片段16 帧,512×512,6 FPS正文基础模型的训练分布
掩码synthetic random masks未公开采样分布、形状与面积范围
结构条件掩码区域内 HEDUNet encoder 权重复制到控制模块,控制模块全部参数训练
评估数据125 个训练未见视频Grounding-DINO + SAM 定位首帧对象,XMem 传播掩码,Llama 生成多种提示
长视频推理\(\omega_a=0.3\),\(o=4\)正文默认;结构比例按任务设为 0 或 1
未公开optimizer、learning rate、batch size、epoch/step、scheduler、warmup、weight decay、precision、GPU、训练时长无法据论文精确重训
同样未公开采样器最终选择、采样步数、CFG、negative prompt、随机种子、窗口尾部策略无法精确复现推理质量与速度
24 帧设置需分开理解:正文训练使用 16 帧。附录 E.1 说明 temporal self-attention 中 sinusoidal position encoding 的最大长度设为 24,并用 24 帧直接推理对比 Temporal MultiDiffusion。该消融展示分布外长度下的质量变化,不代表主模型改用 24 帧训练。
Paper verified测试时首帧对象区域由 Grounding-DINO 与 SAM 标注,之后由 XMem 传播。Llama 用于给测试对象生成多个任务提示。论文没有公开这些提示的完整模板、模型版本或人工筛选流程。
06 · Experiments

主表验证 16 帧基础模型,不覆盖任意长度量化

所有方法在 16 帧视频上比较,作者有意关闭 Sec. 3.4 的长视频推理以控制条件。BP 是未编辑区域 L1 距离,表格以 \(10^{-3}\) 为单位,越低越好;例如 43.1 对应 0.0431;TA 是 CLIP text-video alignment,越高越好;TC 是相邻帧 CLIP-Image 特征余弦相似度,越高越好。

方法UncroppingObject swapRe-texturing
BP ↓TA ↑TC ↑BP ↓TA ↑TC ↑BP ↓TA ↑TC ↑
Per-frame LDM43.131.393.641.431.192.541.431.292.4
Text2Video-Zero49.031.496.547.330.194.947.930.695.0
VideoComposer55.731.296.471.031.596.564.532.195.5
AVID42.331.397.241.131.596.540.732.096.3

AVID 在所有任务取得最低 BP,并在 uncropping 与 re-texturing 获得最高 TC;object swap 的 TA/TC 与 VideoComposer 持平。VideoComposer 的重纹理 TA 32.1 略高于 AVID 32.0。比较条件仍有差异:VideoComposer 重纹理同时接收 sketch、depth 与 motion vector,AVID 只接收 HED。

69.1%
Uncropping 用户首选 AVID
74.5%
Object swap 用户首选 AVID
73.2%
Re-texturing 用户首选 AVID

用户对比同时考虑时间一致、文本匹配和背景保持。Figure 6 给出四种方法的偏好比例,但正文没有披露参与者数量、每人样本数、随机化、置信区间或显著性检验,因此只能把百分比视为偏好证据,不能推导总体效应大小。

TokenFlow 附录比较

任务方法BP ↓TA ↑TC ↑
Object swapTokenFlow93.331.597.5
Object swapAVID41.131.596.5
Re-texturingTokenFlow90.832.297.8
Re-texturingAVID40.732.096.3

TokenFlow 在 TC 与 TA 略高,却严重改变掩码外背景。作者将其归因于 TokenFlow 依靠文本决定编辑区域且使用 DDIM inversion;AVID 的显式 mask 对背景保护更强。TC 的高分也可能来自内容变化较小,无法单独证明运动真实。

Paper verified正文和附录对长视频只提供定性对比:32 帧非重叠独立片段、24 帧直接推理、32 帧 attention scale 等。论文没有报告任意长度任务的独立数值表、速度、显存或随长度增长的质量曲线。
07 · Ablation

消融定位了三种机制的作用边界

结构比例 \(\omega_s\)

重纹理从 0 增至 1 时,金色熊猫更好保持原运动与轮廓;对象替换中高比例会让浣熊继承熊猫形状并削弱文本匹配。该结果支持按任务调节结构强度。

重叠窗口

32 帧案例中,两个独立 16 帧片段在第 15/16 帧出现鹅颈纹理和浣熊姿态突变;Temporal MultiDiffusion 使边界过渡连续。该实验将 \(\omega_a\) 设为 0,以隔离窗口融合。

中间帧 \(\omega_a\)

\(\omega_a=0\) 时汽车颜色由红向暗红漂移;适度引导维持颜色与图案;\(\omega_a=1\) 引入后轮伪影。正文默认 0.3 来平衡锚定与帧内适应。

参考帧选择

附录比较 SC Attention、middle-frame SC Attention、first-frame guidance 与 middle-frame guidance。中间帧方案在示例中同时稳定颜色和图案;这些结果为定性样例,没有量化汇总。

24 帧直接推理

位置编码上限为 24 使模型可以接收 24 帧,但直接处理的浣熊耳朵错位,另一案例无法自然融合;窗口推理改善质量。它证明“可以执行”和“保持训练分布质量”存在差别。

掩码误差

扩大掩码仍可补绘,归因于随机合成掩码训练;腐蚀掩码没有覆盖全部旧对象时,区域外保护机制会保留残片,阻碍目标形状改变。

因果证据强度:Temporal MultiDiffusion 和 attention guidance 的分离实验方向清晰;结构比例也显示任务差异。主要证据来自少量精选视频,缺少多随机种子、总体均值与失败率,因此支持组件可用,尚不足以精确估计其普遍收益。
08 · Paper vs. implementation

官方仓库没有发布可运行模型代码

仓库实际内容

zhang-zx/AVID
assets/ · LICENSE · README.md
3 commits · MIT license
无训练代码、推理代码、config、checkpoint 或 release

README 声明仓库包含论文代码,但当前可见文件只有素材与说明。页面致谢 diffusers、AnimateDiff 与 Tune-A-Video,无法由此恢复 AVID 的具体实现。

可依据论文重建

两阶段冻结策略、16 帧输入、HED 控制、13 个多尺度特征、窗口均值融合、中间帧注意力公式、默认 \(o=4\)、\(\omega_a=0.3\) 与任务相关 \(\omega_s\)。

关键缺口

模型 checkpoint、optimizer 与学习率、噪声 schedule、采样器与步数、CFG、训练时长、显存、随机掩码生成器、测试 prompt pipeline 和视频预处理细节。

依赖风险

Shutterstock 训练集不可下载核查;Grounding-DINO、SAM、XMem、Llama 与 HED 构成外部工具链。每个版本变化都可能改变 mask、提示和最终分数。

Our interpretation当前公开材料足以复现方法族,无法复现论文数值。任何第三方实现选择的 Stable Diffusion checkpoint、采样步数与 mask 策略都应明确标为实现假设。
09 · Reviewer commentary

系统组合具有实用价值,长程证据和复现透明度偏弱

Strengths

  • 问题拆分与组件职责清楚:motion、structure、window consensus、identity anchor 分别处理不同失败模式。
  • 冻结主图像 LDM,训练量集中在时序和结构适配器,合理保护预训练生成能力。
  • 显式 mask 让背景保护可测且明显优于依靠文本编辑区域的方法。
  • 附录诚实展示复杂动作、掩码遗漏和离开后返回画面的失败案例。

Weaknesses

  • 长视频贡献缺乏量化表与长度扩展曲线,主表只覆盖 16 帧。
  • TC 是相邻帧 CLIP 相似度,可能奖励静态或欠运动结果,缺少 optical flow、轨迹、重投影或动作真实性指标。
  • 用户研究缺少人数、样本分配、置信区间与显著性统计。
  • 训练集、代码、权重和大量超参数未公开,精确复现受阻。

Claim audit

“背景保持”由 BP 强力支持;“基础视频一致性”由 TC、用户偏好和定性样例支持;“任意长度”由算法可执行性和最长 256 帧网页样例支持;跨长程运动真实性和任意长度质量稳定性没有得到充分验证。

Limitations

马头左右转动会消失后再出现,狮腿发生穿插;对象离开画面后返回时容易失去一致性。作者将前两项归因于 motion foundation model 能力,将后者指向缺少更精细的 cross-clip attention injection。

Research extension将单一中间帧锚点扩展为少量可更新的 object-centric memory tokens,记录外观、姿态、遮挡与进入离开事件;窗口以可见性门控读取记忆,可专门处理对象离开后返回。
Research extension在 test time 联合搜索 \(\omega_s\)、\(\omega_a\)、步幅与参考帧集合,目标包含 BP、文本匹配、光流重投影误差、轨迹连续性和生成区域的 perceptual quality;需要在独立验证集选择超参数,避免对单个视频调优。
Research extension让结构比例由文本任务、mask 几何和局部运动自动预测,并对腐蚀 mask 引入“旧对象残留”检测与可学习 mask expansion。该方向直接对应补绘区域不完整的附录失败案例。
10 · Sources

一手资料与版本说明

  1. Zhang et al., arXiv:2312.03816v3,16 页合订本:正文 11 页,随后附录 5 页;方法、公式、主表与失败案例的主要来源。
  2. CVPR 2024 Open Access,正式会议信息与 pp. 7162–7172。
  3. CVPR Supplementary Material,6 页独立附录版本;额外任务、效率、TokenFlow、24 帧与 attention 消融、mask 鲁棒性及局限。
  4. 官方项目页,方法总览、定性视频、80/256 帧 text-to-video 延伸结果。
  5. 官方补充网页,对比、消融和失败视频的动态展示。
  6. 官方 GitHub,用于核查实际公开内容与可复现边界。
书目信息:论文作者顺序为 Zhixing Zhang、Bichen Wu、Xiaoyan Wang、Yaqiao Luo、Luxin Zhang、Yinan Zhao、Peter Vajda、Dimitris Metaxas、Licheng Yu。论文和项目页没有通讯作者标记;Obsidian 按仓库规则把末位作者 Licheng Yu 保留为未核实候选。项目页只标注 Zhixing Zhang 的工作主要在 Meta AI 实习期间完成,该星号仅表示实习期间完成工作,不表示通讯身份。