木星链 木星链
Ctrl+D收藏木星链

atc:别再用平均池化层了!Meta AI把注意力塞到池化层,性能提升0.3

作者:

时间:1900/1/1 0:00:00

编辑:LRS

注意力机制这么好用,怎么不把它塞到卷积网络里?最近MetaAI的研究人员提出了一个基于注意力的池化层,仅仅把平均池化层替换掉,就能获得+0.3%的性能提升!

VisualTransformer作为计算机视觉领域的新兴霸主,已经在各个研究任务中逐渐替换掉了卷积神经网络CNN。

ViT与CNN之间存在着许多不同点,例如ViT的输入是imagepatch,而非像素;分类任务中,ViT是通过对类标记进行决策等等。

classtoken实际上是ViT论文原作者提出,用于整合模型输入信息的token。classtoken与每个patch进行信息交互后,模型就能了解到具体的分类信息。

并且在自注意力机制中,最后一层中的softmax可以作为注意力图,根据classtoken和不同patch之间的交互程度,就能够了解哪些patch对最终分类结果有影响及具体程度,也增加了模型可解释性。

但这种可解释性目前仍然是很弱的,因为patch和最后一层的softmax之间还隔着很多层和很多个header,信息之间的不断融合后,很难搞清楚最后一层softmax是否真的可以解释分类。

新世界:与多家银行在新世界开展近10次数字货币试点,取得良好效果:金色财经报道,新世界4月7日披露投资者关系活动记录表显示,公司作为数字货币首批商业应用单位,与多家银行在新世界开展了近10次数字货币的试点,取得了数字货币试点的良好效果。未来,新世界将持续发挥在商、旅、文、康、体多业态联动的优势,拓展在线经济,吸纳线上流量资源,整合创意思考,实现多方共赢,为打响上海购物品牌做出努力和贡献。[2023/4/9 13:52:34]

所以如果ViT和CNN一样有视觉属性就好了!

最近MetaAI就提出了一个新模型,用attentionmap来增强卷积神经网络,说简单点,其实就是用了一个基于注意力的层来取代常用的平均池化层。

仔细一想,池化层和attention好像确实很配啊,都是对输入信息的加权平均进行整合。加入了注意力机制以后的池化层,可以明确地显示出不同patch所占的权重。

PUMA公布Super PUMA NFT发行计划,60%将空投给NITRO NFT持有者:2月17日消息,据PUMA“彪马”官方社交媒体,该时尚运动品牌公布了Super PUMA NFT发行计划细节,该系列将是2D NFT,总供应量为10,000枚,其中6,000枚将分配给NITRO NFT持有者,每持有一枚NITRO将会被自动空投一枚Super PUMA NFT;剩余的4,000枚Super PUMA NFT将于美国东部时间2月22日上午10:00开始向公众开放,初始价格为0.15 ETH。PUMA公司将会在后续一系列产品和计划中使用Super PUMA,包括服装、毛绒玩具和漫画书。[2023/2/17 12:12:44]

并且与经典ViT相比,每个patch都会获得一个单一的权重,无需考虑多层和多头的影响,这样就可以用一个简单的方法达到对注意力可视化的目的了。

在分类任务中更神奇,如果对每个类别使用不同颜色进行单独标记的话,就会发现分类任务也能识别出图片中的不同物体。

PoS验证服务商Smart Stake将于2月21日关闭其Secret网络验证器:1月30日消息,PoS验证服务商Smart Stake将于2月21日关闭其Secret网络验证器,用户需重新委托给另一个验证器。它将其归因于Secret上复杂的验证器操作、操作的成本/工作量、社区内最近发生的事。[2023/1/30 11:36:27]

基于Attention的池化层

文章中新提出的模型叫做PatchConvNet,核心组件就是可学习的、基于attention的池化层。

模型架构的主干是一个卷积网络,相当于是一个轻量级的预处理操作,它的作用就是把图像像素进行分割,并映射为一组向量,和ViT中patchextraction操作对应。

Circle Yield在偿还客户贷款后暂停新贷款:12月2日消息,USDC发行商Circle发言人在一封电子邮件中表示,公司旗下理财产品Circle Yield在偿还客户贷款后不接受新的贷款。而在几周前,随着FTX破产的蔓延,Circle将Circle Yield的收益率降至零。

据悉,Circle Yield以USDC为基础,允许人们将现金配置到基于加密货币的投资中,该投资由比特币超额抵押,并获得高于传统银行利率的收益率。

此前11月16日消息,Genesis加密货币借贷部门暂停客户赎回和新贷款发放。11月17日消息,Circle表示,Genesis是其理财产品的交易对手方,此产品客户未偿贷款为260万美元。(Blockworks)[2022/12/2 21:17:00]

最近也有研究表明,采用卷积的预处理能让模型的性能更加稳定。

模型的第二部分column,包含了整个模型中的大部分层、参数和计算量,它由N个堆叠的残差卷积块组成。每个块由一个归一化、1*1卷积,3*3卷积用来做空间处理,一个squeeze-and-excitation层用于混合通道特征,最后在残差连接前加入一个1*1的卷积。

彭博社:拜登政府正考虑对马斯克的收购交易进行国家安全审查:10月21日消息,据彭博社援引消息人士报道,拜登政府正考虑对马斯克的投资项目进行国家安全审查,包括收购推特和 Starlink 的交易。

消息人士表示,美国官方对马斯克近期关于停止向乌克兰提供 Starlink 服务的相关言论感到不安,并对马斯克与外国投资者购买推特的计划感到担忧。目前进行国家安全审查的讨论仍处于初期阶段。美国官员正在权衡哪些工具可以让联邦审查马斯克的企业。[2022/10/21 16:34:12]

