考虑图像增强中的不确定性,提升暗光下细节和色彩一致性。
Uncertainty-Aware Spatial Color Correlation for Low-Light Image Enhancement
- 引入基于熵的不确定性建模,优化噪声与梯度问题。
- 在多个数据集上达到当前最优效果,泛化性强。
- 适合关注暗光图像质量与可靠性研究的开发者。
现有低光图像增强方法多聚焦于架构创新,常忽视特征表示中的内在不确定性,尤其在极端黑暗条件下,退化的梯度与噪声主导严重影响模型可靠性与因果推理能力。为此,我们提出U2CLLIE框架,融合不确定性感知增强与空间-颜色因果关联建模。从熵基不确定性视角出发,设计两个关键组件:(1) 不确定性感知双域去噪(UaD)模块,利用高斯引导自适应频域特征增强(G2AF)抑制频域噪声并优化熵驱动表示,有效缓解梯度消失与噪声主导问题,提升空间纹理提取与结构精炼能力;(2) 分层因果感知框架,亮度增强网络(LEN)首先对暗区进行粗粒度亮度增强,随后在编码器-解码器阶段,邻域相关状态空间(NeCo)与自适应空间-颜色校准(AsC)两个非对称因果关联建模模块协同构建分层因果约束,重构并强化特征空间中的邻域结构与色彩一致性。大量实验表明,U2CLLIE在多个基准数据集上均取得领先性能,展现出强鲁棒性与跨场景泛化能力。
原文摘要 · Abstract (English)
Most existing low-light image enhancement approaches primarily focus on architectural innovations, while often overlooking the intrinsic uncertainty within feature representations particularly under extremely dark conditions where degraded gradient and noise dominance severely impair model reliability and causal reasoning. To address these issues, we propose U2CLLIE, a novel framework that integrates uncertainty-aware enhancement and spatial-color causal correlation modeling. From the perspective of entropy-based uncertainty, our framework introduces two key components: (1) An Uncertainty-Aware Dual-domain Denoise (UaD) Module, which leverages Gaussian-Guided Adaptive Frequency Domain Feature Enhancement (G2AF) to suppress frequency-domain noise and optimize entropy-driven representations. This module enhances spatial texture extraction and frequency-domain noise suppression/structure refinement, effectively mitigating gradient vanishing and noise dominance. (2) A hierarchical causality-aware framework, where a Luminance Enhancement Network (LEN) first performs coarse brightness enhancement on dark regions. Then, during the encoder-decoder phase, two asymmetric causal correlation modeling modules Neighborhood Correlation State Space (NeCo) and Adaptive Spatial-Color Calibration (AsC) collaboratively construct hierarchical causal constraints. These modules reconstruct and reinforce neighborhood structure and color consistency in the feature space. Extensive experiments demonstrate that U2CLLIE achieves state-of-the-art performance across multiple benchmark datasets, exhibiting robust performance and strong generalization across various scenes.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。