← 返回 2026-03-20

EffectErase:联合视频物体移除与插入实现高质量效果擦除 EffectErase: Joint Video Object Removal and Insertion for High-Quality Effect Erasing

Yang Fu, Yike Zheng, Ziyun Dai, Henghui Ding 📅 2026-03-19 👍 18 2026-07-13 08:36
扩散模型 数据集构建 物体移除 视频修复 计算机视觉

联合学习视频物体移除与插入,利用逆任务一致性擦除物体副作用效果

前置知识

视频修复(Video Inpainting)

视频修复是指根据给定的掩码序列,对视频中缺失或需要移除的区域进行内容重建的技术。早期方法使用卷积网络进行时空建模,但难以处理长距离传播问题。后续工作如 ProPainter 利用光流进行额外的运动线索补充,提升时间一致性。近年来,基于扩散模型的方法(如 COCOCO、FloED、VACE)进一步提升了视频修复的质量和可控性,但这些通用修复方法通常忽略了物体引起的视觉副作用效果。

EffectErase 是视频修复的一个特化方向,理解通用视频修复方法的局限性是理解本文动机的关键基础。

物体副作用效果(Object-Induced Effects)

物体副作用效果是指目标物体在场景中引起的视觉影响,这些影响远超出物体本身的遮挡区域。本文将其归类为五种代表性类型:遮挡(Occlusion,包括不透明、半透明、透明物体)、阴影(Shadow,物体遮挡光线产生的阴影区域)、光照(Lighting,移除光源后场景亮度和色彩平衡的变化)、反射(Reflection,物体在镜子、水面或瓷砖上的反射)以及变形(Deformation,物体对周围结构如窗帘、草地、网的物理变形)。

这些副作用效果是本文要解决的核心问题。现有方法虽然能移除物体本身,但往往无法有效擦除这些副作用,这正是 EffectErase 的突破点。

扩散变换器(Diffusion Transformer, DiT)

DiT 是一种将 Transformer 架构应用于扩散模型的生成架构,用 Transformer 块替代了传统扩散模型中的 U-Net 结构。在本文中,EffectErase 基于 Wan 2.1 视频生成模型(一个基于 DiT 的大规模视频生成模型)进行微调。DiT 块中的交叉注意力机制被用于建模物体与其副作用之间的时空关联,每个 DiT 块产生注意力图,通过聚合和最大池化得到最大激活图。

DiT 是 EffectErase 的骨干网络架构,理解其工作原理对于理解 TARG 模块如何利用交叉注意力来定位受影响区域至关重要。

LoRA(Low-Rank Adaptation)

LoRA 是一种参数高效的微调方法,通过在预训练模型的权重矩阵中注入低秩分解的可训练参数,以极小的参数量实现模型适配。在本文中,EffectErase 对 Wan 2.1 模型的注意力投影(q, k, v, o)和前馈层(ffn.0, ffn.2)应用 LoRA,秩设置为 256,所有 LoRA 权重使用 Kaiming 初始化。这使得模型能够在不修改原始骨干网络参数的情况下,高效地学习视频物体移除和插入的任务。

理解 LoRA 机制有助于理解本文的训练策略和计算效率,以及如何在有限算力条件下适配大规模预训练模型。

交叉注意力(Cross-Attention)

交叉注意力是一种注意力机制,其中查询(Query)来自一个序列,而键(Key)和值(Value)来自另一个序列。在 EffectErase 中,融合特征 $\dot{x}_t$ 作为查询,而任务感知区域引导模块生成的提示嵌入 $e_{\text{prompt}}$ 作为键和值。通过交叉注意力,模型能够捕捉物体与其副作用之间的时空关联,产生突出受影响区域的注意力图。这些注意力图被聚合后用于效果一致性损失的计算。

交叉注意力是 TARG 模块的核心机制,也是效果一致性损失计算的基础,理解它是理解整个方法框架的关键。

KL 散度(Kullback-Leibler Divergence)

KL 散度是衡量两个概率分布之间差异的非对称度量,定义为 $D_{\text{KL}}(P \| Q) = \sum_i P(i) \log \frac{P(i)}{Q(i)}$。在 EffectErase 中,效果一致性损失 $\mathcal{L}_{\text{EC}}$ 使用 KL 散度来对齐移除分支和插入分支的软效应区域估计,使得两个逆任务关注相同的效果区域。由于 KL 散度是非对称的,论文中对两个方向都进行了计算:$\mathcal{L}_{\text{EC}} = D_{\text{KL}}(f^{\text{diff}} \| f^{\text{rm}}) + D_{\text{KL}}(f^{\text{diff}} \| f^{\text{in}})$。

