用低秩注意力提升图像修复效率,减少计算量与参数。
GLMHA A Guided Low-rank Multi-Head Self-Attention for Efficient Image Restoration and Spectral Reconstruction
- 提出实例引导的低秩多头自注意力机制。
- 可减少7.7亿次浮点运算和37万参数,性能接近原模型。
- 适合需要高效推理的图像修复与光谱重建任务。
图像修复与光谱重建是计算机视觉中的长期挑战。当前表现最优的模型多为CNN-Transformer混合结构,其核心设计依赖通道自注意力(CSA)。本文首次揭示CSA整体为低秩操作,并提出实例引导的低秩多头自注意力(GLMHA)以替代CSA,实现显著计算效率提升,同时保持原模型性能。与现有方法(如Linformer、Performer、Reformer)不同,GLMHA在短序列和长序列输入下均能降低浮点运算量(FLOPs)和参数量,而后者在短序列时优势不明显且忽略参数压缩。我们在从RGB重构光谱、快照压缩成像光谱重建、运动去模糊与图像去雨等任务上,将最佳模型替换为GLMHA。结果表明,最多可减少7.7亿浮点运算和37万参数,性能几乎不变。
原文摘要 · Abstract (English)
Image restoration and spectral reconstruction are longstanding computer vision tasks. Currently, CNN-transformer hybrid models provide state-of-the-art performance for these tasks. The key common ingredient in the architectural designs of these models is Channel-wise Self-Attention (CSA). We first show that CSA is an overall low-rank operation. Then, we propose an instance-Guided Low-rank Multi-Head selfattention (GLMHA) to replace the CSA for a considerable computational gain while closely retaining the original model performance. Unique to the proposed GLMHA is its ability to provide computational gain for both short and long input sequences. In particular, the gain is in terms of both Floating Point Operations (FLOPs) and parameter count reduction. This is in contrast to the existing popular computational complexity reduction techniques, e.g., Linformer, Performer, and Reformer, for whom FLOPs overpower the efficient design tricks for the shorter input sequences. Moreover, parameter reduction remains unaccounted for in the existing methods.We perform an extensive evaluation for the tasks of spectral reconstruction from RGB images, spectral reconstruction from snapshot compressive imaging, motion deblurring, and image deraining by enhancing the best-performing models with our GLMHA. Our results show up to a 7.7 Giga FLOPs reduction with 370K fewer parameters required to closely retain the original performance of the best-performing models that employ CSA.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。