Paper Reading AI Learner

DiTPainter: Efficient Video Inpainting with Diffusion Transformers

2025-04-22 07:36:45
Xian Wu, Chang Liu

Abstract

Many existing video inpainting algorithms utilize optical flows to construct the corresponding maps and then propagate pixels from adjacent frames to missing areas by mapping. Despite the effectiveness of the propagation mechanism, they might encounter blurry and inconsistencies when dealing with inaccurate optical flows or large masks. Recently, Diffusion Transformer (DiT) has emerged as a revolutionary technique for video generation tasks. However, pretrained DiT models for video generation all contain a large amount of parameters, which makes it very time consuming to apply to video inpainting tasks. In this paper, we present DiTPainter, an end-to-end video inpainting model based on Diffusion Transformer (DiT). DiTPainter uses an efficient transformer network designed for video inpainting, which is trained from scratch instead of initializing from any large pretrained models. DiTPainter can address videos with arbitrary lengths and can be applied to video decaptioning and video completion tasks with an acceptable time cost. Experiments show that DiTPainter outperforms existing video inpainting algorithms with higher quality and better spatial-temporal consistency.

Abstract (translated)

许多现有的视频修复算法利用光流来构建相应的映射,然后通过映射将相邻帧的像素传播到缺失区域。虽然这种传播机制非常有效,但在处理不准确的光流或大面积遮罩时可能会导致模糊和不一致的问题。最近,扩散变换器(Diffusion Transformer, DiT)作为视频生成任务中的革命性技术崭露头角。然而,用于视频生成的预训练DiT模型参数量巨大,这使得它们在应用于视频修复任务时非常耗时。 在这篇论文中,我们介绍了DiTPainter,这是一个基于扩散变换器(Diffusion Transformer, DiT)的端到端视频修复模型。DiTPainter使用了一种专门设计用于视频修复的有效Transformer网络,并且是从头开始训练的,而不是从任何大型预训练模型初始化的。DiTPainter可以处理任意长度的视频,并能以可接受的时间成本应用于视频去遮挡和视频完成任务。实验表明,DiTPainter在质量和时空一致性方面优于现有的视频修复算法。

URL

https://arxiv.org/abs/2504.15661

PDF

https://arxiv.org/pdf/2504.15661.pdf


Tags
3D Action Action_Localization Action_Recognition Activity Adversarial Agent Attention Autonomous Bert Boundary_Detection Caption Chat Classification CNN Compressive_Sensing Contour Contrastive_Learning Deep_Learning Denoising Detection Dialog Diffusion Drone Dynamic_Memory_Network Edge_Detection Embedding Embodied Emotion Enhancement Face Face_Detection Face_Recognition Facial_Landmark Few-Shot Gait_Recognition GAN Gaze_Estimation Gesture Gradient_Descent Handwriting Human_Parsing Image_Caption Image_Classification Image_Compression Image_Enhancement Image_Generation Image_Matting Image_Retrieval Inference Inpainting Intelligent_Chip Knowledge Knowledge_Graph Language_Model LLM Matching Medical Memory_Networks Multi_Modal Multi_Task NAS NMT Object_Detection Object_Tracking OCR Ontology Optical_Character Optical_Flow Optimization Person_Re-identification Point_Cloud Portrait_Generation Pose Pose_Estimation Prediction QA Quantitative Quantitative_Finance Quantization Re-identification Recognition Recommendation Reconstruction Regularization Reinforcement_Learning Relation Relation_Extraction Represenation Represenation_Learning Restoration Review RNN Robot Salient Scene_Classification Scene_Generation Scene_Parsing Scene_Text Segmentation Self-Supervised Semantic_Instance_Segmentation Semantic_Segmentation Semi_Global Semi_Supervised Sence_graph Sentiment Sentiment_Classification Sketch SLAM Sparse Speech Speech_Recognition Style_Transfer Summarization Super_Resolution Surveillance Survey Text_Classification Text_Generation Time_Series Tracking Transfer_Learning Transformer Unsupervised Video_Caption Video_Classification Video_Indexing Video_Prediction Video_Retrieval Visual_Relation VQA Weakly_Supervised Zero-Shot