基于布朗桥扩散模型的单幅图像去雾研究

    Single hazy image dehazing based on Brownian Bridge diffusion model

    • 摘要: 针对现有扩散模型在图像去雾任务中因域间特征差异导致的跨域转换质量受限问题,本研究突破传统条件生成框架对有雾与清晰域间分布差异建模的局限性,提出基于布朗桥扩散模型的新型去雾方法。基于布朗桥扩散模型的新型转换框架,将有雾域与清晰域之间的图像转换建模为随机布朗桥过程。通过设计双向扩散机制,构建雾化图像与清晰图像间的双向扩散轨迹,实现两个域间特征分布的端到端联合学习。实验结果表明,本方法在跨域图像转换任务中展现出显著优势,其双向扩散机制有效克服了传统条件生成模型的域间差异问题,在RESIDE数据集上取得37.26 dB的峰值信噪比、0.9813的结构相似度以及2.67的自然图像质量评估。布朗桥扩散模型通过构建有雾域与清晰域间的双向布朗运动轨迹,实现了雾气退化特征与清晰图像特征的定向解耦与精准对齐,显著提升了复杂大气散射场景下的图像复原精度。

       

      Abstract: Aiming at the problem of limited cross-domain transformation quality caused by inter-domain feature differences in the existing diffusion models in image defogging task, this study breaks through the limitation of the traditional conditional generation framework in modeling the distribution differences between foggy and clear domains and proposes a new defogging method based on the Brownian Bridge diffusion model. A new transformation framework based on the Brownian Bridge diffusion model was proposed, and the image transformation between the foggy domain and the clear domain was modeled as a random Brownian bridge process. By designing a bidirectional diffusion mechanism, a bidirectional diffusion trajectory between the fogged image and the clear image was constructed to achieve end-to-end joint learning of the feature distribution between the two domains. The experimental results show that this method demonstrates significant advantages in the cross-domain image transformation task. Its bidirectional diffusion mechanism effectively overcomes the inter-domain difference problem of traditional conditional generation models, achieving a peak signal-to-noise ratio of 37.26 dB, a structural similarity of 0.9813, and a natural image quality assessment of 2.67 on the RESIDE dataset. The Brownian Bridge diffusion model achieves the directional decoupling and precise alignment of fog degradation features and clear image features by constructing a bidirectional Brownian motion trajectory between the fog domain and the clear domain, significantly improving the image restoration accuracy in complex atmospheric scattering scenarios.

       

    /

    返回文章
    返回