用微分方程重建雾霾模型,单步去雾更真实。
HazeFlow: Revisit Haze Physical Model as ODE and Non-Homogeneous Haze Generation for Real-World Dehazing
- 将雾霾物理模型转为微分方程,学习最优去雾路径。
- 在多个真实场景数据集上达到当前最佳性能。
- 自动生成多样雾霾图像,缓解真实配对数据不足。
去雾旨在通过估计大气散射效应,去除图像中的雾霾以恢复清晰度。尽管深度学习方法表现优异,但真实世界配对数据稀缺导致域差异,限制其在实际场景中的泛化能力。在此背景下,基于物理规律的学习至关重要;然而,传统基于大气散射模型(ASM)的方法难以应对真实复杂性和多样的雾霾模式。为此,我们提出 HazeFlow,一种基于常微分方程(ODE)的新型框架,将 ASM 重新形式化为 ODE。受修正流(Rectified Flow)启发,HazeFlow 学习最优 ODE 轨迹,实现从有雾图像到清晰图像的映射,仅需一步推理即可提升真实场景去雾效果。此外,我们引入基于马尔可夫链布朗运动(MCBM)的非均匀雾霾生成方法,模拟逼真的雾霾分布,解决真实配对数据匮乏问题。通过大量实验,HazeFlow 在多个真实世界去雾基准数据集上均取得领先性能。
原文摘要 · Abstract (English)
Dehazing involves removing haze or fog from images to restore clarity and improve visibility by estimating atmospheric scattering effects. While deep learning methods show promise, the lack of paired real-world training data and the resulting domain gap hinder generalization to real-world scenarios. In this context, physics-grounded learning becomes crucial; however, traditional methods based on the Atmospheric Scattering Model (ASM) often fall short in handling real-world complexities and diverse haze patterns. To solve this problem, we propose HazeFlow, a novel ODE-based framework that reformulates ASM as an ordinary differential equation (ODE). Inspired by Rectified Flow (RF), HazeFlow learns an optimal ODE trajectory to map hazy images to clean ones, enhancing real-world dehazing performance with only a single inference step. Additionally, we introduce a non-homogeneous haze generation method using Markov Chain Brownian Motion (MCBM) to address the scarcity of paired real-world data. By simulating realistic haze patterns through MCBM, we enhance the adaptability of HazeFlow to diverse real-world scenarios. Through extensive experiments, we demonstrate that HazeFlow achieves state-of-the-art performance across various real-world dehazing benchmark datasets.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。