UniFlowRestore: A General Video Restoration Framework via Flow Matching and Prompt Guidance
On the map: Diffusion models → Inverse problems and image restoration
Abstract (arXiv)
Video imaging is often affected by complex degradations such as blur, noise, and compression artifacts. Traditional restoration methods follow a "single-task single-model" paradigm, resulting in poor generalization and high computational cost, limiting their applicability in real-world scenarios with diverse degradation types. We propose UniFlowRestore, a general video restoration framework that models restoration as a time-continuous evolution under a prompt-guided and physics-informed vector field. A physics-aware backbone PhysicsUNet encodes degradation priors as potential energy, while PromptGenerator produces task-relevant prompts as momentum. These components define a Hamiltonian system whose vector field integrates inertial dynamics, decaying physical gradients, and prompt-based guidance. The system is optimized via a fixed-step ODE solver to achieve efficient and unified restoration across tasks. Experiments show that UniFlowRestore delivers stateof-the-art performance with strong generalization and efficiency. Quantitative results demonstrate that UniFlowRestore achieves state-of-the-art performance, attaining the highest PSNR (33.89 dB) and SSIM (0.97) on the video denoising task, while maintaining top or second-best scores across all evaluated tasks.
Key facts. Cited by 2 papers in the map (data through 14 Oct 2026). Placed in Inverse problems and image restoration, one of 14 directions in Diffusion models. Builds on 33 of its 48 references that are in the map. First posted on arXiv 12 Apr 2025.
Builds on
33 of 48 references.
References · 33
- 2022Hierarchical Text-Conditional Image Generation with CLIP Latentscited by 6,585
- 2022Flow Matching for Generative Modelingcited by 5,702
- 2024Scaling Rectified Flow Transformers for High-Resolution Image Synthesiscited by 4,181
- 2021Restormer: Efficient Transformer for High-Resolution Image Restorationcited by 1,627
- 2017The 2017 DAVIS Challenge on Video Object Segmentationcited by 957
- 2021Learning to Prompt for Continual Learningcited by 944
- 2016Deep Multi-scale Convolutional Neural Network for Dynamic Scene Deblurringcited by 727
- 2017FFDNet: Toward a Fast and Flexible Solution for CNN based Image Denoisingcited by 510
- 2019EDVR: Video Restoration with Enhanced Deformable Convolutional Networkscited by 494
- 2017Benchmarking Single Image Dehazing and Beyondcited by 447
- 2017Learning Deep CNN Denoiser Prior for Image Restorationcited by 408
- 2024Discrete Flow Matchingcited by 317
21 more
- 2018Density-aware Single Image De-raining using a Multi-stream Dense Networkcited by 248
- 2021TransWeather: Transformer-based Restoration of Images Degraded by Adverse Weather Conditionscited by 244
- 2022VRT: A Video Restoration Transformercited by 220
- 2020Human-Aware Motion Deblurringcited by 175
- 2023Multitask Prompt Tuning Enables Parameter-Efficient Transfer Learningcited by 115
- 2023ProRes: Exploring Degradation-aware Visual Prompt for Universal Image Restorationcited by 77
- 2019A General Decoupled Learning Framework for Parameterized Image Operatorscited by 72
- 2022A Simple Baseline for Video Restoration with Grouped Spatial-temporal Shiftcited by 68
- 2018Robust Video Content Alignment and Compensation for Rain Removal in a CNN Frameworkcited by 53
- 2021NTIRE 2021 Challenge on Image Deblurringcited by 50
- 2021BANet: Blur-aware Attention Networks for Dynamic Scene Deblurringcited by 45
- 2023Prompt-In-Prompt Learning for Universal Image Restorationcited by 43
- 2024PromptCIR: Blind Compressed Image Restoration with Prompt Learningcited by 20
- 2024How Do Flow Matching Models Memorize and Generalize in Sample Data Subspaces?cited by 19
- 2025FLowHigh: Towards Efficient and High-Quality Audio Super-Resolution with Single-Step Flow Matchingcited by 18
- 2024Blur-aware Spatio-temporal Sparse Transformer for Video Deblurringcited by 17
- 2022Multi-Scale Memory-Based Video Deblurringcited by 16
- 2025Denoising Hamiltonian Network for Physical Reasoningcited by 10
- 2023Cross-Consistent Deep Unfolding Network for Adaptive All-In-One Video Restorationcited by 5
- 2025Efficient Point Clouds Upsampling via Flow Matchingcited by 4
- 2024Review Learning: Advancing All-in-One Ultra-High-Definition Image Restoration Training Methodcited by 3
Built on by
Citing papers per year: 2025: 1; 2026: 1.
2 papers.
Similar, not cited
Close in wording, no citation either way
- 2026D2-VR: Degradation-Robust and Distilled Video Restoration with Synergistic Optimization Strategy
- 2025TDM: Temporally-Consistent Diffusion Model for All-in-One Real-World Video Restoration
- 2025HaineiFRDM: Structure-Preserving Diffusion for Film Restoration under Fast Motion and Diverse Defects
- 2025LVTINO: LAtent Video consisTency INverse sOlver for High Definition Video Restoration
- 2025SeedVR: Seeding Infinity in Diffusion Transformer Towards Generic Video Restoration
- 2025DiTVR: Zero-Shot Diffusion Transformer for Video Restoration
- 2024DiffIR2VR-Zero: Zero-Shot Video Restoration with Diffusion-based Image Restoration Models
- 2026Flow Straight to Reality: Perceptually Consistent Flow Matching for Efficient Image Restoration