Enhancing Underwater Images via Adaptive Semantic-aware Codebook Learning

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: Lin, Bosen, Gao, Feng, Yu, Yanwei, Dong, Junyu, Du, Qian
Formato: Preprint
Publicado: 2026
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866912896991100928
author Lin, Bosen
Gao, Feng
Yu, Yanwei
Dong, Junyu
Du, Qian
author_facet Lin, Bosen
Gao, Feng
Yu, Yanwei
Dong, Junyu
Du, Qian
contents Underwater Image Enhancement (UIE) is an ill-posed problem where natural clean references are not available, and the degradation levels vary significantly across semantic regions. Existing UIE methods treat images with a single global model and ignore the inconsistent degradation of different scene components. This oversight leads to significant color distortions and loss of fine details in heterogeneous underwater scenes, especially where degradation varies significantly across different image regions. Therefore, we propose SUCode (Semantic-aware Underwater Codebook Network), which achieves adaptive UIE from semantic-aware discrete codebook representation. Compared with one-shot codebook-based methods, SUCode exploits semantic-aware, pixel-level codebook representation tailored to heterogeneous underwater degradation. A three-stage training paradigm is employed to represent raw underwater image features to avoid pseudo ground-truth contamination. Gated Channel Attention Module (GCAM) and Frequency-Aware Feature Fusion (FAFF) jointly integrate channel and frequency cues for faithful color restoration and texture recovery. Extensive experiments on multiple benchmarks demonstrate that SUCode achieves state-of-the-art performance, outperforming recent UIE methods on both reference and no-reference metrics. The code will be made public available at https://github.com/oucailab/SUCode.
format Preprint
id arxiv_https___arxiv_org_abs_2602_10586
institution arXiv
publishDate 2026
record_format arxiv
spellingShingle Enhancing Underwater Images via Adaptive Semantic-aware Codebook Learning
Lin, Bosen
Gao, Feng
Yu, Yanwei
Dong, Junyu
Du, Qian
Computer Vision and Pattern Recognition
Image and Video Processing
Underwater Image Enhancement (UIE) is an ill-posed problem where natural clean references are not available, and the degradation levels vary significantly across semantic regions. Existing UIE methods treat images with a single global model and ignore the inconsistent degradation of different scene components. This oversight leads to significant color distortions and loss of fine details in heterogeneous underwater scenes, especially where degradation varies significantly across different image regions. Therefore, we propose SUCode (Semantic-aware Underwater Codebook Network), which achieves adaptive UIE from semantic-aware discrete codebook representation. Compared with one-shot codebook-based methods, SUCode exploits semantic-aware, pixel-level codebook representation tailored to heterogeneous underwater degradation. A three-stage training paradigm is employed to represent raw underwater image features to avoid pseudo ground-truth contamination. Gated Channel Attention Module (GCAM) and Frequency-Aware Feature Fusion (FAFF) jointly integrate channel and frequency cues for faithful color restoration and texture recovery. Extensive experiments on multiple benchmarks demonstrate that SUCode achieves state-of-the-art performance, outperforming recent UIE methods on both reference and no-reference metrics. The code will be made public available at https://github.com/oucailab/SUCode.
title Enhancing Underwater Images via Adaptive Semantic-aware Codebook Learning
topic Computer Vision and Pattern Recognition
Image and Video Processing
url https://arxiv.org/abs/2602.10586