arXiv:2501.15774cs.CV2025-01AAAI被引 23

轻量级图像超分模型,用注意力共享和信息蒸馏提升效率

Efficient Attention-Sharing Information Distillation Transformer for Lightweight Single Image Super-Resolution

  • 通过注意力共享与信息蒸馏减少自注意力计算开销
  • 仅30万参数即达主流模型性能,参数量显著更少
  • 适合移动端或资源受限场景的高效图像超分辨率

基于Transformer的图像超分方法虽能捕捉长程依赖关系,表现优于传统卷积网络,但计算复杂度高。为此,本文提出针对Transformer的轻量级超分网络ASID,融合注意力共享与信息蒸馏结构。将原用于高效卷积网络的信息蒸馏机制改造为降低堆叠自注意力层的计算负担,同时引入跨块注意力共享以进一步减少自注意力运算成本。该设计使ASID在仅约30万参数下实现与现有方法相当的性能,显著低于现有的基于CNN和Transformer的超分模型。在参数量匹配时,其表现优于当前最优方法,验证了其高效性与有效性。代码与补充材料可在项目页面获取。

原文摘要 · Abstract (English)

Transformer-based Super-Resolution (SR) methods have demonstrated superior performance compared to convolutional neural network (CNN)-based SR approaches due to their capability to capture long-range dependencies. However, their high computational complexity necessitates the development of lightweight approaches for practical use. To address this challenge, we propose the Attention-Sharing Information Distillation (ASID) network, a lightweight SR network that integrates attention-sharing and an information distillation structure specifically designed for Transformer-based SR methods. We modify the information distillation scheme, originally designed for efficient CNN operations, to reduce the computational load of stacked self-attention layers, effectively addressing the efficiency bottleneck. Additionally, we introduce attention-sharing across blocks to further minimize the computational cost of self-attention operations. By combining these strategies, ASID achieves competitive performance with existing SR methods while requiring only around 300K parameters - significantly fewer than existing CNN-based and Transformer-based SR models. Furthermore, ASID outperforms state-of-the-art SR methods when the number of parameters is matched, demonstrating its efficiency and effectiveness. The code and supplementary material are available on the project page.

图像超分Transformer轻量化注意力机制

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。