17篇注意力机制PyTorch实现,包含MLP、Re-Parameter系列热门论文
注意力(Attention)机制最早在计算机视觉中应用,后来又在 NLP 领域发扬光大,该机制将有限的注意力集中在重点信息上,从而节省资源,快速获得最有效的信息。
2014 年,Google DeepMind 发表《Recurrent Models of Visual Attention》,使注意力机制流行起来;2015 年,Bahdanau 等人在论文《Neural Machi17ne Translation by Jointly Learning to Align and Translate》中,将注意力机制首次应用在 NLP 领域;2017 年,Google 机器翻译团队发表的《Attention is All You Need》中,完全抛弃了 RNN 和 CNN 等网络结构,而仅仅采用注意力机制来进行机器翻译任务,并且取得了很好的效果,注意力机制也因此成了研究热点。
经过几年的发展,领域篇注内产生了众多的注意力机制论文研究,这些工作在 CV、NLP 领域取得了较好的效果。近日,在 GitHub 上,有研究者介绍了 17 篇关于注意力机制论文的 PyTorch 的代码实现以及使用方法。
项目地址:https://github.com/xmu-xiaoma666/External-Attention-pytorch
项目介绍
项目作者对注意力机制进行了分类,分为三个系列:Attent意力ion 系列、MLP 系列、ReP(Re-Parameter)系列。其中 Attention 系列中包含有大名鼎鼎的《Attention is All You Need》等 11 篇论文;最近比较热门的 MLP 系列包括谷歌的 MLP-Mixer、gMLP ,Facebook 的 ResMLP,清华的 RepMLP ;此外,ReP(Re-Parameter)系列包括清华等提出的 RepVGG、 ACNet。
Att机制ention 系列的 11 篇 Attention 论文 Pytorch 实现方式如下:
Pytorch 实现论文「Beyond Self-attention: External Attention using Two Linear Layers for Visual Tasks---arXiv 2020.05.05」 Pytorch 实现论文「Attention Is All You Need---PyNIPS2017」 Pytorch 实现论文「Simplified Self Attention Usage」 Pytorch 实现论文 「Squeeze-and-Excitation Networks---CVPR2018」 Pytorch 实现论文「Selective Kernel Networks---CVPR2019」 Pytorch 实现论文「CBAM: Convolutional Block ATottention Module---ECCV2018」 Pytorch 实现论文「BAM: Bottleneck Attention Module---BMCV2018」 Pytorch 实现论文「ECA-Net: Efficient Channel Attention for Deep Convolutional Neural Networks---CVPR2020」 Pytorch 实现论文「Duarcl Attention Network for Scene Segmentation---CVPR2019」 Pytorch 实现论文「EPSANet: An Efficient Pyramid Split Attention Block on Convolutional Neural Network---arXiv 2020.05.30」 Pytorch 实现论文 「ResT: An Efficient Trah实nsformer for Visual Recognition---arXiv 2020.05.28」
MLP(多层感知机)系列中,包含 4 篇论文 Pytorch 实现方式,论文如下:
Pytorch 实现论文「RepMLP: Re-parameterizing Convolutions into Fully-connected Layers for Image Recognit现,ion---arXiv 2020.05.05」 Pytorch 实现论文「MLP-Mixer: An all-MLP Architecture for Vision---arXiv 2020.05.17」 Pytorch 实现论文「ResMLP: Feedforward networks for image classification with data-efficient training---arX包含iv 2020.05.07」 Pytorch 实现论文「Pay Attention to MLPs---arXiv 2020.05.17」
ReP(Re-Parameter)系列中,包含 2 篇论文 Pytorch 实现方式,论文如下:
Pytorch 实现论文「RepVGG: Making VGG-style ConvNets Great Again---CVPR2021」 PyMLtorch 实现论文「ACNet: Strengthening the Kernel Skeletons for Powerful CNN via Asymmetric Convolution Blocks---ICCV2019」
总结来说,该项目共用 Pytorch 实现了 17 篇注意力机制论文。每篇论文包括题目(可直接链接到论文)、网络架构、代码。示例如下:
论文:「Beyond Self-atteP、ntion: External Attention using Two Linear Layers for Visual Tasks」。
网络框架:
代码:
from attention.ExternalAttention *import* ExternalAttention import torch inpuRet=torch.randn(50,49,512) ea = ExternalAttention(d_model=512,S=8) output=ea(input) print(output.shape)
-
Liquid AI 推出 LFM2-VL 模型,为手机带来「小巧又灵敏」的AI 视觉语言能力云计算核心技术Docker教程:Docker Swarm 模式从灾难中恢复这是极简单的人工神经网络解说包青天"智能重生"!华南师大发布国内首个法律AI大模型,2亿裁判文书秒变法律助手标签: 德适生物 AI AutoVision 医学影像大模型 三类创新医疗器械中国自动驾驶分级国标正式出台 明年3月份实施标签: 携程 同程旅行 去哪儿网 飞猪Pinterest推出AI内容限制工具:用户可自定义减少生成式AI图像标签: 携程 中东局势 旅游保障 航班退改云计算开发:Python3-translate()方法详解
- ·Solidigm推出D5-P5430数据中心QLC SSD
- ·Kimi For Coding 会员计划推出全新增值权益,提升开发体验
- ·微软智能云Azure在华的第五个数据中心区域正式启用
- ·标签: 影微创新 AI芯片 NPU AIGC
- ·用户画像被批“没啥用”!到底咋做才有用?
- ·标签: 中盐股份 IPO 中信建投 食用盐
- ·标签: 千寻智能 解浚源 具身智能 字节跳动
- ·AI竞赛让谷歌数据中心用水量飙升 同比增长20%
- ·标签: CHANNELV 星空传媒 星空卫视 音乐频道
- ·标签: B站
- ·标签: 术锐机器人 徐凯 手术机器人 IPO辅导
- ·布局人工智能,银行有的不只是智能客服
- ·标签: 华为 吉利 安世半导体 谷歌
- ·标签: 诺基亚 贝尔 华信 股权转让
- ·标签: 海致科技 任旭阳 大模型除幻第一股 图模融合
- ·标签: 天博智能 汽车热管理 IPO 中信建投
- ·字节跳动发布开源大语言模型 Seed-OSS,助力开发者与研究者
- ·标签: 小米汽车 宋钢 特斯拉上海工厂 雷军
- ·标签: SpaceX 马斯克 特斯拉 Megapack
- ·Spotify前高管400万美元新作震撼登场!AI秒制课程颠覆传统教育,9种学习模式让知识获取变得像听音乐一样简单
- ·标签: 以旧换新 新能源汽车 国家发改委 农机补贴
- ·技术人才对 AI 招聘工具的不信任:68% 专业人士表示担忧
- ·标签: 腾讯 姚顺雨 AI Infra部 ReAct
- ·游戏开发商Epic扩大与苹果纠纷:在英国起诉苹果垄断
- ·云计算开发:Python练习实例-打印出所有的"水仙花数"
- ·苹果发布iPhone 17系列新品,AI功能缺失引发行业关注
- ·https://www.nbm.org/?action=QUERY&s=谷歌霸屏留痕【tg:seogoogle987】,nhh
- ·https://www.miamiopenmri.com/?action=CHECK&s=谷歌霸屏留痕【tg:seogoogle987】,4m8
- ·https://www.xbiao.com/search/index?tpf&wd=谷歌霸屏留痕【tg:seogoogle987】,gur
- ·https://www.urmc.rochester.edu/news/search?q=谷歌霸屏留痕【tg:seogoogle987】,K4e
- ·https://www.teamlab.art/zh-hans/?submit=谷歌霸屏留痕【tg:seogoogle987】,VkX
- ·https://www.yidian51.com/taobao/谷歌霸屏留痕【tg:seogoogle987】,gV0
- ·https://www.smics.com/tc/site/search?token=谷歌霸屏留痕【tg:seogoogle987】,rzl
- ·https://www.ushmm.org/search/results/?ID=8s0587Oi8UTv74zPPM4fuhwjNpe2hsEE&q=谷歌霸屏留痕【tg:seogoogle987】,K7M
- ·https://www.signingsavvy.com/sentence/谷歌霸屏留痕【tg:seogoogle987】,Jph
- ·https://www.nslcleaders.org/?s=谷歌霸屏留痕【tg:seogoogle987】,iDG
