PVRF: All-in-one Adverse Weather Removal via Prior-modulated and Velocity-constrained Rectified Flow

· Source: cs.CV updates on arXiv.org · Field: Technology & Digital — Artificial Intelligence & Machine Learning, Computer Vision · Depth: Expert, extended

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

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

Topics

Code references

Best for: Research Scientist, AI Scientist, Machine Learning Engineer, Computer Vision Engineer

Related on AIssential

Open in AIssential →

Editorial summary, takeaway, and curation by AIssential. Original article published by cs.CV updates on arXiv.org.