Frequency-guided Dual Sparse Self-Attention Algorithm for Single Image Deraining

被引:0
|
作者
Wen Y.-B. [1 ]
Gao T. [1 ]
Chen T. [1 ]
Zhang Q.-X. [1 ]
机构
[1] School of Information Engineering, Chang’an University, Shaanxi, Xi’an
来源
基金
中国国家自然科学基金;
关键词
computer vision; frequency-guided learning; image deraining; sparse attention; spatial shift; Transformer;
D O I
10.12263/DZXB.20221420
中图分类号
学科分类号
摘要
Existing Transformer-based algorithms for single image deraining achieve state-of-the-art performance but leading to reasonable computational loads while failing to effectively process real-world rainy images. To this end, we propose a frequency-guided dual sparse self-attention Transformer for single image deraining (FDSATFormer). Initially, our proposed method utilizes the spatial sparse factor and the channel reduction factor to extract accurate global information and significantly decreases the amount of computation. Furthermore, we present dual sparse self-attention feature learning network (DSFL) to deal with the problem that Transformer is difficult to represent self-attention on high-resolution feature maps. Meanwhile, by exploring the spectral changes of rainy image before and after removing rain streaks, we develop a frequency-guided feature enhancement module (FFE), which exploits the global information from the frequency domain to guide the accurate learning of spatial features in network encoders. In addition, the encoder and decoder of most existing methods follow the similar principles, resulting in almost double computational burden. To handle with this issue, we propose a hierarchical feature decoding and reconstructing network (HFDR), which uses non-parametric spatial neighborhood shift (SNS) to construct the feature decoding network and achieves fine results while further reducing the overall computing burden. Experimental results show that, our method improves the average peak signal noise ratio by 3.13 dB and 0.12 dB, and achieves performance gains of 1.39% and 1.07% in average structure similarity over the state-of-the-art Uformer and Restormer. © 2023 Chinese Institute of Electronics. All rights reserved.
引用
收藏
页码:2812 / 2820
页数:8
相关论文
共 34 条
  • [31] LI Y, TAN R T, GUO X J, Et al., Rain streak removal using layer priors, 2016 IEEE Conference on Computer Vision and Pattern Recognition (CVPR), pp. 2736-2744, (2016)
  • [32] FU X Y, LIANG B R, HUANG Y, Et al., Lightweight pyramid networks for image deraining, IEEE Transactions on Neural Networks and Learning Systems, 31, 6, pp. 1794-1807, (2020)
  • [33] KINGMA D P, BA J., Adam: A method for stochastic optimization
  • [34] ZHANG H, CISSE M, DAUPHIN Y N, Et al., MixUp: Beyond empirical risk minimization