18 March 2024 UKD-Net: efficient image enhancement with knowledge distillation
Xiaoyan Zhao, Xiaowen Cai, Ying Xue, Yipeng Liao, Liqun Lin, Tiesong Zhao
Author Affiliations +
Abstract

Underwater images often suffer from color distortion, blurred details, and low contrast. Therefore, more researchers are exploring underwater image enhancement (UIE) methods. However, UIE models based on deep learning suffer from high computational complexity, thus limiting their integration into underwater devices. In this work, we propose a lightweight UIE network based on knowledge distillation (UKD-Net), which includes a teacher network (T-Net) and a student network (S-Net). T-Net uses our designed multi-scale fusion block and parallel attention block to achieve excellent performance. We utilize knowledge distillation technology to transfer the rich knowledge of the T-Net onto a deployable S-Net. Additionally, S-Net employs blueprint separable convolutions and multistage distillation block to reduce parameter count and computational complexity. Results demonstrate that our UKD-Net successfully achieves a lightweight model design while maintaining superior enhanced performance.

© 2024 SPIE and IS&T
Xiaoyan Zhao, Xiaowen Cai, Ying Xue, Yipeng Liao, Liqun Lin, and Tiesong Zhao "UKD-Net: efficient image enhancement with knowledge distillation," Journal of Electronic Imaging 33(2), 023024 (18 March 2024). https://doi.org/10.1117/1.JEI.33.2.023024
Received: 11 August 2023; Accepted: 28 February 2024; Published: 18 March 2024
Advertisement
Advertisement
RIGHTS & PERMISSIONS
Get copyright permission  Get copyright permission on Copyright Marketplace
KEYWORDS
Image enhancement

Image quality

Convolution

Machine learning

Education and training

Image processing

Ablation

Back to Top