效果一致性损失是本文三大核心组件之一,理解 KL 散度的作用对于理解移除-插入联合学习如何实现跨任务一致性至关重要。

研究动机

视频物体移除旨在消除视频中的动态目标物体及其视觉效果(如变形、阴影、反射等),同时恢复无缝的背景。尽管基于扩散的视频修复和物体移除方法(如 ProPainter、DiffuEraser、VACE 等)已经展示了令人期待的性能,但它们在处理具有复杂视觉效果的物体时仍然面临严峻挑战。如论文 Figure 2 所示,现有方法如 ROSE 和 MinMax-Remover 虽然能移除输入掩码区域内的物体主体,但常常无法发现和移除物体引起的副作用(如反射)。这一局限性的根本原因在于大多数方法过度依赖输入掩码作为引导,导致忽略了物体引入到场景中的副作用。部分方法如 MinMax-Remover 隐式训练模型来发现这些效果,ROSE 显式预测副作用差异掩码,但它们都缺乏对物体与其副作用之间时空关联的显式建模,在复杂真实场景中的鲁棒性有限。此外,该领域还缺乏一个大规模、公开可用的数据集来系统性地捕捉各种场景中的常见物体副作用效果。

本文的目标是本文的目标是解决效果感知的视频物体移除问题,具体包括三个层面:首先,构建一个大规模、高质量的混合数据集 VOR(Video Object Removal),包含 60K 对配对视频,涵盖 366 个物体类别、67 种场景类型,总时长超过 145 小时,系统性地覆盖五种代表性物体副作用效果(遮挡、阴影、光照、反射、变形);其次,提出 EffectErase 方法,通过联合学习视频物体移除和插入来实现高质量的效果擦除,利用两个逆任务之间的互补关系来增强对受影响区域的感知和定位能力;最后,建立两个评估基准(VOR-Eval 和 VOR-Wild),为该领域的未来研究提供标准化的评估框架。

与已有工作不同的是,本文的独特切入角度在于将视频物体移除和插入视为一对互逆任务,并利用这种对称关系进行联合学习。与以往将移除视为孤立任务的方法不同,EffectErase 观察到移除和插入操作作用于相同的效果区域(如 Figure 3 所示),因此可以通过联合优化来提供互补的监督信号。具体而言,论文提出了三大创新:一是 Task-Aware Region Guidance(TARG)模块,通过交叉注意力机制建模物体与其副作用之间的时空关联,并支持灵活的任务切换;二是 Effect Consistency(EC)损失,强制移除和插入两个分支关注相同的效果区域,利用插入任务作为移除任务的辅助监督;三是混合数据集构建策略,结合相机捕获的真实数据和 3D 渲染引擎生成的合成数据,在保证物理真实性的同时大幅扩展数据多样性。这种以插入辅助移除的双学习范式是本文区别于所有已有方法的核心创新。

核心方法

EffectErase 的整体思路可以这样理解:想象你要从视频中擦除一个物体,不仅要让它消失,还要擦除它留下的所有痕迹——阴影、反射、光照变化等。现有方法只关注物体本身,就像用橡皮擦只擦掉字却留下了压痕。EffectErase 的直觉是:如果我能完美地把物体放回去(插入),那我就能知道哪些区域受到了物体的影响;反过来,如果我知道了受影响的区域,移除就能做得更干净。因此,论文将移除和插入这对逆任务放在一起训练,让它们相互教对方。技术路线上,EffectErase 基于 Wan 2.1 视频生成模型(一个 1.3B 参数的扩散变换器),使用 VAE 将配对视频编码到潜空间,通过一个轻量级适配器融合条件与噪声潜变量,然后在 DiT 骨干网络上进行去噪。在此基础上,TARG 模块注入任务感知的区域引导信息,EC 损失对齐两个任务的效果区域估计,三者共同实现高质量的效果擦除。

