DiTVR: Zero-Shot Diffusion Transformer for Video Restoration

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Gao, Sicheng, Mehta, Nancy, Wu, Zongwei, Timofte, Radu
Format: Preprint
Published: 2025
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866908484377772032
author Gao, Sicheng
Mehta, Nancy
Wu, Zongwei
Timofte, Radu
author_facet Gao, Sicheng
Mehta, Nancy
Wu, Zongwei
Timofte, Radu
contents Video restoration aims to reconstruct high quality video sequences from low quality inputs, addressing tasks such as super resolution, denoising, and deblurring. Traditional regression based methods often produce unrealistic details and require extensive paired datasets, while recent generative diffusion models face challenges in ensuring temporal consistency. We introduce DiTVR, a zero shot video restoration framework that couples a diffusion transformer with trajectory aware attention and a wavelet guided, flow consistent sampler. Unlike prior 3D convolutional or frame wise diffusion approaches, our attention mechanism aligns tokens along optical flow trajectories, with particular emphasis on vital layers that exhibit the highest sensitivity to temporal dynamics. A spatiotemporal neighbour cache dynamically selects relevant tokens based on motion correspondences across frames. The flow guided sampler injects data consistency only into low-frequency bands, preserving high frequency priors while accelerating convergence. DiTVR establishes a new zero shot state of the art on video restoration benchmarks, demonstrating superior temporal consistency and detail preservation while remaining robust to flow noise and occlusions.
format Preprint
id arxiv_https___arxiv_org_abs_2508_07811
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle DiTVR: Zero-Shot Diffusion Transformer for Video Restoration
Gao, Sicheng
Mehta, Nancy
Wu, Zongwei
Timofte, Radu
Computer Vision and Pattern Recognition
Video restoration aims to reconstruct high quality video sequences from low quality inputs, addressing tasks such as super resolution, denoising, and deblurring. Traditional regression based methods often produce unrealistic details and require extensive paired datasets, while recent generative diffusion models face challenges in ensuring temporal consistency. We introduce DiTVR, a zero shot video restoration framework that couples a diffusion transformer with trajectory aware attention and a wavelet guided, flow consistent sampler. Unlike prior 3D convolutional or frame wise diffusion approaches, our attention mechanism aligns tokens along optical flow trajectories, with particular emphasis on vital layers that exhibit the highest sensitivity to temporal dynamics. A spatiotemporal neighbour cache dynamically selects relevant tokens based on motion correspondences across frames. The flow guided sampler injects data consistency only into low-frequency bands, preserving high frequency priors while accelerating convergence. DiTVR establishes a new zero shot state of the art on video restoration benchmarks, demonstrating superior temporal consistency and detail preservation while remaining robust to flow noise and occlusions.
title DiTVR: Zero-Shot Diffusion Transformer for Video Restoration
topic Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2508.07811