arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2501.02269cs.CV

TDM:面向一体化真实世界视频修复的时间一致性扩散模型

TDM: Temporally-Consistent Diffusion Model for All-in-One Real-World Video Restoration

Institute of Science Tokyo(东京科学大学)

Yizhou Li, Zihua Liu, Yusuke Monno, Masatoshi Okutomi

AI总结:

提出基于Stable Diffusion与ControlNet的一体化视频修复模型TDM,通过TPG、DDIM反演和SW-CFA提升多退化修复、内容保持与时间一致性。

摘要:

本文提出首个基于扩散模型的一体化视频修复方法,该方法利用预训练 Stable Diffusion 和微调后的 ControlNet 的能力。我们的方法能够通过单一统一模型修复多种类型的视频退化,克服了标准方法需要为每个修复任务配备特定模型的局限。我们的贡献包括:一种高效训练策略,其采用任务提示引导(Task Prompt Guidance,TPG)以处理多样的修复任务;一种推理策略,将去噪扩散隐式模型(Denoising Diffusion Implicit Models,DDIM)反演与新型滑动窗口跨帧注意力(Sliding Window Cross-Frame Attention,SW-CFA)机制相结合,以增强内容保持和时间一致性;以及一种可扩展流程,使我们的方法成为一体化方案,可适应不同的视频修复任务。通过在五项视频修复任务上开展大量实验,我们证明,相较于现有最先进方法,本方法在对真实世界视频的泛化能力以及时间一致性保持方面具有优越性。本方法通过提供可在多种应用中提升视频质量的统一解决方案,推动了视频修复任务的发展。

英文摘要:

In this paper, we propose the first diffusion-based all-in-one video restoration method that utilizes the power of a pre-trained Stable Diffusion and a fine-tuned ControlNet. Our method can restore various types of video degradation with a single unified model, overcoming the limitation of standard methods that require specific models for each restoration task. Our contributions include an efficient training strategy with Task Prompt Guidance (TPG) for diverse restoration tasks, an inference strategy that combines Denoising Diffusion Implicit Models~(DDIM) inversion with a novel Sliding Window Cross-Frame Attention (SW-CFA) mechanism for enhanced content preservation and temporal consistency, and a scalable pipeline that makes our method all-in-one to adapt to different video restoration tasks. Through extensive experiments on five video restoration tasks, we demonstrate the superiority of our method in generalization capability to real-world videos and temporal consistency preservation over existing state-of-the-art methods. Our method advances the video restoration task by providing a unified solution that enhances video quality across multiple applications.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