EffectErase 的核心创新在于将视频物体移除和插入作为互逆任务进行联合学习,这与已有方法的本质区别体现在三个维度。第一,任务建模方式不同:现有方法(如 ROSE、MinMax-Remover)将移除视为孤立任务,只学习怎么擦,而 EffectErase 同时学习怎么擦和怎么放,通过两个逆操作的互补性来增强对效果区域的感知。第二,效果区域定位方式不同:以往方法依赖二值掩码或隐式学习来定位副作用区域,丢失了变化强度信息;EffectErase 通过 TARG 模块的交叉注意力机制显式建模物体与副作用之间的时空关联,并使用软分布来保留效果的渐变信息(如光照和阴影的强度变化)。第三,监督信号不同:EC 损失利用 KL 散度对齐移除和插入分支的注意力图,使得插入任务为移除提供额外的效果区域监督,这是前所未有的跨任务一致性约束。

方法步骤详情

EffectErase 的方法包含以下完整步骤:(1)数据编码与融合:视频输入首先通过预训练 VAE 编码到潜空间。对于移除任务,物体视频 $V^o$、背景视频 $V^b$(无物体)和对应掩码 $M$ 分别编码为 $x_o$、$x_b$、$x_m$,其中干净潜变量 $x = x_b$;对于插入任务,$x = x_o$。随机噪声 $z \sim \mathcal{N}(0, I)$ 通过前向过程 $x_t = t x + (1-t) z$ 加入,其中 $t$ 从 logit-normal 分布采样。一个轻量级适配器 $A_\phi(\cdot)$ 将条件 $c$ 与噪声潜变量融合:$\dot{x}_t = A_\phi([x_t; c])$,其中移除时 $c = [x_o; x_m]$,插入时 $c = [x_b; x_f]$($x_f = x_o \odot x_m$ 为前景掩码乘积)。(2)Task-Aware Region Guidance:从语言模型提取任务嵌入 $e_{\text{task}}$,从 CLIP 图像编码器提取前景视觉嵌入 $e_f$(输入为裁剪的前景图像块 $V^f = V^o \odot M$),通过轻量投影器 $P_\psi(\cdot)$ 映射到令牌空间,替换 $e_{\text{task}}$ 中的占位符 object,形成 $e_{\text{prompt}} = e_{\text{task}}[\text{object}] \leftarrow P_\psi(e_f)$。该嵌入通过交叉注意力注入骨干网络。(3)去噪训练:去噪模型 $v_\theta$ 预测速度 $v = x - z$,损失为 $\mathcal{L}_{\text{denoise}} = \mathbb{E}_{z,x,t,c} \| v_\theta(x_t, t, c) - v \|^2$,分别对移除和插入计算。(4)Effect Consistency Loss:收集所有 DiT 块的交叉注意力图,通过最大池化得到 $A^{\text{rm}}$ 和 $A^{\text{in}}$,经轻量映射器 $G_\omega(\cdot)$ 投影为软效应区域估计 $f^{\text{rm}}$ 和 $f^{\text{in}}$。同时从 $V^o$ 和 $V^b$ 的下采样差异构建差异图先验 $f^{\text{diff}}$(保留变化强度信息而非二值掩码)。EC 损失为 $\mathcal{L}_{\text{EC}} = D_{\text{KL}}(f^{\text{diff}} \| f^{\text{rm}}) + D_{\text{KL}}(f^{\text{diff}} \| f^{\text{in}})$。(5)联合优化:总损失 $\mathcal{L}_{\text{total}} = \mathcal{L}_{\text{denoise}}^{\text{remove}} + \mathcal{L}_{\text{denoise}}^{\text{insert}} + \lambda \mathcal{L}_{\text{EC}}$。

技术新颖性

EffectErase 的技术新颖性体现在多个层面。首先,将视频物体移除和插入作为互逆任务进行联合学习是一个全新的范式。以往的视频物体移除工作(如 ROSE、MinMax-Remover)都将移除作为孤立任务处理,而 EffectErase 首次利用两个逆任务之间的对称性来提供互补监督,这种以插入辅助移除的思路在该领域是首创。其次,TARG 模块的设计具有创新性:通过将 CLIP 视觉嵌入与语言模型任务嵌入融合,形成任务感知的区域引导表示,并利用 DiT 的交叉注意力机制进行注入,这种多模态融合的区域引导方式不同于以往简单的掩码引导或隐式学习。特别是,TARG 中的任务令牌设计使得同一个模型可以通过修改输入灵活切换移除和插入任务,无需额外训练。第三,EC 损失的软分布监督方式也是新颖的:与 ROSE 使用二值掩码不同,EffectErase 使用从 $V^o$ 和 $V^b$ 差异构建的软分布作为先验,保留了光照、阴影等效果的渐变信息,这对于准确定位和擦除这些微妙效果至关重要。最后,VOR 数据集的混合构建策略(相机捕获加 3D 合成)以及 Ken Burns 效果的扩展应用(14 种相机运动模式)也为数据集构建方法论做出了贡献。

