基于物理模型的扩散方法,让去雾更贴近真实雾霾形成过程。
HNDiff: Haze-Noise Diffusion for Image Dehazing

- 用物理散射模型做先验,正向过程分区域加雾和噪声
- 逆过程同步去雾去噪,恢复效果优于现有方法
- 可嵌入现有网络,适合做图像去雾研究者参考
现有基于扩散的方法在图像去雾方面取得了显著进展,但通常忽略雾霾形成的物理机制,从纯高斯噪声中重建清晰图像,限制了恢复潜力。为此,我们提出雾-噪扩散(HNDiff),一种将大气散射模型作为归纳偏置的新扩散框架。通过将扩散过程建立在物理原理之上,HNDiff确保恢复结果更符合雾霾形成机制。正向过程中,引入联合雾-噪扩散与雾感知噪声调度器,逐步向图像添加雾和噪声;调度器根据雾密度自适应调整噪声强度——雾越重区域注入更强噪声以促进内容生成,较清区域注入较弱噪声以更好保留细节,直接关联前向退化过程与雾霾物理特性。反向过程中,推导出物理一致的去雾-去噪流程,同步去除雾和噪声,恢复清晰图像,且与前向过程对齐。为进一步提升实用性,提出潜空间版HNDiff,整合干净潜在先验,可无缝集成至现有去雾网络以提升性能。大量实验表明,该方法显著提升主流去雾骨干网络,在基准数据集上达到最先进水平。
原文摘要 · Abstract (English)
Existing diffusion-based methods have recently made significant progress in image dehazing. However, they typically neglect the physics of haze formation and reconstruct clean images from pure Gaussian noise, thereby limiting their restoration potential. To address this issue, we propose Haze-Noise Diffusion (HNDiff), a novel diffusion framework that embeds the atmospheric scattering model as an inductive bias. By grounding diffusion in physical principles, HNDiff ensures that the restoration aligns more closely with underlying mechanisms of haze formation. In its forward process, we introduce joint haze-noise diffusion with a haze-aware noise scheduler, which progressively adds both haze and noise to an image. Essentially, the scheduler adapts noise levels according to haze density, meaning that regions with heavier haze receive stronger noise injection to encourage content generation, while clearer regions receive lighter noise to better preserve details, which directly links the forward degradation process with the physics of haze. In the reverse process, we then derive a physically consistent dehazing-denoising process that simultaneously removes haze and noise to restore a clean image in a manner aligned with the forward degradation process. To further enhance practicality, we propose Latent HNDiff, which compiles clean latent priors that can be seamlessly integrated into existing dehazing networks to boost performance. Extensive experiments show that our work significantly improves leading dehazing backbones and achieves state-of-the-art results on benchmark datasets. The project page is available at https://jin-ting-he.github.io/HNDiff .
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。