cosFormer ICLR 重新思考注意力机制中的Softmax 2022 (cosforge)

文章编号:40986 最新资讯 2024-12-03 Softmax cosFormer 商汤

导读:Transformer在自然语言处理、计算机视觉和音频处理方面取得了巨大成功。作为其核心组成部分之一,Softmax Attention模块能够捕捉长距离的依赖关系,但由于Softmax算子关于序列长度的二次空间和时间复杂性,使其很难扩展。针对这点,研究者提出利用核方法以及稀疏注意力机制的方法来近似Softmax算子,从而降低时间空间复杂度。但是,由于误差的存在,效果往往不尽如人意。

我们(商汤多模态研究组)认为,近似操作本身存在的误差使得其效果很难超越Softmax Attention。我们的观点是,与其近似Softmax,不如设计一种方式代替Softmax,并且同时降低时间空间复杂度。因此,本文提出了名为cosFormer的方法,在时间空间复杂度关于序列长度为线性复杂度的同时,其性能接近或者超越Softmax Attention,并在LRA benchmark上取得SOTA结果。我们的设计核心理念基于两点,首先是注意力矩阵的非负性,其次是对局部注意力的放大(非极大值抑制)。

本文主要介绍已收录于ICLR 2022的一篇文章 cosFormer : RethinkingSoftmaxinAttention。

 2022 cosFormer:重新思考注意力机制中的Sofx

论文地址:

部分开源代码:

 2022 cosFormer:重新思考注意力机制中的Sofx

图片表示各种transformer在LRA benchmark上的表现,其中y轴表示性能,x轴表示速度,圆圈大小表示内存,我们提出的cosFormer取得了明显的优势。

一、背景

 2022 cosFormer:重新思考注意力机制中的Sofx

3、Softmax的两大性质

我们经过分析以及实验,归纳出Softmax Attention中比较重要的性质,这两个性质可以指导我们的模型设计:

1.注意力矩阵的非负性

2.局部注意力的放大(非极大值抑制)

 2022 cosFormer:重新思考注意力机制中的Sofx

所以我们的方法需要在加了reweighting操作后也更加集中在对角线附近。注意并非所有的有类似权重的函数均适用,这个reweighting的函数需要跟前面的QK一样可以拆分成两个矩阵的乘法的形式。

至此,就可以引入我们的cosFormer了。

二、cosFormer

1、方法

我们的方法基于线性Attention,首先给出符号定义:

 2022 cosFormer:重新思考注意力机制中的Sofx

 2022 cosFormer:重新思考注意力机制中的Sofx

2、实验结果

我们在单向模型,双向模型以及LRA benchmark上测试了我们的方法,均取得了非常不错的效果。

单向语言模型,指标表示困惑度(越低越好):  2022 cosFormer:重新思考注意力机制中的Sofx

论文地址:

版权文章,未经授权禁止转载。详情见 转载须知 。


本文地址: /article/be0e91d84d8569e6a9a1.html
全局中部横幅
全局中部横幅
江苏开锁培训_学开锁_配汽车遥控培训_学配汽车钥匙_【安信开锁培训学校】

江苏开锁培训,教开锁,学配汽车遥控,学开锁技术,开锁学校,包学会免费住宿。欢迎各地学员实地考察,报销路费,学开锁找安信!

忍者必须死3下载_忍者必须死3安卓版下载_最新版-4399手机游戏网

4399安卓游戏为您提供忍者必须死3手机版下载、忍者必须死3安卓版下载,忍者必须死3最新版攻略、评测,更多忍者必须死3下载尽在4399手机游戏网。

人民网_网上的人民日报

人民网,是世界十大报纸之一《人民日报》建设的以新闻为主的大型网上信息发布平台,也是互联网上最大的中文和多语种新闻网站之一。作为国家重点新闻网站,人民网以新闻报道的权威性、及时性、多样性和评论性为特色,在网民中树立起了“权威媒体、大众网站”的形象。

防爆电器-中通智能防爆行业领导品牌