Removal-Insertion. Video object removal and insertion are inverse tasks that operate on the same affected regions.
Figure 3: Removal-Insertion. Video object removal and insertion are inverse tasks that operate on the same affected regions.
Dataset Construction Pipeline of VOR.
Figure 4: Dataset Construction Pipeline of VOR.
Representative side effects in VOR dataset.
Figure 5: Representative side effects in VOR dataset.
The framework of EffectErase.
Figure 6: The framework of EffectErase.

实验结果

EffectErase 在三个数据集和所有评估指标上都取得了最优性能,全面超越现有方法。在 ROSE-Benchmark 上,EffectErase 的 PSNR 达到 32.161(比 ROSE 的 31.122 高出 1.039),LPIPS 降至 0.039(比 ROSE 的 0.077 降低近 50%),FVD 降至 55.578(比 ROSE 的 72.177 降低 23%),SSIM 提升至 0.948(比 ROSE 的 0.917 高出 0.031)。在 VOR-Eval 上,EffectErase 的 PSNR 为 23.750(比 ROSE 的 22.966 高出 0.784),FVD 为 342.871(比 ROSE 的 383.084 降低 10.5%)。在无 ground truth 的 VOR-Wild 上,EffectErase 获得最高的 QScore(9.280)和用户评分(7.20),显著优于 ROSE 的 6.38 和 MinMax-Remover 的 5.90。消融实验(Table 3)表明,各组件均有贡献:加入 EC 损失后 FVD 从 368.664 降至 354.545;加入 TARG 后 SSIM 从 0.737 显著提升至 0.780;加入合成数据后 LPIPS 从 0.193 明显降至 0.170。此外,EffectErase 无需额外训练即可直接适配物体插入任务,生成逼真的物体副作用效果(如阴影、反射)。

Comparison of video object removal datasets.
Table 1: Comparison of video object removal datasets.
Quantitative results on ROSE and VOR.
Table 2: Quantitative results on ROSE and VOR.
Ablation study on VOR-Eval.
Table 3: Ablation study on VOR-Eval.
Qualitative results on VOR-Eval.
Figure 7: Qualitative results on VOR-Eval.
Qualitative results on VOR-Wild.
Figure 8: Qualitative results on VOR-Wild.
查看结构化数据
任务指标本文基线提升
ROSE-Benchmark 视频物体移除 PSNR 32.161 ROSE: 31.122, MinMax-Remover: 26.770, Propainter: 27.200 比最优基线 ROSE 提升 1.039 dB(+3.3%)
ROSE-Benchmark 视频物体移除 LPIPS 0.039 ROSE: 0.077, MinMax-Remover: 0.099, ObjectClear: 0.076 比最优基线 ObjectClear 的 0.076 降低 48.7%
ROSE-Benchmark 视频物体移除 FVD 55.578 ROSE: 72.177, MinMax-Remover: 137.840, Propainter: 171.020 比最优基线 ROSE 的 72.177 降低 23.0%
VOR-Eval 视频物体移除 PSNR 23.750 ROSE: 22.966, MinMax-Remover: 21.963, Propainter: 21.975 比最优基线 ROSE 提升 0.784 dB(+3.4%)
VOR-Eval 视频物体移除 FVD 342.871 ROSE: 383.084, MinMax-Remover: 539.427, Propainter: 589.012 比最优基线 ROSE 的 383.084 降低 10.5%
VOR-Wild 无 ground truth 评估 QScore 9.280 ROSE: 9.240, MinMax-Remover: 8.984, DiffuEraser: 9.113 比最优基线 ROSE 提升 0.040(+0.4%)
VOR-Wild 无 ground truth 评估 User Score 7.20 ROSE: 6.38, MinMax-Remover: 5.90, DiffuEraser: 5.50 比最优基线 ROSE 提升 0.82(+12.9%)

局限与改进

