# DWMamba: a structure-aware adaptive state space network for image quality improvement

## TL;DR

DWMamba is a new image enhancement network that improves image quality in challenging conditions while being computationally efficient.

## Contribution

DWMamba introduces a structure-aware, adaptive state space network with novel modules for efficient and effective image quality improvement.

## Key findings

- DWMamba achieves superior performance in restoring degraded images under diverse lighting conditions.
- The Adaptive State Space Module (ASSM) effectively handles non-uniform degradations with linear complexity.
- The Structure-guided Residual Fusion (SGRF) module enhances detail restoration and low-light texture quality.

## Abstract

Overcoming visual degradation in challenging imaging scenarios is essential for accurate scene understanding. Although deep learning methods have integrated various perceptual capabilities and achieved remarkable progress, their high computational cost limits practical deployment under resource-constrained conditions. Moreover, when confronted with diverse degradation types, existing methods often fail to effectively model the inconsistent attenuation across color channels and spatial regions. To tackle these challenges, we propose DWMamba, a degradation-aware and weight-efficient Mamba network for image quality enhancement. Specifically, DWMamba introduces an Adaptive State Space Module (ASSM) that employs a dual-stream channel monitoring mechanism and a soft fusion strategy to capture global dependencies. With linear computational complexity, ASSM strengthens the models ability to address non-uniform degradations. In addition, by leveraging explicit edge priors and region partitioning as guidance, we design a Structure-guided Residual Fusion (SGRF) module to selectively fuse shallow and deep features, thereby restoring degraded details and enhancing low-light textures. Extensive experiments demonstrate that the proposed network delivers superior qualitative and quantitative performance, with strong generalization to diverse extreme lighting conditions. The code is available at https://github.com/WindySprint/DWMamba.