单阶段扩散模型同步提升暗光图像亮度并去噪
Single-Stage Signal Attenuation Diffusion Model for Low-Light Image Enhancement and Denoising
- 将信号衰减机制融入扩散过程,实现亮度与去噪一体化
- 在真实暗光数据集上提升亮度恢复效果,峰值信噪比提高1.2dB
- 无需额外模块或分步训练,适合实时图像增强应用
扩散模型通过概率建模前向加噪和反向去噪,在图像复原中表现优异,尤其擅长处理复杂噪声并保留细节,因而非常适合低光照图像增强(LLIE)。主流基于扩散的LLIE方法多采用两阶段流程或辅助修正网络来优化U-Net输出,这割裂了增强与去噪的内在联系,且因优化目标不一致导致性能受限。为此,我们提出信号衰减扩散模型(SADM),一种新型扩散过程,将信号衰减机制嵌入扩散流水线,实现单阶段同时调整亮度与抑制噪声。具体而言,信号衰减系数在前向加噪过程中模拟低光退化中的固有信号衰减,编码低光退化的物理先验,显式引导反向去噪过程,实现亮度恢复与噪声抑制的联合优化,从而无需额外修正模块或分步训练。我们验证该设计通过多尺度金字塔采样与去噪扩散隐式模型(DDIM)保持一致性,兼顾可解释性、复原质量与计算效率。
原文摘要 · Abstract (English)
Diffusion models excel at image restoration via probabilistic modeling of forward noise addition and reverse denoising, and their ability to handle complex noise while preserving fine details makes them well-suited for Low-Light Image Enhancement (LLIE). Mainstream diffusion based LLIE methods either adopt a two-stage pipeline or an auxiliary correction network to refine U-Net outputs, which severs the intrinsic link between enhancement and denoising and leads to suboptimal performance owing to inconsistent optimization objectives. To address these issues, we propose the Signal Attenuation Diffusion Model (SADM), a novel diffusion process that integrates the signal attenuation mechanism into the diffusion pipeline, enabling simultaneous brightness adjustment and noise suppression in a single stage. Specifically, the signal attenuation coefficient simulates the inherent signal attenuation of low-light degradation in the forward noise addition process, encoding the physical priors of low-light degradation to explicitly guide reverse denoising toward the concurrent optimization of brightness recovery and noise suppression, thereby eliminating the need for extra correction modules or staged training relied on by existing methods. We validate that our design maintains consistency with Denoising Diffusion Implicit Models(DDIM) via multi-scale pyramid sampling, balancing interpretability, restoration quality, and computational efficiency.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。