论文在多个方面存在局限性。首先,EffectErase 需要输入掩码来指定移除区域,这在实际使用中可能不够用户友好。作者在论文末尾明确指出,未来方向是支持更自然的交互方式,如文本和语音指令。其次,从消融实验可以看到,即使加入所有组件,在 VOR-Eval 上的 PSNR 仅为 23.750,与 ROSE-Benchmark 上的 32.161 差距较大,说明在更复杂的真实场景中(VOR-Eval 包含 43 对配对视频,涵盖多种效果类型),方法的性能仍有提升空间。第三,论文的失败案例分析(Supplementary Figure VII)揭示了一个根本性问题:当物体的副作用或配件无法明确归属于目标物体时(如狗的阴影与人的阴影纠缠、狗链无法清晰归属),方法会出现困难。这种歧义性是效果感知移除的本质难题。此外,方法基于 Wan 2.1 的 1.3B 参数模型,训练需要 8 张 H100 GPU 进行 120K 次迭代,推理需要 50 步去噪,计算成本较高,可能限制其在资源受限场景下的应用。最后,论文仅评估了五种效果类型,实际场景中可能存在更复杂的效果组合(如多物体交互产生的复合效果),方法的泛化能力有待进一步验证。

独立分析的弱点

EffectErase 存在几个值得深入分析的弱点。第一,掩码依赖性问题:方法要求用户提供精确的物体掩码视频作为输入,这在实际视频编辑场景中是一个显著的使用门槛。虽然 SAM2 等分割工具可以辅助生成掩码,但对于长视频和复杂场景,掩码的生成和传播仍然是一个挑战。改进方向可以探索无掩码或稀疏提示(如点击、框选)的物体移除方案。第二,计算效率问题:模型基于 1.3B 参数的扩散变换器,训练需要 8 张 H100 GPU 和 120K 次迭代,推理需要 50 步去噪。对于实际部署场景,可以探索知识蒸馏、模型剪枝或更高效的采样策略来降低计算成本。第三,效果类型覆盖有限:虽然论文覆盖了五种代表性效果类型,但实际场景中还存在其他复杂效果(如运动模糊、景深变化、颜色溢出等),且论文未评估多种效果同时存在时的处理能力。第四,数据集偏差:VOR 的真实数据使用固定三脚架相机拍摄后施加 Ken Burns 效果模拟相机运动,这与真实手持拍摄的运动模式可能存在差异,可能影响模型在真实场景中的泛化性能。第五,评估指标局限:在 VOR-Wild 上使用 QScore(基于 Qwen-VL)和用户研究进行评估,这些主观评估方法的可重复性和客观性有待商榷。

未来方向

基于 EffectErase 的成果,可以从多个方向延伸未来研究。作者提出的方向是支持更用户友好的交互方式,如文本描述指定要移除的物体、语音指令控制移除操作,这需要将 TARG 模块扩展为支持多模态输入。更广泛的方向包括:(1)无监督或弱监督的效果感知移除,利用视频中的时间一致性线索自动发现和定位副作用区域,减少对配对数据的依赖;(2)实时视频物体移除,通过模型压缩和高效采样技术实现实时推理,满足视频编辑和直播场景的需求;(3)多物体交互效果处理,当场景中多个物体相互影响时(如一个物体的阴影投射到另一个物体上),如何准确归属和处理这些复合效果;(4)将 EffectErase 的联合学习范式扩展到其他视频编辑任务,如视频风格迁移、视频超分辨率等,探索互逆任务联合学习的通用性;(5)结合 3D 场景理解,在知道场景几何和光照信息的条件下,更精确地建模物体与环境的交互效果,实现物理上更准确的效果擦除。

复现评估

论文在复现方面提供了较好的条件。数据集方面,VOR 包含 60K 对配对视频,总时长 145.33 小时,数据量充足。论文描述了详细的数据构建流程:真实数据使用固定三脚架相机拍摄,合成数据使用 Blender 3D 引擎渲染,并使用 SAM2 进行掩码生成和人工精修。两个评估基准 VOR-Eval(43 对视频)和 VOR-Wild(195 个视频)也已建立。方法方面,论文基于开源的 Wan 2.1 模型,使用 LoRA 微调,超参数明确(输入分辨率 832x480,81 帧,学习率 $1 \times 10^{-5}$,LoRA 秩 256,批大小 8,120K 迭代,50 步去噪)。然而,复现仍面临一些挑战:训练需要 8 张 H100 GPU,这是大多数研究团队难以获得的算力资源;VOR 数据集的完整版本是否开源尚未明确;3D 合成数据的构建需要 Blender 渲染管线和 3D 资产,有一定门槛。总体而言,在算力充足的条件下,论文的复现可行性较高。