中通智能专业生产防爆产品,防爆电器,防爆小屋,防爆柜,防爆配电箱,防爆空调,防爆控制箱,防爆空调,防爆监控,led防爆灯等防爆电器产品。

首页_浙江奔克汽车部件有限公司,汽车内外饰,晴雨挡,挡泥板,电镀件,扶手箱,官方网站

浙江奔克汽车部件有限公司(前身是瑞安奔克汽车部件有限公司)成立于1993年,是一家集产品研发、制造和销售于一体的汽车用品生产厂家。专业研发和制造汽车冲压、五金、塑胶、电镀、吸塑、注塑等内外饰零部件。

花卉谷 - 您身边的养花专家

花卉谷提供花卉种植基地和种植技术,包括室内花卉、阳台花卉、盆栽花卉等,适合各种场合和地域的花卉种植方案。多肉花卉、温室花卉、苗木花卉种植技术齐全,让你更好地了解每个花卉的特点和养护方法,让你的花儿更加健康美丽。

首页-高速石墨立式带锯床_高速石墨卧式带锯床_金刚砂环线锯切机

高速石墨立式带锯床_高速石墨卧式带锯床_金刚砂环线锯切机

塑料造粒机,废旧塑料造粒机,再生塑料造粒机-镇江克锐特塑料机械有限公司

江苏克锐特塑料机械有限公司生产PP、ABS、PC、HIPS、HDPE、LDPE等塑料造粒机.废旧塑料回收造粒机,再生塑料造粒机,塑料再生造粒机。源头厂家,售后保障。

重庆搬家,重庆搬家电话,价格优,服务好【本站出租】-重庆搬家公司

重庆搬家公司,正规靠谱,优质服务,无损搬家,全部归位,搬家价格透明,拒绝乱收费,提供重庆小型搬家,长途搬家,公司搬迁,重庆搬家电话提供咨询预约,拨打搬家电话,了解重庆搬家公司收费标准,重庆市搬家公司哪家好等事宜.服务区域:万州区,黔江区,涪陵区,渝中区,大渡口区,江北区,沙坪坝区,九龙坡区,南岸区,北碚区,渝北区,巴南区,长寿区,江津区,合川区,永川区,南川区,綦江区,大足区,璧山区,铜梁区,潼南区,荣昌区,开州区,梁平区,武隆区及其重庆周边地区搬家服务

红外热成像摄像机|森林防火监控测温预警热成像|中远距离激光夜视摄像机|SWIR短波红外相机InGaAs|多光谱光电转台监控云台|船舶夜视仪|森林防火软件地理信息系统GIS软件|深圳市华下数字技术有限公司

森林防火监控、长波非制冷机芯、中波制冷机芯、短波红外机芯镜头、船舶红外夜视热成像辅助航行摄像机、中远距离激光夜视摄像机、手持式热成像仪、测温报警热成像、远距离高清透雾日夜摄像机、车载热成像助行仪、热成像瞄准镜、多光谱云台、抗风转台、高清电动变焦镜头

三维电子沙盘|数字沙盘|AI|GIS地理|无人机|人工智能|成都趋势电子|VR AR MR XR虚拟现实|大数据|数字孪生|武警部队军事公安应急消防安监煤矿石油水电|一张图可视化|多点触摸屏幕-成都趋势电子有限责任公司

国产银河麒麟二三维数字沙盘大数据可视化研训推演模拟仿真地理信息系统(T3D&M3DGIS)是成都趋势电子有限责任公司秉承贰拾多载(2001)工匠精神自主知识产权的通过手势多点触摸控制的大型二、三维一体化地理信息系统;以三维地理信息(3DMAX、BIM模型)、部队部署、三维态势要图标绘推演、综合查询等为基础支撑,深度开发联网标图、无人机三维倾斜摄影、实时视频防控、智慧消防救援、公安接处警、武警反恐防爆、人防应急抢险救灾、交通事故热点路况查询分析等功能;符合军用技术体制和标准规范,提供丰富的二次开发接口(公众号:m3dgis2001)

全局底部横幅