论文阅读笔记《Few-shot Classification via Adaptive Attention》

小样本学习&元学习经典论文整理||持续更新

核心思想

  本文提出一种基于注意力机制的小样本学习算法。作者认为基于参数优化的元学习算法优化过程过于复杂,而基于度量学习的小样本学习算法,虽然更加简单有效,但缺少对于新任务的适应能力。本文利用注意力机制根据支持集图像的特征图和查询集图像的特征图,得到对应的注意力图(Attention Maps),将注意力图与查询集图像的特征图做逐元素相乘,得到了优化后的特征图,再利用其进行分类。整个网络包含三个部分:特征提取网络,自适应注意力模块,分类器,网络结构如下图所示
在这里插入图片描述
  首先,支持集图像和查询集图像分别经过特征提取器 F ( ; θ F ) F(;\theta_F) F(;θF),得到对应的特征图 f s f_s fs f q f_q fq。然后,将 f s f_s fs f q f_q fq输入到自适应注意力模块 A ( ; θ A ) A(;\theta_A) A(;θA)中,用于计算注意力图,自适应注意力模块的结构如下图所示
在这里插入图片描述
  如图所示,自适应注意力模块主要由元权重生成器 A R ( ; θ R ) A_R(;\theta_R) AR(;θR)和空间注意力生成器 A S ( ; θ S ) A_S(;\theta_S) AS(;θS)构成,支持集图像的特征图经过元权重生成器得到对应的权重向量 w s = A R ( f s ; θ R ) w_s=A_R(f_s;\theta_R) ws=AR(fs;θR),然后将 w s w_s ws与查询集图像的特征图 f q f_q fq做逐通道相乘,得到类别相关的特征图 f q y s f_q^{y_s} fqys
在这里插入图片描述
这里其实是采用了通道注意力机制,作者认为支持集图像特征图的每个通道都表示不同的特征,但这些特征有些是重要的,有些是无关的。因此利用元权重生成器,将每个通道的特征图都转化为一个权重值,再与查询集图像特征图逐通道相乘,得到加权后的查询集图像特征图,这一特征图包含着每个类别支持集特征图的信息,因此称为类别相关的特征图 f q y s f_q^{y_s} fqys
  得到的类别相关特征图 f q y s f_q^{y_s} fqys接下来要输入到空间注意力模块 A S A_S AS中,该模块的作用是得到查询集特征图中与支持集相关的重要空间区域,输出对应的注意力图 M q y s M_q^{y_s} Mqys
在这里插入图片描述
在注意力图 M q y s M_q^{y_s} Mqys中,与支持集类别 y s y_s ys相关的区域其特征值要更大,反映到可视化的图像中就是那个部分的区域会更加明亮,而其他无关区域的色彩会更加暗沉,这本质上是一种空间注意力机制。将得到的注意力图 M q y s M_q^{y_s} Mqys与查询集特征图 f q f_q fq做逐元素相乘计算,就得到了优化后的特征图 f q M s f^{M_s}_q fqMs
在这里插入图片描述
  最后,将优化后的特征图 f q M s f^{M_s}_q fqMs输入到分类器 C C C中,分别得到每个支持集类别 y s y_s ys对应的分类得分 s y s q s_{y_s}^q sysq
在这里插入图片描述
整个过程可以表示为
在这里插入图片描述
其中 A ( f s , f q ) A(f_s,f_q) A(fs,fq)表示自适应注意力模块
在这里插入图片描述
即使在最后利用ReLU函数将输出值统一为正值,这个过程还是不对称的(指查询集特征图和支持集特征图不能交换位置),因此作者又改进为对称形式
在这里插入图片描述
d ( f s , f q ) d(f_s,f_q) d(fs,fq)表示最终的分类得分。

实现过程

网络结构

  特征提取网络采用4-conv或ResNet网络,元权重生成器采用SPP+3个全连接层结构,空间注意力生成器采用两个全连接层+全局平均池化层结构,分类器采用SPP+3个全连接层结构。

损失函数

  损失函数包含两个部分:注意力损失 L A t t L_{Att} LAtt和分类损失 L C E L_{CE} LCE
在这里插入图片描述
其中注意力损失为
在这里插入图片描述
a j q a_j^q ajq是将自适应注意力模块输出的注意力图 M q y i M_q^{y_i} Mqyi经过全局平均池化得到的
在这里插入图片描述
分类损失为
在这里插入图片描述
s j q s^q_j sjq表示分类得分,是对类别 y i y_i yi对应的 K K K个支持集图像得到的分类预测得分求平均值得到的
在这里插入图片描述

创新点

  • 引入了通道注意力机制和空间注意力机制对特征图进行优化,从而改善小样本分类效果
  • 设计了元权重生成器和空间注意力生成器结构,并将分类预测得分改为对称形式

算法评价

  本文利用注意力机制对特征图进行了优化,使其具备根据任务(查询集图像)进行自适应调整的能力。根据最后的可视化实验结果来看,得到的注意力图的确在对应的类别,对应的区域上有更明显的反应,这从一个侧面证实了注意力机制的作用。

如果大家对于深度学习与计算机视觉领域感兴趣,希望获得更多的知识分享与最新的论文解读,欢迎关注我的个人公众号“深视”。在这里插入图片描述

评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

深视

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值