PVRF: All-in-one Adverse Weather Removal via Prior-modulated and Velocity-constrained Rectified Flow
Summary
PVRF is a novel, unified framework designed for all-in-one adverse weather removal (AWR) in real-world images, addressing challenges like heterogeneous degradations and overly smooth restoration results. Developed by researchers from McMaster University and Shanghai Jiao Tong University, PVRF integrates zero-shot soft weather perceptions with velocity-constrained rectified-flow refinement. It features an AWR-specific question answering module (AWR-QA) that leverages frozen vision-language models (VLMs) to estimate soft probabilities of weather types and low-level attribute scores. These perceptions condition restoration networks through Attribute-Modulated Normalization (AMN) and Weather-Weighted Adapters (WWA), providing an anchor estimate for refinement. PVRF then learns a terminal-consistent residual rectified flow with perception-adaptive source perturbation and terminal-consistent velocity parameterization, stabilizing learning near the terminal regime. Extensive experiments demonstrate that PVRF significantly improves both fidelity and perceptual quality over state-of-the-art baselines, showing strong cross-dataset generalization on single and combined degradations.
Key takeaway
For research scientists developing robust image restoration models, PVRF offers a compelling approach to overcome limitations of traditional AWR. You should consider integrating VLM-extracted soft perceptions and terminal-consistent rectified flow into your models to achieve superior fidelity and perceptual quality, especially for handling diverse and combined weather degradations in real-world scenarios.
Key insights
PVRF unifies adverse weather removal by combining VLM-derived soft perceptions with a perception-guided rectified flow for enhanced fidelity and realism.
Principles
- Soft conditioning is superior to hard one-hot encoding for mixed degradations.
- Perception-distortion trade-off requires balancing fidelity and perceptual quality.
- Terminal-consistent velocity parameterization stabilizes rectified flow learning.
Method
PVRF uses a VLM-based AWR-QA module for soft weather and attribute perceptions, then applies AMN and WWA for degradation-aware posterior estimation, followed by a terminal-consistent residual rectified flow with adaptive source perturbation for photo-realistic refinement.
In practice
- Use VLMs for zero-shot, soft weather perception.
- Modulate normalization and adapters with perception priors.
- Employ residual rectified flow for photo-realistic refinement.
Topics
- Adverse Weather Removal
- Rectified Flow
- Vision-Language Models
- Zero-shot Perception
- Degradation-aware Conditioning
Code references
Best for: Research Scientist, AI Scientist, Machine Learning Engineer, Computer Vision Engineer
Related on AIssential
Editorial summary, takeaway, and curation by AIssential. Original article published by cs.CV updates on arXiv.org.