研究人员对模型块的选择也提出了一些建议,例如在batchsize够大的情况下,BatchNorm往往效果比LayerNorm更好。但训练大模型或者高分辨率的图像输入时,由于batchsize更小,所以BatchNorm在这种情况下就不太实用了。

下一个模块就是基于注意力的池化层了。

在主干模型的输出端,预处理后的向量通过类似Transformer的交叉注意力层的方式进行融合。

注意力层中的每个权重值取决于预测patch与可训练向量之间的相似度,结果和经典ViT中的classtoken类似。

然后将产生的d维向量添加到CLS向量中,并经过一个前馈网络处理。

与之前提出的class-attentiondecoder不同之处在于,研究人员仅仅只用一个block和一个head,大幅度简化了计算量,也能够避免多个block和head之间互相影响,从而导致注意力权重失真。

因此,classtoken和预处理patch之间的通信只发生在一个softmax中,直接反映了池化操作者如何对每个patch进行加权。

也可以通过将CLS向量替换为k×d矩阵来对每个类别的attentionmap进行归一化处理,这样就可以看出每个块和每个类别之间的关联程度。

但这种设计也会增加内存的峰值使用量,并且会使网络的优化更加复杂。通常只在微调优化的阶段以一个小的学习率和小batchsize来规避这类问题。

实验结果

在图像分类任务上,研究人员首先将模型与ImageNet1k和ImageNet-v2上的其他模型从参数量,FLOPS,峰值内存用量和256张图像batchsize下的模型推理吞吐量上进行对比。

实验结果肯定是好的,可以看到PatchConvNet的简单柱状结构相比其他模型更加简便和易于扩展。对于高分辨率图像来说,不同模型可能会针对FLOPs和准确率进行不同的平衡,更大的模型肯定会取得更高的准确率,相应的吞吐量就会低一些。

在语义分割任务上,研究人员通过ADE20k数据集上的语义分割实验来评估模型,数据集中包括2万张训练图像和5千张验证图像,标签超过150个类别。由于PatchConvNet模型不是金字塔式的,所以模型只是用模型的最后一层输出和UpperNet的多层次网络输出,能够简化模型参数。研究结果显示,虽然PatchConvNet的结构更简单,但与最先进的Swin架构性能仍处于同一水平,并且在FLOPs-MIoU权衡方面优于XCiT。

在检测和实例分割上,研究人员在COCO数据集上对模型进行评估,实验结果显示PatchConvNet相比其他sota架构来说,能够在FLOPs和AP之间进行很好的权衡。

在消融实验中,为了验证架构问题,研究人员使用不同的架构对比了Transformer中的classattention和卷积神经网络的平均池化操作,还对比了卷积主干和线性投影之间的性能差别等等。实验结果可以看到卷积主干是模型取得最佳性能的关键,class-attention几乎没有带来额外的性能提升。

另一个重要的消融实验时attention-basedpooling和ConvNets之间的对比,研究人员惊奇地发现可学习的聚合函数甚至可以提高一个ResNet魔改后模型的性能。

通过把attention添加到ResNet50中,直接在Imagenet1k上获得了80.1%的最高准确率,比使用平均池化层的baseline模型提高了+0.3%的性能,并且attention-based只稍微增加了模型的FLOPs数量,从4.1B提升到4.6B。

参考资料:

https://arxiv.org/abs/2112.13692

标签:atcFLOPFLOatc币是什么币FLOP币FLOP价格FLO价格FLO币

pepe最新价格热门资讯
NFT:Highstreet与Taiko将于1月24日联合发售NFC-NEKO猫毯NFT

鞭牛士1月24日消息,NFT限量版产品兑换市场Highstreet与国际创意机构Taiko合作推出NFC-NEKO猫毯NFT.

1900/1/1 0:00:00
元宇宙:元宇宙是未来还是局,元宇宙是什么意思?

书中都逃不开两个起源,首先元宇宙这个词的来源,科幻小说家尼尔斯蒂芬森在1992年写的科幻小说《雪崩》,在这本书中人们不仅在现实中生活,更生活在一个叫迈尔斯的虚拟空间中.

1900/1/1 0:00:00
元宇宙:什么是元宇宙?元宇宙在哪儿?它真的能改变人类世界的运行方式吗

当下有一个事物非常的火,它就是元宇宙。对于绝大多数人来说,只知道宇宙这个名词,那么究竟什么是元宇宙?元宇宙到底在哪?它真的能改变人类世界的运行方式吗?今天这条视频,我们就来聊聊这个话题.

1900/1/1 0:00:00
APP:破获虚拟币案件259起,收缴虚拟币价值110亿余元!

防大数据 部网络安全保卫局二级巡视员孔长青1月14日下午向记者介绍,当前网络黑灰产背后埋藏巨大“陷阱”,针对其不同环节,分别打击治理.

1900/1/1 0:00:00
元宇宙:春晚带火元宇宙?一虚拟币暴涨几百倍

1,概念 元宇宙的解释:「元宇宙」的内涵是吸纳了信息革命、人工智能革命,以及VR、AR、MR,特别是游戏引擎在内的虚拟现实技术革命的成果.

1900/1/1 0:00:00
GRS:电动化之后再无驾驶乐趣?特斯拉/宝马/丰田等纯电跑车展望

易车原创在中国、欧洲、美国等引领的电动化时代中,我们见到了太多的纯电SUV和轿车,但是纯电跑车好像几乎处于空白状态.

1900/1/1 0:00:00