CELAT-DiffNet: channel-enhanced local attention transformer for underwater image enhancement based on diffusion models
摘要
This paper proposes a novel denoising diffusion probabilistic model for underwater image enhancement (UIE). It effectively addresses common issues in underwater images such as color distortion, low contrast, and detail blurring. We design an innovative Transformer denoising network that integrates a multidimensional collaborative attention (MCA) mechanism, which employs a three branch parallel structure to model feature interactions across channel, height, and width dimensions, further improving enhancement performance. Additionally, we propose the channel-enhanced local attention (CELA) module, which first performs global interaction at the channel level and then applies local neighborhood attention. This design ensures that local details are guided by global context, enabling effective integration of long-range dependencies and local semantics. To address the limitations of conventional feed-forward networks (FFNs) in spatial and channel interaction, we further design the channel-enhanced feed-forward network (CEFN). This module combines grouped convolution with a lightweight yet efficient channel attention mechanism, facilitating synergistic enhancement of spatial patterns and channel features. Extensive experimental results on diverse datasets (UIEB, LSUI, UFO, C60, U45) demonstrate that our method outperforms state-of-the-art approaches in both qualitative and quantitative evaluations.