UM  > Faculty of Science and Technology
Residential Collegefalse
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 Name11th IEEE International Conference on Data Science and Advanced Analytics, DSAA 2024
Source Publication2024 IEEE 11th International Conference on Data Science and Advanced Analytics, DSAA 2024
Conference Date06-10 October 2024
Conference PlaceSan Diego, CA
CountryUSA
PublisherInstitute 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.

KeywordDeep Unfolding Network Underwater Image Enhancement Vision Transformer
DOI10.1109/DSAA61799.2024.10722842
URLView the original
Language英語English
Scopus ID2-s2.0-85209369091
Fulltext Access
Citation statistics
Document TypeConference paper
CollectionFaculty of Science and Technology
DEPARTMENT OF COMPUTER AND INFORMATION SCIENCE
Corresponding AuthorPun, Chi Man
Affiliation1.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 AffilicationUniversity of Macau
Corresponding Author AffilicationUniversity 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.
Related Services
Recommend this item
Bookmark
Usage statistics
Export to Endnote
Google Scholar
Similar articles in Google Scholar
[Lei, Yingtie]'s Articles
[Yu, Jia]'s Articles
[Dong, Yihang]'s Articles
Baidu academic
Similar articles in Baidu academic
[Lei, Yingtie]'s Articles
[Yu, Jia]'s Articles
[Dong, Yihang]'s Articles
Bing Scholar
Similar articles in Bing Scholar
[Lei, Yingtie]'s Articles
[Yu, Jia]'s Articles
[Dong, Yihang]'s Articles
Terms of Use
No data!
Social Bookmark/Share
All comments (0)
No comment.
 

Items in the repository are protected by copyright, with all rights reserved, unless otherwise indicated.