Uncertainty-Aware Spatial Color Correlation for Low-Light Image Enhancement
2508.04176v1
cs.CV
2025-08-09
Авторы:
Jin Kuang, Dong Liu, Yukuang Zhang, Shengsheng Wang
Резюме на русском
Многие подходы к улучшению изображений в условиях низкого освещения фокусируются на развитии архитектуры, но часто игнорируют встроенную неопределенность в представлениях признаков, особенно при использовании экстремальных темных условий. Наличие шума и падения производных сильно снижает надежность модели и возможность выполнения логических выводов. Мы предлагаем U2CLLIE — новую модель, объединяющую нейросетевое улучшение изображений с моделированием пространственно-цветовых зависимостей на основе причинности. Модель включает два ключевых модуля: (1) нейросеть Uncertainty-Aware Dual-domain Denoise (UaD), которая устраняет шум в двух доменах (спектральном и частотном) и оптимизирует сигналы с применением G2AF-функции; (2) цепная система с причинным рассуждением, включающую LEN, NeCo и AsC, которые восстанавливают структуру и цветовую консистентность. Эксперименты показывают, что U2CLLIE превосходит современные подходы, показывая высокую точность и генерализуемость на нескольких датасетах.
Abstract
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.
Ссылки и действия
Дополнительные ресурсы: