Underwater Image Enhancement via Adaptive Group Attention-Based Multiscale Cascade Transformer

Abstract

The absorption and scattering caused by the underwater medium degrade the quality of underwater optical imaging, which limits the further development of underwater tasks. Recently, Transformer-based methods have shown the same excellent performance as Convolutional Neural Networks (CNNs) in various vision tasks, but the huge parameters of such networks hinder their application deployment. In this paper, we propose a novel adaptive group attention (AGA), which can dynamically select visually complementary channels based on the dependencies, reducing the number of further attention parameters. The AGA is applied in the Swin Transformer module and used to design an end-to-end underwater image enhancement network. The network also introduces the multiscale cascade module and the channel attention mechanism. This paper conducted ablation study, qualitative and quantitative comparisons on public datasets, and the results show that the application of AGA significantly compresses the model size while ensuring performance, and other application components have significant gain on the network. Compared with other advanced methods, the network in this paper has outstanding performance.