arXiv:2505.04369cs.CV2025-05被引 17

利用小波分析发现雾霾信息集中于低频,提出分阶段去雾新框架。

WDMamba: When Wavelet Degradation Prior Meets Vision Mamba for Image Dehazing

  • 通过小波变换发现雾霾信息主要在低频成分,据此设计先恢复低频再增强细节的两阶段流程。
  • 在低频恢复阶段使用Mamba模块实现线性复杂度建模,高效去除整体雾霾并生成粗略恢复图。
  • 引入自引导对比正则化,提升细节保留能力,使结果更自然,适合图像质量要求高的场景。

本文通过小波变换分析揭示了一种新的雾霾特有小波退化先验:雾霾相关信息主要集中在低频成分。基于此,我们提出新型去雾框架WDMamba,将去雾任务分解为两个顺序阶段——低频恢复与细节增强。该粗到精策略使模型能有效捕捉各阶段特征,从而生成高质量恢复图像。具体而言,在低频恢复阶段,集成Mamba模块以线性复杂度重建全局结构,高效去除整体雾霾并生成粗略恢复图像;随后,细节增强阶段恢复前一阶段可能忽略的精细信息,最终输出去雾结果。此外,为提升细节保留能力并实现更自然的去雾效果,我们在训练中引入自引导对比正则化,利用粗略恢复输出作为硬负例,促使模型学习更具区分性的表征,显著提升整体性能。在多个公开去雾基准上的广泛评估表明,本方法在定性和定量上均优于现有最先进方法。代码已开源:https://github.com/SunJ000/WDMamba。

原文摘要 · Abstract (English)

In this paper, we reveal a novel haze-specific wavelet degradation prior observed through wavelet transform analysis, which shows that haze-related information predominantly resides in low-frequency components. Exploiting this insight, we propose a novel dehazing framework, WDMamba, which decomposes the image dehazing task into two sequential stages: low-frequency restoration followed by detail enhancement. This coarse-to-fine strategy enables WDMamba to effectively capture features specific to each stage of the dehazing process, resulting in high-quality restored images. Specifically, in the low-frequency restoration stage, we integrate Mamba blocks to reconstruct global structures with linear complexity, efficiently removing overall haze and producing a coarse restored image. Thereafter, the detail enhancement stage reinstates fine-grained information that may have been overlooked during the previous phase, culminating in the final dehazed output. Furthermore, to enhance detail retention and achieve more natural dehazing, we introduce a self-guided contrastive regularization during network training. By utilizing the coarse restored output as a hard negative example, our model learns more discriminative representations, substantially boosting the overall dehazing performance. Extensive evaluations on public dehazing benchmarks demonstrate that our method surpasses state-of-the-art approaches both qualitatively and quantitatively. Code is available at https://github.com/SunJ000/WDMamba.

去雾小波分析Mamba图像恢复

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。