MambaLLIE: Implicit Retinex-Aware Low Light Enhancement with Global-then-Local State Space

Read original: arXiv:2405.16105 - Published 5/28/2024 by Jiangwei Weng, Zhiqiang Yan, Ying Tai, Jianjun Qian, Jian Yang, Jun Li
Total Score

0

MambaLLIE: Implicit Retinex-Aware Low Light Enhancement with Global-then-Local State Space

Sign in to get full access

or

If you already have an account, we'll log you in

Overview

  • The paper introduces MambaLLIE, a novel low-light image enhancement method that combines global and local enhancement techniques.
  • It leverages the Retinex theory, which models the human visual system's ability to perceive constant object colors despite varying illumination.
  • MambaLLIE uses a global-then-local state space model to first enhance the overall brightness and contrast, then apply localized adjustments to preserve details and prevent over-enhancement.

Plain English Explanation

The MambaLLIE paper presents a new way to make dark or low-light images look brighter and clearer. It's based on the Retinex theory, which explains how our eyes can see the true colors of objects even when the lighting changes.

MambaLLIE works in two steps. First, it globally enhances the overall brightness and contrast of the image. This brings out the general details and makes the whole picture look better. Then, it applies more localized adjustments to specific areas. This helps preserve important details and prevent over-brightening in certain spots.

The researchers use a "global-then-local state space model" to achieve this. This means they have a multi-stage process that first looks at the image as a whole, then zooms in on smaller regions. This allows MambaLLIE to balance the enhancements across the entire image for natural-looking results.

Technical Explanation

The MambaLLIE approach builds on the Retinex theory, which models the human visual system's ability to perceive constant object colors despite varying illumination. Retinex-based methods have shown promise for low-light image enhancement, but often struggle with preserving local details.

To address this, the authors propose a global-then-local state space model for low-light image enhancement. First, a global enhancement module adjusts the overall brightness and contrast of the image. Then, a local enhancement module applies targeted adjustments to specific regions to preserve details and prevent over-enhancement.

This two-stage process is implemented using a deep neural network architecture. The global module uses a convolutional encoder-decoder structure, while the local module employs a novel state space model to capture localized adjustments. The outputs of the two modules are combined to produce the final enhanced image.

Experiments on benchmark low-light image datasets demonstrate the effectiveness of the MambaLLIE approach, with improvements over state-of-the-art methods in terms of both objective metrics and subjective visual quality. The authors also showcase the flexibility of the MambaLLIE framework by extending it to other image-to-image tasks, such as multi-modal medical image synthesis and 3D point cloud enhancement.

Critical Analysis

The MambaLLIE paper presents a well-designed and effective approach for low-light image enhancement. The authors' use of a global-then-local state space model is a clever way to balance overall brightness and contrast adjustments with targeted local enhancements.

However, the paper does not address some potential limitations of the method. For example, the global enhancement module may struggle with scenes that have highly uneven lighting, and the local adjustments may introduce artifacts in some cases. Additionally, the computational complexity of the two-stage model could be a concern for real-time applications.

Further research could explore ways to address these limitations, such as incorporating adaptive weighting between the global and local modules or developing more efficient neural network architectures. The authors' extension of the MambaLLIE framework to other tasks is promising, and it would be interesting to see how the method performs on a wider range of image-to-image problems.

Conclusion

The MambaLLIE paper presents a novel low-light image enhancement method that combines global and local adjustment techniques. By leveraging the Retinex theory and a global-then-local state space model, the authors have developed an effective approach that can improve the brightness, contrast, and detail preservation of low-light images.

The flexible and extensible nature of the MambaLLIE framework suggests that it could have broader applications beyond low-light enhancement, such as in medical imaging and 3D point cloud processing. As the field of image processing continues to advance, contributions like MambaLLIE that combine theoretical insights with practical innovations will be crucial for developing robust and versatile visual enhancement solutions.



This summary was produced with help from an AI and may contain inaccuracies - check out the links to read the original source documents!

Follow @aimodelsfyi on 𝕏 →

Related Papers

MambaLLIE: Implicit Retinex-Aware Low Light Enhancement with Global-then-Local State Space
Total Score

0

MambaLLIE: Implicit Retinex-Aware Low Light Enhancement with Global-then-Local State Space

Jiangwei Weng, Zhiqiang Yan, Ying Tai, Jianjun Qian, Jian Yang, Jun Li

Recent advances in low light image enhancement have been dominated by Retinex-based learning framework, leveraging convolutional neural networks (CNNs) and Transformers. However, the vanilla Retinex theory primarily addresses global illumination degradation and neglects local issues such as noise and blur in dark conditions. Moreover, CNNs and Transformers struggle to capture global degradation due to their limited receptive fields. While state space models (SSMs) have shown promise in the long-sequence modeling, they face challenges in combining local invariants and global context in visual data. In this paper, we introduce MambaLLIE, an implicit Retinex-aware low light enhancer featuring a global-then-local state space design. We first propose a Local-Enhanced State Space Module (LESSM) that incorporates an augmented local bias within a 2D selective scan mechanism, enhancing the original SSMs by preserving local 2D dependency. Additionally, an Implicit Retinex-aware Selective Kernel module (IRSK) dynamically selects features using spatially-varying operations, adapting to varying inputs through an adaptive kernel selection process. Our Global-then-Local State Space Block (GLSSB) integrates LESSM and IRSK with LayerNorm as its core. This design enables MambaLLIE to achieve comprehensive global long-range modeling and flexible local feature aggregation. Extensive experiments demonstrate that MambaLLIE significantly outperforms state-of-the-art CNN and Transformer-based methods. Project Page: https://mamballie.github.io/anon/

