2025-11-20T02:40:14.869934

A PDE-Based Image Dehazing Method via Atmospheric Scattering Theory

Hu, Wang, Gao et al.

This paper introduces a novel partial differential equation (PDE) framework for single-image dehazing. We embed the atmospheric scattering model into a PDE featuring edge-preserving diffusion and a nonlocal operator to maintain both local details and global structures. A key innovation is an adaptive regularization mechanism guided by the dark channel prior, which adjusts smoothing strength based on haze density. The framework's mathematical well-posedness is rigorously established by proving the existence and uniqueness of its weak solution in $H_0^1(Î©)$. An efficient, GPU-accelerated fixed-point solver is used for implementation. Experiments confirm our method achieves effective haze removal while preserving high image fidelity, offering a principled alternative to purely data-driven techniques.

academic

A PDE-Based Image Dehazing Method via Atmospheric Scattering Theory

基本信息

论文ID: 2506.08793
标题: A PDE-Based Image Dehazing Method via Atmospheric Scattering Theory
作者: Liubing Hu, Pu Wang, Guangwei Gao, Chunyan Wang, Zhuoran Zheng
分类: cs.CV (Computer Vision), eess.IV (Image and Video Processing)
发表时间: 2025年（arXiv预印本）
论文链接: https://arxiv.org/abs/2506.08793

摘要

本文提出了一种基于偏微分方程（PDE）的单图像去雾新框架。该方法将大气散射模型嵌入到具有边缘保持扩散和非局部算子的PDE中，以维持局部细节和全局结构。关键创新是基于暗通道先验的自适应正则化机制，根据雾霾密度调整平滑强度。通过证明H₀¹(Ω)空间中弱解的存在性和唯一性，严格建立了框架的数学适定性。采用高效的GPU加速定点求解器进行实现。实验证实该方法在有效去雾的同时保持高图像保真度，为纯数据驱动技术提供了有原则的替代方案。

研究背景与动机

问题定义

图像去雾是计算机视觉中的关键挑战，主要解决大气散射导致的图像退化问题。雾霾图像遵循大气散射模型：I(x) = J(x)t(x) + A(1-t(x))，其中I(x)是观测到的雾霾图像，J(x)是清晰图像，t(x)是透射率图，A是大气光值。

问题重要性

实际应用需求：雾霾天气严重影响户外视觉系统性能，如自动驾驶、监控系统等
图像质量提升：去雾能显著改善图像对比度和可视性
后续处理基础：为目标检测、场景理解等高级视觉任务提供高质量输入

现有方法局限性

传统物理模型方法：
- 暗通道先验（DCP）在天空区域和均匀场景中失效
- 导致不准确的透射率图估计
数据驱动方法：
- 需要大量标注数据
- 缺乏可解释性
- 在分布外雾霾条件下表现不佳

研究动机

结合物理建模与数学严谨性，提供一个既有理论保证又实用有效的去雾框架，弥补传统方法和深度学习方法的不足。

核心贡献

创新的PDE框架：首次将大气散射模型嵌入到具有边缘保持扩散和非局部算子的PDE中
自适应正则化机制：基于暗通道先验的自适应正则化参数，根据雾霾密度动态调整平滑强度
数学理论保证：严格证明了PDE弱解在H₀¹(Ω)空间中的存在性和唯一性
高效实现：GPU加速的定点迭代求解器，实现实时处理
优越性能：在多个无参考图像质量评估指标上达到最先进性能

方法详解

任务定义

输入：雾霾图像I(x) ∈ ℝᴴˣᵂˣ³ 输出：去雾图像u(x) ∈ ℝᴴˣᵂˣ³ 约束：遵循大气散射物理模型，保持图像结构和细节

模型架构

1. 大气散射模型嵌入

重构算子定义为：

Φ(I, t, A) = (I - A(1-t)) / max(t, t₀)

其中t₀ > 0是避免数值不稳定的小阈值。

2. 核心PDE模型

-div(D(∇u)∇u) + λ(t)G(u) = Φ(I, t, A)

组成部分：

边缘保持扩散项：-div(D(∇u)∇u)
非局部正则化项：λ(t)G(u)
数据保真项：Φ(I, t, A)

3. 边缘保持扩散机制

扩散系数设计为：

D(∇u) = (|∇u| + ε)⁻¹

其中ε = 10⁻³。该设计确保：

在强边缘处（大梯度）抑制扩散
在平滑区域（小梯度）促进扩散

4. 非局部正则化

高斯卷积算子：

G(u)(x) = ∫_Ω K(x,y)u(y)dy

高斯核：

K(x,y) = (1/2πσ²)exp(-‖x-y‖²/2σ²)

5. 自适应正则化参数

基于暗通道先验的自适应参数：

λ(t) = λ₀exp(-β(1-t))

其中λ₀ = 0.5，β = 3.0。该设计确保：

雾霾密集区域（小t值）：强正则化
清晰区域（大t值）：弱正则化，保持细节

技术创新点

物理一致性：将大气散射模型直接嵌入PDE约束，确保物理合理性
自适应性：根据局部雾霾浓度动态调整正则化强度
多尺度处理：结合局部扩散和非局部算子，同时处理局部细节和全局结构
数学严谨性：提供完整的理论分析，证明解的存在唯一性

实验设置

数据集

真实世界数据集：包含多样化雾霾图像
场景类型：城市街道、自然景观、建筑物
雾霾浓度：从轻度到重度不等
评估方式：由于缺乏真实对应的清晰图像，采用无参考图像质量评估

评价指标

采用6个广泛认可的无参考图像质量评估（NR-IQA）指标：

NIQE、BRISQUE、PIQE、FADE：越小越好
DHQI、NRBP：越大越好

对比方法

传统方法：DCP、CAP、Color-Lines、ICAP、BCCR
融合方法：VarFusion
深度学习方法：CORUN、DiffDehaze

实现细节

平台：Python + PyTorch，NVIDIA RTX 4090 GPU
参数设置：
- 暗通道块大小：15×15
- 扩散稳定项：ε = 10⁻³
- 高斯核：5×5，σ = 2.0
- 松弛参数：τ = 0.2
- 自适应参数：λ₀ = 0.5，β = 3.0

实验结果

主要结果

在6个NR-IQA指标中，本方法在5个指标上达到最佳性能：

方法	NIQE↓	BRISQUE↓	PIQE↓	FADE↓	DHQI↑	NRBP↑
DCP	5.34	32.97	48.09	48.81	55.73	24.08
DiffDehaze	4.85	30.15	39.50	43.55	56.20	74.32
Ours	4.51	28.46	35.68	50.66	57.85	76.82

消融实验

系统性移除各组件验证其贡献：

模型变体	NIQE↓	BRISQUE↓	PIQE↓	NRBP↑
无PDE优化	5.032	31.71	37.64	41.49
无非局部正则化	8.25	43.58	36.92	39.87
无自适应正则化	8.41	33.43	36.67	40.15
完整模型	4.51	28.46	35.68	76.82