U-shaped Dual Attention Transformer: An Efficient Transformer Based on Channel and Spatial Attention
Transformer-based methods have demonstrated remarkable performance on image super-resolution tasks. Due to high computational complexity, researchers have been working to achieve a balance between computation costs and performance. Restormer has achieved commendable balance by utilizing global channel attention. However, the performance is limited by insufficient local pixel reconstruction. In this paper, we propose a U-shaped Dual Attention Transformer (UDAT) with local-global receptive field, addressing the limitation of Restormer in local pixel reconstruction. We propose a dense window channel attention to enhance the local feature representation, more efficient in computational complexity. Experiments demonstrate that our UDAT achieve superior performance compared with Restormer on benchmark datasets, surpassing Restormer by 0.57 dB on the Urban100 dataset. On-par with SwinIR, our method reduces computational complexity by 3.2 times and improves inference speed by 2 times, achieving a better balance between computational costs and performance.
Authors
- Zhen Ping Zhou (ORCID: https://orcid.org/0000-0003-3232-9903)
- Peng Qiao (ORCID: https://orcid.org/0000-0001-6752-7892)
- Rongchun Li (ORCID: https://orcid.org/0000-0001-5922-7961)
- Z. Zhai (ORCID: https://orcid.org/0009-0009-4095-1832)
Institutions
- National University of Defense Technology (CN)
Publication Details
- Published
- 2024-12-27
- DOI
- https://doi.org/10.1109/icairc64177.2024.10900137
- Primary Topic
- CCD and CMOS Imaging Sensors
- Type
- article
- Field-Weighted Citation Impact
- 0.00