HASN: Hybrid Attention Separable Network for Efficient Image Super-resolution

Weifeng Cao, Xiaoyan Lei, Jun Shi, Wanyong Liang, Jie Liu +1 more
10/13/2024
eess.IVcs.CV

Official code: nathan66666/HASN (8 stars, last commit 2024-07-12).

Abstract

Recently, lightweight methods for single image super-resolution (SISR) have gained significant popularity and achieved impressive performance due to limited hardware resources. These methods demonstrate that adopting residual feature distillation is an effective way to enhance performance. However, we find that using residual connections after each block increases the model's storage and computational cost. Therefore, to simplify the network structure and learn higher-level features and relationships between features, we use depthwise separable convolutions, fully connected layers, and activation functions as the basic feature extraction modules. This significantly reduces computational load and the number of parameters while maintaining strong feature extraction capabilities. To further enhance model performance, we propose the Hybrid Attention Separable Block (HASB), which combines channel attention and spatial attention, thus making use of their complementary advantages. Additionally, we use depthwise separable convolutions instead of standard convolutions, significantly reducing the computational load and the number of parameters while maintaining strong feature extraction capabilities. During the training phase, we also adopt a warm-start retraining strategy to exploit the potential of the model further. Extensive experiments demonstrate the effectiveness of our approach. Our method achieves a smaller model size and reduced computational complexity without compromising performance. Code can be available at https://github.com/nathan66666/HASN.git

View on arXivView PDF

Code Implementations(1)

nathan66666/HASNOfficial100%

Accepted by Visual Computer

80C++, CudaJul 3, 20242 years ago

Similar Papers

Cite this paper

@article{cao2024hasn,
  title  = {HASN: Hybrid Attention Separable Network for Efficient Image Super-resolution},
  author = {Weifeng Cao and Xiaoyan Lei and Jun Shi and Wanyong Liang and Jie Liu and Zongfei Bai},
  year   = {2024},
  eprint = {2410.09844},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2410.09844v1}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2410.09844.svg)](https://codeofpaper.com/paper/2410.09844)

Discussion