Computer Engineering and Applications ›› 2024, Vol. 60 ›› Issue (19): 250-258.DOI: 10.3778/j.issn.1002-8331.2307-0100

• Graphics and Image Processing • Previous Articles     Next Articles

Lightweight Transformer Deraining Network Based on Reduced Biquaternion Attention Mechanism

XIONG Gonghe, CHEN Feilong, SUN Chengli, GUO Qiaosheng   

  1. 1. School of Information and Engineering, Nanchang Hangkong University, Nanchang 330063, China
    2. Key Laboratory of Jiangxi Province for Image Processing and Pattern Recognition, Nanchang Hangkong University, Nanchang 330063, China
    3. Chaoyang Gevotai (Xin Feng) Technology Co., Ltd., Ganzhou, Jiangxi 341600, China
  • Online:2024-10-01 Published:2024-09-30

基于退化四元数注意力机制的轻量化Transformer去雨网络

熊贡鹤,陈飞龙,孙成立,郭桥生   

  1. 1. 南昌航空大学  信息工程学院,南昌  330063
    2. 南昌航空大学  江西省图像处理与模式识别重点实验室,南昌  330063
    3. 朝阳聚声泰(信丰)科技有限公司,江西  赣州  341600

Abstract: The existing mainstream image deraining methods focus on improving the deraining performance but ignore the problem of excessive network computation overhead. A few lightweight network studies are restricted to changing the network’s structure to reduce computation. The capability of the reduced biquaternion (RQ) to obtain more a priori information is used to propose a reduced biquaternion image deraining network as a solution to the above issue. The network’s primary feature extraction module is the reduced biquaternion Swin-Transformer block (RQSTB). The RQSTB incorporates the reduced biquaternion Transformer block, which utilizes the reduced biquaternion multi-headed attention mechanism to extract global feature information. Additionally, the reduced biquaternion multi-scale convolution module is interleaved to capture local multi-scale feature information. This combination compensates for the inherent lack of CNN’s inductive biases that are absent in Transformer. It is experimentally demonstrated that this method achieves advanced levels of deraining performance, outperforming most existing image rain removal methods in terms of network parameters and computational complexity and demonstrating significant results in terms of both quantitative and qualitative metrics.

Key words: image deraining, reduced biquaternion network, Transformer, lightweight

摘要: 现有主流图像去雨方法专注于提升去雨性能,而忽略了网络计算开销过大的问题。少数轻量化网络的研究只局限于修改网络结构来简化网络计算。针对上述问题,利用退化四元数可以获得更多图像先验信息的特性提出了一个基于退化四元数图像去雨网络。网络使用退化四元数Swin-Transformer块(reduced biquaternion Swin-Transformer block,RQSTB)作为主要特征提取模块。其中设计了使用基于退化四元数多头注意力机制的Transformer块提取全局特征信息,同时穿插使用退化四元数多尺度卷积模块提取局部多尺度特征信息,用以弥补Transformer缺乏卷积神经网络自带的一些归纳偏置的缺陷。经实验证明,该方法在网络参数和计算复杂度方面都优于很多现有的图像去雨方法,并且在去雨性能方面也达到了先进的水平,无论是从定量还是定性的指标来看,都展现了显著的效果。

关键词: 图像去雨, 退化四元数网络, Transformer, 轻量化