清华计|arXiv爆款:想了解Attention就看它!清华计图胡事民团队出品

博雯 发自 凹非寺
量子位 报道 | 公众号 QbitAI
清华计图胡事民团队的这篇注意力机制的综述火了!
在上周的arXiv上,这是最热的一篇论文:
清华计|arXiv爆款:想了解Attention就看它!清华计图胡事民团队出品

推特以及GitHub上也有不低的热度:
清华计|arXiv爆款:想了解Attention就看它!清华计图胡事民团队出品

而这篇论文引用近200篇内容,对计算机视觉领域中的各种注意力机制进行了全面回顾。
在大量调查之后,论文将注意力机制分为多个类别,GitHub还给出了各类别下提到内容的PDF下载文件:
清华计|arXiv爆款:想了解Attention就看它!清华计图胡事民团队出品

现在,就来一起看看这篇论文。
文章主要内容论文首先将基于注意力的模型在计算机视觉领域中的发展历程大致归为了四个阶段:

  1. 将深度神经网络与注意力机制相结合,代表性方法为RAM
  2. 明确预测判别性输入特征,代表性方法为STN
  3. 隐性且自适应地预测潜在的关键特征,代表方法为SENet
  4. 自注意力机制
清华计|arXiv爆款:想了解Attention就看它!清华计图胡事民团队出品

同时,注意力机制也被分为了通道注意、空间注意、时间注意、分支注意,以及两个混合类别:
清华计|arXiv爆款:想了解Attention就看它!清华计图胡事民团队出品

针对不同类别,研究团队给出了其代表性方法和发展背景:
通道注意力(Channel Attention)
在深度神经网络中,不同特征图的不同通道常代表不同对象。
而通道注意力作为一个对象选择过程,可以自适应地重新校准每个通道的权重,从而决定关注什么。
因此,按照类别和出版日期将代表性通道关注机制进行分类,应用范围有分类(Cls)、语义分割(SSeg)、实例分割(ISeg)、风格转换(ST)、动作识别(Action)。
其中,(A)代表Channel-wise product,(I)强调重要通道,(II)捕捉全局信息。
清华计|arXiv爆款:想了解Attention就看它!清华计图胡事民团队出品

空间注意力(Spatial Attention)
空间注意力可以被看作是一种自适应的空间区域选择机制。
其应用范围比通道注意力多出了精细分类(FGCls)和图像字幕(ICap)。
清华计|arXiv爆款:想了解Attention就看它!清华计图胡事民团队出品

时间注意力(Temporal Attention)
时间注意力可以被看作是一种动态的时间选择机制,决定了何时进行注意,因此通常用于视频处理。
清华计|arXiv爆款:想了解Attention就看它!清华计图胡事民团队出品

分支注意力(Branch Attention)
分支注意可以被看作是一种动态的分支选择机制,通过多分支结构决定去注意什么。
清华计|arXiv爆款:想了解Attention就看它!清华计图胡事民团队出品

通道空间注意力(Channel & Spatial Attention)
通道和空间结合的注意力机制可以自适应地选择重要的对象和区域,由残差注意力(Residual Attention)网络开创了这一内容。
在残差注意力之后,为了有效利用全局信息,后来的工作又相继引入全局平均池化(Global Average Pooling),引入自注意力机制等内容。
清华计|arXiv爆款:想了解Attention就看它!清华计图胡事民团队出品

时空注意力(Spatial & Temporal Attention)
时空注意力机制可以自适应地选择重要区域和关键帧。
清华计|arXiv爆款:想了解Attention就看它!清华计图胡事民团队出品

最后,作者也提出了注意力机制在未来的一些研究方向: