Underwater Image Restoration via Polymorphic Large Kernel CNNs

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Guo, Xiaojiao, Dong, Yihang, Chen, Xuhang, Chen, Weiwen, Li, Zimeng, Zheng, FuChen, Pun, Chi-Man
Format: Preprint
Published: 2024
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866913625360302080
author Guo, Xiaojiao
Dong, Yihang
Chen, Xuhang
Chen, Weiwen
Li, Zimeng
Zheng, FuChen
Pun, Chi-Man
author_facet Guo, Xiaojiao
Dong, Yihang
Chen, Xuhang
Chen, Weiwen
Li, Zimeng
Zheng, FuChen
Pun, Chi-Man
contents Underwater Image Restoration (UIR) remains a challenging task in computer vision due to the complex degradation of images in underwater environments. While recent approaches have leveraged various deep learning techniques, including Transformers and complex, parameter-heavy models to achieve significant improvements in restoration effects, we demonstrate that pure CNN architectures with lightweight parameters can achieve comparable results. In this paper, we introduce UIR-PolyKernel, a novel method for underwater image restoration that leverages Polymorphic Large Kernel CNNs. Our approach uniquely combines large kernel convolutions of diverse sizes and shapes to effectively capture long-range dependencies within underwater imagery. Additionally, we introduce a Hybrid Domain Attention module that integrates frequency and spatial domain attention mechanisms to enhance feature importance. By leveraging the frequency domain, we can capture hidden features that may not be perceptible to humans but are crucial for identifying patterns in both underwater and on-air images. This approach enhances the generalization and robustness of our UIR model. Extensive experiments on benchmark datasets demonstrate that UIR-PolyKernel achieves state-of-the-art performance in underwater image restoration tasks, both quantitatively and qualitatively. Our results show that well-designed pure CNN architectures can effectively compete with more complex models, offering a balance between performance and computational efficiency. This work provides new insights into the potential of CNN-based approaches for challenging image restoration tasks in underwater environments. The code is available at \href{https://github.com/CXH-Research/UIR-PolyKernel}{https://github.com/CXH-Research/UIR-PolyKernel}.
format Preprint
id arxiv_https___arxiv_org_abs_2412_18459
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Underwater Image Restoration via Polymorphic Large Kernel CNNs
Guo, Xiaojiao
Dong, Yihang
Chen, Xuhang
Chen, Weiwen
Li, Zimeng
Zheng, FuChen
Pun, Chi-Man
Computer Vision and Pattern Recognition
Image and Video Processing
Underwater Image Restoration (UIR) remains a challenging task in computer vision due to the complex degradation of images in underwater environments. While recent approaches have leveraged various deep learning techniques, including Transformers and complex, parameter-heavy models to achieve significant improvements in restoration effects, we demonstrate that pure CNN architectures with lightweight parameters can achieve comparable results. In this paper, we introduce UIR-PolyKernel, a novel method for underwater image restoration that leverages Polymorphic Large Kernel CNNs. Our approach uniquely combines large kernel convolutions of diverse sizes and shapes to effectively capture long-range dependencies within underwater imagery. Additionally, we introduce a Hybrid Domain Attention module that integrates frequency and spatial domain attention mechanisms to enhance feature importance. By leveraging the frequency domain, we can capture hidden features that may not be perceptible to humans but are crucial for identifying patterns in both underwater and on-air images. This approach enhances the generalization and robustness of our UIR model. Extensive experiments on benchmark datasets demonstrate that UIR-PolyKernel achieves state-of-the-art performance in underwater image restoration tasks, both quantitatively and qualitatively. Our results show that well-designed pure CNN architectures can effectively compete with more complex models, offering a balance between performance and computational efficiency. This work provides new insights into the potential of CNN-based approaches for challenging image restoration tasks in underwater environments. The code is available at \href{https://github.com/CXH-Research/UIR-PolyKernel}{https://github.com/CXH-Research/UIR-PolyKernel}.
title Underwater Image Restoration via Polymorphic Large Kernel CNNs
topic Computer Vision and Pattern Recognition
Image and Video Processing
url https://arxiv.org/abs/2412.18459