# A Mamba U-Net Model for Reconstruction of Extremely Dark RGGB Images

## TL;DR

This paper introduces a Mamba U-Net model to efficiently enhance extremely dark RGGB images using a state-space model for better quality and lower resource use.

## Contribution

The novel contribution is the development of a Mamba U-Net model using a state-space model for efficient and effective restoration of extremely dark RGGB images.

## Key findings

- The proposed Mamba U-Net model significantly reduces resource consumption compared to existing methods.
- The model achieves improved PSNR and SSIM metrics for dark RGGB image restoration.
- The method was trained and validated using the see-in-the-dark (SID) dataset.

## Abstract

Currently, most images captured by high-pixel devices such as mobile phones, camcorders, and drones are in RGGB format. However, image quality in extremely dark scenes often needs improvement. Traditional methods for processing these dark RGGB images typically rely on end-to-end U-Net networks and their enhancement techniques, which require substantial resources and processing time. To tackle this issue, we first converted RGGB images into RGB three-channel images by subtracting the black level and applying linear interpolation. During the training stage, we leveraged the computational efficiency of the state-space model (SSM) and developed a Mamba U-Net end-to-end model to enhance the restoration of extremely dark RGGB images. We utilized the see-in-the-dark (SID) dataset for training, assessing the effectiveness of our approach. Experimental results indicate that our method significantly reduces resource consumption compared to existing single-step training and prior multi-step training techniques, while achieving improved peak signal-to-noise ratio (PSNR) and structural similarity (SSIM) outcomes.