PAUNet: A Lightweight Network for Real-Time Underwater Image Enhancement.
- 1Ocean College, Zhejiang University, Zhoushan 316000, China.
- 2Donghai Laboratory, Zhoushan 316000, China.
Abstract
Underwater images are degraded by wavelength-dependent absorption and scattering, while enhancement on resource-constrained platforms requires a small model and low inference latency. We propose PAUNet, a lightweight encoder-decoder network combining partial axial depthwise convolution (PADW), multiscale axial spatial adaptive attention (MASA), and attention-guided skip connections. PADW filters half of the feature channels and preserves the remainder through an identity branch. MASA combines three axial scales using spatial branch attention. A joint objective constrains reconstruction, structure, perceptual features, and color. PAUNet achieves PSNR values of 21.508, 28.696, and 28.632 dB on UIEB, EUVP, and LSUI, respectively, with SSIM values of 0.751, 0.843, and 0.895. Among the compared methods, PAUNet has the highest PSNR on EUVP and LSUI, ranks second in UIEB PSNR, and achieves the highest LSUI SSIM. PAUNet contains 64,495 parameters, requires 0.493 GFLOPs, and runs at 288.95 frames per second on an NVIDIA RTX 3090. On an NVIDIA Jetson Xavier NX, TensorRT FP16 inference reaches 74.55 frames per second at 256 × 256 resolution and batch size one. The comparisons and ablations evaluate this compact architecture for underwater image enhancement.
Related
- Rethinking underwater image enhancement via a hybrid Mamba–transformer network
- Physics aware joint low light and color correction for underwater image enhancement
- DUIE: Depth-Aware Underwater Image Enhancement via Spatial Prior Modeling
- A Lightweight Channel-Frequency Joint Attention Network for Underwater Ship-Radiated Noise Separation.
- Phys-Mamba: Physics-informed selective state-space fusion network for high-fidelity underwater image restoration
BibTeX
@article{yin2026paunet,
title = {{PAUNet: A Lightweight Network for Real-Time Underwater Image Enhancement.}},
author = {Kelong Yin and Dongjin Jiang and Hongyang Xu and Yongqing Zeng and Mingyu Qiu and Xiaoping Wang},
journal = {Sensors (Basel, Switzerland)},
year = {2026},
month = oct,
doi = {10.3390/s26206546},
url = {https://europepmc.org/article/MED/42944407},
note = {CC BY},
}
RIS
TY - JOUR
TI - PAUNet: A Lightweight Network for Real-Time Underwater Image Enhancement.
AU - Kelong Yin
AU - Dongjin Jiang
AU - Hongyang Xu
AU - Yongqing Zeng
AU - Mingyu Qiu
AU - Xiaoping Wang
PY - 2026
DA - 2026/10/15
JO - Sensors (Basel, Switzerland)
DO - 10.3390/s26206546
UR - https://europepmc.org/article/MED/42944407
AB - Underwater images are degraded by wavelength-dependent absorption and scattering, while enhancement on resource-constrained platforms requires a small model and low inference latency. We propose PAUNet, a lightweight encoder-decoder network combining partial axial depthwise convolution (PADW), multiscale axial spatial adaptive attention (MASA), and attention-guided skip connections. PADW filters half of the feature channels and preserves the remainder through an identity branch. MASA combines three axial scales using spatial branch attention. A joint objective constrains reconstruction, structure, perceptual features, and color. PAUNet achieves PSNR values of 21.508, 28.696, and 28.632 dB on UIEB, EUVP, and LSUI, respectively, with SSIM values of 0.751, 0.843, and 0.895. Among the compared methods, PAUNet has the highest PSNR on EUVP and LSUI, ranks second in UIEB PSNR, and achieves the highest LSUI SSIM. PAUNet contains 64,495 parameters, requires 0.493 GFLOPs, and runs at 288.95 frames per second on an NVIDIA RTX 3090. On an NVIDIA Jetson Xavier NX, TensorRT FP16 inference reaches 74.55 frames per second at 256 × 256 resolution and batch size one. The comparisons and ablations evaluate this compact architecture for underwater image enhancement.
SN - 1424-8220
ER -