PVRF: All-in-one Adverse Weather Removal via Prior-modulated and Velocity-constrained Rectified Flow
About
Adverse weather removal (AWR) in real-world images remains challenging due to heterogeneous and unseen degradations, while distortion-driven training often yields overly smooth results. We propose PVRF, a unified framework that integrates zero-shot soft weather perceptions with velocity-constrained rectified-flow refinement. PVRF introduces an AWR-specific question answering module (AWR-QA) that uses frozen vision--language models (VLMs) to estimate soft probabilities of weather types and low-level attribute scores. These perceptions condition restoration networks via attribute-modulated normalization (AMN) and weather-weighted adapters (WWA), producing an anchor estimate for refinement. We then learn a terminal-consistent residual rectified flow with perception-adaptive source perturbation and a terminal-consistent velocity parameterization to stabilize learning near the terminal regime. Extensive experiments show that PVRF improves both fidelity and perceptual quality over state-of-the-art baselines, with strong cross-dataset generalization on single and combined degradations. Code will be released at https://github.com/dongw22/PVRF.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Low-light enhancement | MEF, NPE, DICM (unseen) | NIQE3.4777 | 9 | |
| Deblurring | HIDE (unseen) | MUSIQ Score55.1979 | 6 | |
| Deblurring | General Image Restoration Setting III (test) | PSNR28.8893 | 6 | |
| Dehazing | CDD-11 (unseen) | MUSIQ Score71.1582 | 6 | |
| Dehazing | General Image Restoration Setting III (test) | PSNR29.5607 | 6 | |
| Deraining | RealRain-1k (unseen) | MUSIQ40.7962 | 6 | |
| Deraining | General Image Restoration Setting III (test) | PSNR32.0514 | 6 | |
| Desnowing | CDD-11 (unseen) | MUSIQ Score70.7852 | 6 | |
| Desnowing | General Image Restoration Setting III (test) | PSNR28.9248 | 6 | |
| Image Restoration | Dataset Haze + Rain | MUSIQ70.5356 | 6 |