Residential College | false |
Status | 已發表Published |
UIE-UnFold: Deep Unfolding Network with Color Priors and Vision Transformer for Underwater Image Enhancement | |
Lei, Yingtie1; Yu, Jia2; Dong, Yihang3; Gong, Changwei3; Zhou, Ziyang2; Pun, Chi Man1![]() ![]() | |
2024 | |
Conference Name | 11th IEEE International Conference on Data Science and Advanced Analytics, DSAA 2024 |
Source Publication | 2024 IEEE 11th International Conference on Data Science and Advanced Analytics, DSAA 2024
![]() |
Conference Date | 06-10 October 2024 |
Conference Place | San Diego, CA |
Country | USA |
Publisher | Institute of Electrical and Electronics Engineers Inc. |
Abstract | Underwater image enhancement (UIE) plays a crucial role in various marine applications, but it remains challenging due to the complex underwater environment. Current learning-based approaches frequently lack explicit incorporation of prior knowledge about the physical processes involved in underwater image formation, resulting in limited optimization despite their impressive enhancement results. This paper proposes a novel deep unfolding network (DUN) for UIE that integrates color priors and inter-stage feature transformation to improve enhancement performance. The proposed DUN model combines the iterative optimization and reliability of model-based methods with the flexibility and representational power of deep learning, offering a more explainable and stable solution compared to existing learning-based UIE approaches. The proposed model consists of three key components: a Color Prior Guidance Block (CPGB) that establishes a mapping between color channels of degraded and original images, a Nonlinear Activation Gradient Descent Module (NAGDM) that simulates the underwater image degradation process, and an Inter Stage Feature Transformer (ISF-Former) that facilitates feature exchange between different network stages. By explicitly incorporating color priors and modeling the physical characteristics of underwater image for-mation, the proposed DUN model achieves more accurate and reliable enhancement results. Extensive experiments on multiple underwater image datasets demonstrate the superiority of the proposed model over state-of-the-art methods in both quantitative and qualitative evaluations. The proposed DUN-based approach offers a promising solution for UIE, enabling more accurate and reliable scientific analysis in marine research. The code is available at https://github.com/CXH-Research/UIE-UnFold. |
Keyword | Deep Unfolding Network Underwater Image Enhancement Vision Transformer |
DOI | 10.1109/DSAA61799.2024.10722842 |
URL | View the original |
Language | 英語English |
Scopus ID | 2-s2.0-85209369091 |
Fulltext Access | |
Citation statistics | |
Document Type | Conference paper |
Collection | Faculty of Science and Technology DEPARTMENT OF COMPUTER AND INFORMATION SCIENCE |
Corresponding Author | Pun, Chi Man |
Affiliation | 1.University of Macau, Chinese Academy of Sciences, Macao 2.Huizhou University, Chinese Academy of Sciences, China 3.Shenzhen Institute of Advanced Technology, Chinese Academy of Sciences, China |
First Author Affilication | University of Macau |
Corresponding Author Affilication | University of Macau |
Recommended Citation GB/T 7714 | Lei, Yingtie,Yu, Jia,Dong, Yihang,et al. UIE-UnFold: Deep Unfolding Network with Color Priors and Vision Transformer for Underwater Image Enhancement[C]:Institute of Electrical and Electronics Engineers Inc., 2024. |
APA | Lei, Yingtie., Yu, Jia., Dong, Yihang., Gong, Changwei., Zhou, Ziyang., & Pun, Chi Man (2024). UIE-UnFold: Deep Unfolding Network with Color Priors and Vision Transformer for Underwater Image Enhancement. 2024 IEEE 11th International Conference on Data Science and Advanced Analytics, DSAA 2024. |
Files in This Item: | There are no files associated with this item. |
Items in the repository are protected by copyright, with all rights reserved, unless otherwise indicated.
Edit Comment