arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2307.07710cs.CVeess.IV

ExposureDiffusion:学习曝光以进行低光照图像增强

ExposureDiffusion: Learning to Expose for Low-light Image Enhancement

Nanyang Technological University(南洋理工大学);Peng Cheng Laboratory(鹏城实验室);The Hong Kong Polytechnic University(香港理工大学)

Yufei Wang, Yi Yu, Wenhan Yang, Lanqing Guo, Lap-Pui Chau, Alex C. Kot, Bihan Wen

AI总结:

本文提出ExposureDiffusion,将扩散模型与基于物理的曝光模型结合,并引入自适应残差层,在低光照增强中实现更快推理、更好性能及对未见放大倍率的更强泛化。

摘要:

以往基于原始图像的低光照图像增强方法主要依赖前馈神经网络来学习从低光照图像到正常曝光图像的确定性映射。然而,它们未能捕捉关键的分布信息,导致产生视觉上不理想的结果。本工作通过将扩散模型与基于物理的曝光模型无缝集成来解决这一问题。与必须执行高斯去噪的普通扩散模型不同,借助注入的基于物理的曝光模型,我们的恢复过程可以直接从含噪图像而非纯噪声开始。因此,与普通扩散模型相比,我们的方法获得了显著提升的性能,并减少了推理时间。为充分利用不同中间步骤的优势,我们进一步提出了一种自适应残差层;当中间结果已经充分曝光时,该层能有效排除迭代细化过程中的副作用。所提出的框架可同时适用于真实配对数据集、SOTA噪声模型以及不同的骨干网络。需要注意的是,该框架兼容真实配对数据集、真实/合成噪声模型以及不同的骨干网络。我们在多个公开基准上评估了所提方法,使用不同曝光模型和骨干网络均取得了有前景的结果和一致的改进。此外,所提方法在未见过的放大倍率上具有更好的泛化能力,并且在采用较少参数时,性能优于更大的前馈神经模型。

英文摘要:

Previous raw image-based low-light image enhancement methods predominantly relied on feed-forward neural networks to learn deterministic mappings from low-light to normally-exposed images. However, they failed to capture critical distribution information, leading to visually undesirable results. This work addresses the issue by seamlessly integrating a diffusion model with a physics-based exposure model. Different from a vanilla diffusion model that has to perform Gaussian denoising, with the injected physics-based exposure model, our restoration process can directly start from a noisy image instead of pure noise. As such, our method obtains significantly improved performance and reduced inference time compared with vanilla diffusion models. To make full use of the advantages of different intermediate steps, we further propose an adaptive residual layer that effectively screens out the side-effect in the iterative refinement when the intermediate results have been already well-exposed. The proposed framework can work with both real-paired datasets, SOTA noise models, and different backbone networks. Note that, the proposed framework is compatible with real-paired datasets, real/synthetic noise models, and different backbone networks. We evaluate the proposed method on various public benchmarks, achieving promising results with consistent improvements using different exposure models and backbones. Besides, the proposed method achieves better generalization capacity for unseen amplifying ratios and better performance than a larger feedforward neural model when few parameters are adopted.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