面向图像复原和增强的轻量级交叉门控Transformer

Lightweight Cross-Gating Transformer for Image Restoration and Enhancement

现有的图像复原和图像增强方法难以同时兼顾在多个子任务上的鲁棒性和维持较小的参数量与计算代价。针对这一问题,提出轻量级交叉门控转换算法(CGT)。一方面,总结了传统全局自注意力机制捕获全局依赖关系的局限性,将全局自注意力机制改进为跨层次交叉门控自注意力机制。同时提出轻量化的前馈神经网络,从而以极小的计算代价学习到跨层次局部依赖关系,在局部邻域内重构清晰特征。另一方面,针对传统方法对编码器和解码器平等地进行加法或拼接的操作易导致信息干扰这一缺陷,提出长距离重置更新模块,分别对无用信息与清晰特征加以抑制和更新。在图像去噪、图像去雨和低亮度图像增强3个不同任务的9个公开数据集上,与最新的25个方法进行的对比实验结果表明,所提出的轻量级交叉门控转换模型以较少的参数量和计算代价,在图像复原和图像增强领域中均取得较高的峰值信噪比和结构相似度,重构出接近真实世界场景的清晰图像,达到了先进的图像复原性能。

Recent image restoration and image enhancement methods are difficult to balance the robustness of multi-ple subtasks with the small number of parameters and computational costs.To solve this problem,this paper proposes a lightweight cross-gating transformer(CGT)for efficient image restoration task.On the one hand,this paper sum-marizes the limitations of traditional global self-attention mechanism to capture global dependencies,and improves the global self-attention mechanism to a cross-level cross-gating self-attention mechanism.Meanwhile,a light-weight feed-forward neural network is proposed to learn cross-level local dependencies at a very small computational cost and reconstruct clear features in the adjacent locality.On the other hand,in view of the defect that the traditional method of adding or concatenating encoder and decoder equally leads to information interference,a long-distance re-set update module is proposed to suppress and update useless information and clear features respectively.This paper conducts extensive quantitative experiments and is compared with 25 state-of-the-art methods on 9 datasets for image denoising,image deraining and low-light image enhancement,respectively.Experimental results prove that the pro-posed lightweight cross-gating transformer achieves high peak signal-to-noise ratio and structural similarity in im-age restoration and image enhancement tasks with a small number of parameters and computation,and reconstructs clear images close to real-world scenes,achieving state-of-the-art image restoration performance.