mlsanityData to 14 Oct 2026

UniFlowRestore: A General Video Restoration Framework via Flow Matching and Prompt Guidance

arXiv 2504.09069 ↗First posted 12 Apr 2025cs.CVcited by 2

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

  1. 2022Hierarchical Text-Conditional Image Generation with CLIP Latentscited by 6,585
  2. 2022Flow Matching for Generative Modelingcited by 5,702
  3. 2024Scaling Rectified Flow Transformers for High-Resolution Image Synthesiscited by 4,181
  4. 2021Restormer: Efficient Transformer for High-Resolution Image Restorationcited by 1,627
  5. 2017The 2017 DAVIS Challenge on Video Object Segmentationcited by 957
  6. 2021Learning to Prompt for Continual Learningcited by 944
  7. 2016Deep Multi-scale Convolutional Neural Network for Dynamic Scene Deblurringcited by 727
  8. 2017FFDNet: Toward a Fast and Flexible Solution for CNN based Image Denoisingcited by 510
  9. 2019EDVR: Video Restoration with Enhanced Deformable Convolutional Networkscited by 494
  10. 2017Benchmarking Single Image Dehazing and Beyondcited by 447
  11. 2017Learning Deep CNN Denoiser Prior for Image Restorationcited by 408
  12. 2024Discrete Flow Matchingcited by 317
21 more
  1. 2018Density-aware Single Image De-raining using a Multi-stream Dense Networkcited by 248
  2. 2021TransWeather: Transformer-based Restoration of Images Degraded by Adverse Weather Conditionscited by 244
  3. 2022VRT: A Video Restoration Transformercited by 220
  4. 2020Human-Aware Motion Deblurringcited by 175
  5. 2023Multitask Prompt Tuning Enables Parameter-Efficient Transfer Learningcited by 115
  6. 2023ProRes: Exploring Degradation-aware Visual Prompt for Universal Image Restorationcited by 77
  7. 2019A General Decoupled Learning Framework for Parameterized Image Operatorscited by 72
  8. 2022A Simple Baseline for Video Restoration with Grouped Spatial-temporal Shiftcited by 68
  9. 2018Robust Video Content Alignment and Compensation for Rain Removal in a CNN Frameworkcited by 53
  10. 2021NTIRE 2021 Challenge on Image Deblurringcited by 50
  11. 2021BANet: Blur-aware Attention Networks for Dynamic Scene Deblurringcited by 45
  12. 2023Prompt-In-Prompt Learning for Universal Image Restorationcited by 43
  13. 2024PromptCIR: Blind Compressed Image Restoration with Prompt Learningcited by 20
  14. 2024How Do Flow Matching Models Memorize and Generalize in Sample Data Subspaces?cited by 19
  15. 2025FLowHigh: Towards Efficient and High-Quality Audio Super-Resolution with Single-Step Flow Matchingcited by 18
  16. 2024Blur-aware Spatio-temporal Sparse Transformer for Video Deblurringcited by 17
  17. 2022Multi-Scale Memory-Based Video Deblurringcited by 16
  18. 2025Denoising Hamiltonian Network for Physical Reasoningcited by 10
  19. 2023Cross-Consistent Deep Unfolding Network for Adaptive All-In-One Video Restorationcited by 5
  20. 2025Efficient Point Clouds Upsampling via Flow Matchingcited by 4
  21. 2024Review Learning: Advancing All-in-One Ultra-High-Definition Image Restoration Training Methodcited by 3

Built on by

Cited by year (2026 to 14 Oct)

Citing papers per year: 2025: 1; 2026: 1.

2 papers.

Similar, not cited

Close in wording, no citation either way

  1. 2026D2-VR: Degradation-Robust and Distilled Video Restoration with Synergistic Optimization Strategy
  2. 2025TDM: Temporally-Consistent Diffusion Model for All-in-One Real-World Video Restoration
  3. 2025HaineiFRDM: Structure-Preserving Diffusion for Film Restoration under Fast Motion and Diverse Defects
  4. 2025LVTINO: LAtent Video consisTency INverse sOlver for High Definition Video Restoration
  5. 2025SeedVR: Seeding Infinity in Diffusion Transformer Towards Generic Video Restoration
  6. 2025DiTVR: Zero-Shot Diffusion Transformer for Video Restoration
  7. 2024DiffIR2VR-Zero: Zero-Shot Video Restoration with Diffusion-based Image Restoration Models
  8. 2026Flow Straight to Reality: Perceptually Consistent Flow Matching for Efficient Image Restoration