双向扩散建模光照退化,提升暗光图像增强质量
Degradation-Consistent Learning via Bidirectional Diffusion for Low-Light Image Enhancement
- 双向扩散同时学习明暗图像退化过程
- 在多个数据集上超越现有方法,视觉效果更自然
- 适合需要高保真图像增强的科研与应用
暗光图像增强旨在提升退化图像的可见性,使其更符合人类视觉感知。尽管基于扩散的方法因其强大的生成能力展现出良好性能,但其单向退化建模难以捕捉真实世界退化的复杂性,常导致结构不一致和像素错位。为此,我们提出一种双向扩散优化机制,联合建模暗光与正常光照图像的退化过程,实现更精准的退化参数匹配,提升生成质量。具体而言,训练时同时进行从暗到亮和从亮到暗的双向扩散,并引入自适应特征交互模块(AFI)以优化特征表示。通过两条路径的互补性,隐式施加光照衰减与噪声分布的对称约束,促进退化一致性学习,增强模型对光照与细节退化的感知能力。此外,设计反射校正模块(RACM)指导去噪后的色彩恢复并抑制过曝区域,确保内容一致性,生成符合人类视觉感知的高质量图像。大量实验表明,该方法在多个基准数据集上均优于现有最先进方法,在定量与定性评价中表现优异,且对多样退化场景具有强泛化能力。代码已开源:https://github.com/hejh8/BidDiff
原文摘要 · Abstract (English)
Low-light image enhancement aims to improve the visibility of degraded images to better align with human visual perception. While diffusion-based methods have shown promising performance due to their strong generative capabilities. However, their unidirectional modelling of degradation often struggles to capture the complexity of real-world degradation patterns, leading to structural inconsistencies and pixel misalignments. To address these challenges, we propose a bidirectional diffusion optimization mechanism that jointly models the degradation processes of both low-light and normal-light images, enabling more precise degradation parameter matching and enhancing generation quality. Specifically, we perform bidirectional diffusion-from low-to-normal light and from normal-to-low light during training and introduce an adaptive feature interaction block (AFI) to refine feature representation. By leveraging the complementarity between these two paths, our approach imposes an implicit symmetry constraint on illumination attenuation and noise distribution, facilitating consistent degradation learning and improving the models ability to perceive illumination and detail degradation. Additionally, we design a reflection-aware correction module (RACM) to guide color restoration post-denoising and suppress overexposed regions, ensuring content consistency and generating high-quality images that align with human visual perception. Extensive experiments on multiple benchmark datasets demonstrate that our method outperforms state-of-the-art methods in both quantitative and qualitative evaluations while generalizing effectively to diverse degradation scenarios. Code at https://github.com/hejh8/BidDiff
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。