中文

面向图像超分辨率的混合注意力聚合 Transformer(HAAT)

图像与视频处理 2024-12-11 v3 人工智能 计算机视觉与模式识别

摘要

在图像超分辨率研究领域,基于 Swin-transformer 的模型因其全局空间建模能力及平移窗口注意力机制而受到青睐。然而,现有方法常限制于非重叠窗口以降低计算成本,忽略跨通道的有用信息。为此,本文提出一种新型模型 Hybrid Attention Aggregation Transformer(HAAT),以更好地利用特征信息。HAAT 由集成 Swin-Dense-Residual-Connected Blocks(SDRCB)与 Hybrid Grid Attention Blocks(HGAB)构成。SDRCB 扩大感受野且保持简洁架构,提升性能。HGAB 融合通道注意力、稀疏注意力与窗口注意力,增强非局部特征融合,实现更具视觉效果的输出。实验表明,HAAT 在基准数据集上超越了当前最先进的方法。

关键词

引用

@article{arxiv.2411.18003,
  title  = {HAAT: Hybrid Attention Aggregation Transformer for Image Super-Resolution},
  author = {Song-Jiang Lai and Tsun-Hin Cheung and Ka-Chun Fung and Kai-wen Xue and Kin-Man Lam},
  journal= {arXiv preprint arXiv:2411.18003},
  year   = {2024}
}

备注

6 pages, 2 figures, 1 table