通过让注意力头学习不同子空间,解决图像修复中注意力冗余问题。
Devil is in the Uniformity: Exploring Diverse Learners within Transformer for Image Restoration

- 让注意力头在不同大小子空间学习,提升特征多样性
- 在12个基准上实现超越现有方法的修复效果
- 适合图像修复、视觉恢复方向的研究者参考
基于Transformer的方法在图像修复中受到广泛关注,其核心组件多头注意力(MHA)对捕捉多样特征和恢复高质量结果至关重要。传统MHA中各注意力头独立计算,从均匀划分的子空间中提取信息,导致冗余问题,影响模型表现。本文提出一种新型架构HINT,包含层级多头注意力(HMHA)与查询-键缓存更新(QKCU)模块,以缓解该冗余。其中,HMHA通过让注意力头在不同尺度和内容的子空间中学习,提取多样化上下文特征;QKCU则通过层内与跨层交互机制,进一步增强头间协作。在5类图像修复任务(低光增强、去雾、去雪、去噪、去雨)共12个基准上的实验表明,HINT显著优于现有方法。源代码见补充材料。
原文摘要 · Abstract (English)
Transformer-based approaches have gained significant attention in image restoration, where the core component, i.e, Multi-Head Attention (MHA), plays a crucial role in capturing diverse features and recovering high-quality results. In MHA, heads perform attention calculation independently from uniform split subspaces, and a redundancy issue is triggered to hinder the model from achieving satisfactory outputs. In this paper, we propose to improve MHA by exploring diverse learners and introducing various interactions between heads, which results in a Hierarchical multI-head atteNtion driven Transformer model, termed HINT, for image restoration. HINT contains two modules, i.e., the Hierarchical Multi-Head Attention (HMHA) and the Query-Key Cache Updating (QKCU) module, to address the redundancy problem that is rooted in vanilla MHA. Specifically, HMHA extracts diverse contextual features by employing heads to learn from subspaces of varying sizes and containing different information. Moreover, QKCU, comprising intra- and inter-layer schemes, further reduces the redundancy problem by facilitating enhanced interactions between attention heads within and across layers. Extensive experiments are conducted on 12 benchmarks across 5 image restoration tasks, including low-light enhancement, dehazing, desnowing, denoising, and deraining, to demonstrate the superiority of HINT. The source code is available in the supplementary materials.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。