arXiv:2604.23709cs.CVeess.IV2026-04

用扩散模型先学先验,推理时丢掉它,实现高效去雾。

ZID-Net: Zero-Inference Diffusion Prior Decoupling Network for Single Image Dehazing

  • 去雾网络分两步:训练时用扩散模型提供先验,推理时纯前向传播。
  • 在RESIDE数据集上达40.75 dB PSNR,比现有方法高1.13 dB。
  • 推理仅需19.35毫秒,适合实时应用,尤其适合遥感图像去雾。

单图像去雾常受限于恢复质量与计算效率之间的权衡。尽管高效,卷积神经网络难以学习密集且非均匀雾霾的鲁棒先验;而扩散模型虽具备强大生成先验,却存在严重推理延迟和采样不稳定性。为此,我们提出ZID-Net,一种显式解耦扩散监督与前向推理的新框架。为实现高效推理,设计了频域-空间解耦的前向骨干网络:通道-空间拉普拉斯掩码(CSLM)过滤雾霾增强噪声,提取纯净结构细节;轻量级全局上下文块(LGCB)建立长程空间依赖,捕捉雾霾的全局变化。动态特征仲裁块(DFAB)自适应融合语义与结构特征以实现鲁棒重建。训练阶段引入零推理先验传播头(ZI-PPH),通过条件扩散过程预测残差噪声,为骨干网络提供退化感知的结构监督。测试时舍弃扩散分支,将扩散先验融入纯前向架构,实现精准高效恢复。ZID-Net在合成RESIDE数据集上取得40.75 dB PSNR,在真实世界数据集上优于现有方法1.13 dB。此外,在StateHaze1k遥感数据集上提升3.06 dB PSNR,推理时间仅19.35毫秒。项目代码已公开。

原文摘要 · Abstract (English)

Single image dehazing is often constrained by a trade-off between restoration quality and computational efficiency. While efficient, CNN networks struggle to learn robust priors for dense and non-homogeneous haze. Conversely, diffusion models provide strong generative priors but suffer from severe inference latency and sampling instability. To address these limitations, we propose ZID-Net, a novel framework that explicitly decouples diffusion supervision from feed-forward inference. For efficient inference, we design a frequency-spatial decoupled feed-forward backbone. Within this backbone, a Channel-Spatial Laplacian Mask (CSLM) filters haze-amplified noise to extract purified structural details, while Lightweight Global Context Blocks (LGCBs) establish long-range spatial dependencies to capture the global variations of haze. A Dynamic Feature Arbitration Block (DFAB) then adaptively fuses these semantic and structural features for robust reconstruction. To provide this backbone with physical priors without the inference cost, we introduce a Zero-Inference Prior Propagation Head (ZI-PPH) during training. ZI-PPH leverages a conditional diffusion process to predict residual noise, providing degradation-aware structural supervision to the backbone. By discarding the diffusion branch at test time, ZID-Net integrates diffusion priors into a pure feed-forward architecture for accurate and efficient restoration. ZID-Net achieves 40.75 dB PSNR on the synthetic RESIDE dataset and outperforms existing methods with a 1.13 dB gain on real-world datasets. Additionally, it yields a 3.06 dB PSNR gain on the StateHaze1k remote sensing dataset with an inference time of just 19.35 ms. The project code is available at: https://github.com/XoomitLXH/ZID-Net.

去雾扩散模型高效推理遥感图像

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