This paper studies a diffusion-based framework to address the low-light image enhancement problem. To harness the capabilities of diffusion models, we delve into this intricate process and advocate for the regularization of its inherent ODE-trajectory. To be specific, inspired by the recent research that low curvature ODE-trajectory results in a stable and effective diffusion process, we formulate a curvature regularization term anchored in the intrinsic non-local structures of image data, i.e., global structure-aware regularization, which gradually facilitates the preservation of complicated details and the augmentation of contrast during the diffusion process. This incorporation mitigates the adverse effects of noise and artifacts resulting from the diffusion process, leading to a more precise and flexible enhancement. To additionally promote learning in challenging regions, we introduce an uncertainty-guided regularization technique, which wisely relaxes constraints on the most extreme regions of the image. Experimental evaluations reveal that the proposed diffusion-based framework, complemented by rank-informed regularization, attains distinguished performance in low-light enhancement. The outcomes indicate substantial advancements in image quality, noise suppression, and contrast amplification in comparison with state-of-the-art methods. We believe this innovative approach will stimulate further exploration and advancement in low-light image processing, with potential implications for other applications of diffusion models. The code is publicly available at https://github.com/jinnh/GSAD.
翻译:本文研究了一种基于扩散模型的框架,以解决低光照图像增强问题。为充分发挥扩散模型的能力,我们深入探究了这一复杂过程,并提出了对其固有常微分方程轨迹的正则化方法。具体而言,受近期关于低曲率常微分方程轨迹能产生稳定且高效扩散过程的研究启发,我们构建了一个基于图像数据内在非局部结构的曲率正则化项,即全局结构感知正则化,它在扩散过程中逐步促进复杂细节的保留和对比度的增强。这种整合减轻了扩散过程中噪声和伪影的负面影响,从而实现了更精确、更灵活的增强效果。为进一步促进困难区域的学习,我们引入了一种不确定性引导的正则化技术,该技术巧妙地对图像中最极端区域施加约束放松。实验评估表明,所提出的扩散模型框架结合排序信息正则化,在低光照增强任务中取得了卓越性能。与现有最先进方法相比,其在图像质量、噪声抑制和对比度增强方面均取得了显著进步。我们相信,这一创新方法将推动低光照图像处理领域的进一步探索与发展,并对扩散模型的其他应用具有潜在影响。代码已开源在 https://github.com/jinnh/GSAD。