Read more

5/28/2024

🖼️

Total Score

0

Retinexmamba: Retinex-based Mamba for Low-light Image Enhancement

Jiesong Bai, Yuhao Yin, Qiyuan He, Yuanxian Li, Xiaofeng Zhang

In the field of low-light image enhancement, both traditional Retinex methods and advanced deep learning techniques such as Retinexformer have shown distinct advantages and limitations. Traditional Retinex methods, designed to mimic the human eye's perception of brightness and color, decompose images into illumination and reflection components but struggle with noise management and detail preservation under low light conditions. Retinexformer enhances illumination estimation through traditional self-attention mechanisms, but faces challenges with insufficient interpretability and suboptimal enhancement effects. To overcome these limitations, this paper introduces the RetinexMamba architecture. RetinexMamba not only captures the physical intuitiveness of traditional Retinex methods but also integrates the deep learning framework of Retinexformer, leveraging the computational efficiency of State Space Models (SSMs) to enhance processing speed. This architecture features innovative illumination estimators and damage restorer mechanisms that maintain image quality during enhancement. Moreover, RetinexMamba replaces the IG-MSA (Illumination-Guided Multi-Head Attention) in Retinexformer with a Fused-Attention mechanism, improving the model's interpretability. Experimental evaluations on the LOL dataset show that RetinexMamba outperforms existing deep learning approaches based on Retinex theory in both quantitative and qualitative metrics, confirming its effectiveness and superiority in enhancing low-light images.

Read more

5/21/2024

LFMamba: Light Field Image Super-Resolution with State Space Model
Total Score

0

LFMamba: Light Field Image Super-Resolution with State Space Model

Wang xia, Yao Lu, Shunzhou Wang, Ziqi Wang, Peiqi Xia, Tianfei Zhou

Recent years have witnessed significant advancements in light field image super-resolution (LFSR) owing to the progress of modern neural networks. However, these methods often face challenges in capturing long-range dependencies (CNN-based) or encounter quadratic computational complexities (Transformer-based), which limit their performance. Recently, the State Space Model (SSM) with selective scanning mechanism (S6), exemplified by Mamba, has emerged as a superior alternative in various vision tasks compared to traditional CNN- and Transformer-based approaches, benefiting from its effective long-range sequence modeling capability and linear-time complexity. Therefore, integrating S6 into LFSR becomes compelling, especially considering the vast data volume of 4D light fields. However, the primary challenge lies in emph{designing an appropriate scanning method for 4D light fields that effectively models light field features}. To tackle this, we employ SSMs on the informative 2D slices of 4D LFs to fully explore spatial contextual information, complementary angular information, and structure information. To achieve this, we carefully devise a basic SSM block characterized by an efficient SS2D mechanism that facilitates more effective and efficient feature learning on these 2D slices. Based on the above two designs, we further introduce an SSM-based network for LFSR termed LFMamba. Experimental results on LF benchmarks demonstrate the superior performance of LFMamba. Furthermore, extensive ablation studies are conducted to validate the efficacy and generalization ability of our proposed method. We expect that our LFMamba shed light on effective representation learning of LFs with state space models.

Read more

6/19/2024

LLEMamba: Low-Light Enhancement via Relighting-Guided Mamba with Deep Unfolding Network
Total Score

0

LLEMamba: Low-Light Enhancement via Relighting-Guided Mamba with Deep Unfolding Network

Xuanqi Zhang, Haijin Zeng, Jinwang Pan, Qiangqiang Shen, Yongyong Chen

Transformer-based low-light enhancement methods have yielded promising performance by effectively capturing long-range dependencies in a global context. However, their elevated computational demand limits the scalability of multiple iterations in deep unfolding networks, and hence they have difficulty in flexibly balancing interpretability and distortion. To address this issue, we propose a novel Low-Light Enhancement method via relighting-guided Mamba with a deep unfolding network (LLEMamba), whose theoretical interpretability and fidelity are guaranteed by Retinex optimization and Mamba deep priors, respectively. Specifically, our LLEMamba first constructs a Retinex model with deep priors, embedding the iterative optimization process based on the Alternating Direction Method of Multipliers (ADMM) within a deep unfolding network. Unlike Transformer, to assist the deep unfolding framework with multiple iterations, the proposed LLEMamba introduces a novel Mamba architecture with lower computational complexity, which not only achieves light-dependent global visual context for dark images during reflectance relight but also optimizes to obtain more stable closed-form solutions. Experiments on the benchmarks show that LLEMamba achieves superior quantitative evaluations and lower distortion visual results compared to existing state-of-the-art methods.

Read more

6/4/2024