A Deep Feature Fusion Underwater Image Enhancement Model Based on Perceptual Vision Swin Transformer

Read full text
Authors: Shasha Tian, Adisorn Sirikham, Jessada Konpang, Chuyang Wang
Cited by: 1
References: 45

Abstract

Underwater optical images are the primary carriers of underwater scene information, playing a crucial role in marine resource exploration, underwater environmental monitoring, and engineering inspection. However, wavelength-dependent absorption and scattering severely deteriorate underwater images, leading to reduced contrast, chromatic distortions, and loss of structural details. To address these issues, we propose a U-shaped underwater image enhancement framework that integrates Swin-Transformer blocks with lightweight attention and residual modules. A Dual-Window Multi-Head Self-Attention (DWMSA) in the bottleneck models long-range context while preserving fine local structure. A Global-Aware Attention Map (GAMP) adaptively re-weights channels and spatial locations to focus on severely degraded regions. A Feature-Augmentation Residual Network (FARN) stabilizes deep training and emphasizes texture and color fidelity. Trained with a combination of Charbonnier, perceptual, and edge losses, our method achieves state-of-the-art results in PSNR and SSIM, the lowest LPIPS, and improvements in UIQM and UCIQE on the UFO-120 and EUVP datasets, with average metrics of PSNR 29.5 dB, SSIM 0.94, LPIPS 0.17, UIQM 3.62, and UCIQE 0.59. Qualitative results show reduced color cast, restored contrast, and sharper details. Code, weights, and evaluation scripts will be released to support reproducibility.

Fields of Study & Subjects

Image Enhancement Techniques Advanced Image Fusion Techniques Generative Adversarial Networks and Image Synthesis

Publication details

Journal / source
Journal of Imaging
Publisher
MDPI AG
Volume
12
Issue
1
Pages
44–44
Publication date
14 January 2026
Language
EN