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

文章编号:40986 资讯动态 2024-12-03 SoftmaxcosFormer商汤

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

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

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

ICLR  cosFormer:重新思考注意力机制中的Soft

论文地址:

部分开源代码:

ICLR  cosFormer:重新思考注意力机制中的Soft

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

一、背景

ICLR  cosFormer:重新思考注意力机制中的Soft

3、Softmax的两大性质

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

1.注意力矩阵的非负性

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

ICLR  cosFormer:重新思考注意力机制中的Soft

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

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

二、cosFormer

1、方法

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

ICLR  cosFormer:重新思考注意力机制中的Soft

ICLR  cosFormer:重新思考注意力机制中的Soft

2、实验结果

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

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

论文地址:

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

全局中部横幅
苏州

苏州浦阳瑞钢铁有限公司销售热线13814814096,专业生产加工苏州钢板切割加工,镀锌板,苏州冷板,常熟花纹板,开平板,酸洗板,张家港方管,镀锌方管,槽钢,热镀锌槽钢,工字钢,H型钢,常熟无缝管,镀锌管,厂家批发直销,价格优惠,有需要请联系我们,公司本着诚信为本,互惠互利的原则为客户提供最佳服务,欢迎新老客户惠顾!高质量产品值得您选购。

专注于广告­­­

【牛云说营销】-用新广告,新策划,新营销的思路,帮助传统企业突破营销困境,利用移动互联网工具,依托于传统店面,进行线上线下互动营销,实现企业互联网+深度提炼,精华解读,用最简单的方式玩转,餐饮、装修、房地产、健身房等传统企业营销,让您迅速成为营销牛人!

PE自封袋,塑料方底袋,塑料包装袋厂家

丹阳康耐包装材料有限公司是一家专业生产塑料包装袋等产品的企业,主要产品有:PE袋、PE自封袋、塑料方底袋、PE塑料包装袋等,欢迎新老客户前来咨询合作。

星空体育·(中国)官方网站

星空体育·(中国)官方网站_StarSkySports简称“星空体育”是亚洲顶级的在线游戏平台,星空体育平台官网面向全世界平台、登录、网站、网址、娱乐。选择星空体育,体验前所未有的体育盛宴。

武汉办公室磨砂膜

【企业推荐】武汉武卓玻璃贴膜公司电话:13797063828(同微信号);是武汉建筑玻璃贴膜领导者,专业从事武汉隔热膜、武汉单向透视膜、武汉安全防爆膜,武汉防晒膜、武汉办公室磨砂膜,武汉装饰膜等;服务于武汉办公室玻璃贴膜、家居玻璃贴膜、幕墙玻璃贴膜,窗户玻璃贴膜等;还可以满足客户的各种玻璃膜定制。湖北省各地都有专业的施工队伍。欢迎您的来电。

武汉祥远复合材料有限公司

武汉祥远复合材料有限公司是一家集玻璃钢设计、研发、制造为一体的技术型企业,有着多年的真空导入工艺经验,能根据客户的要求来设计产品的成型工艺(轻质RTM、拉挤、手糊)、模具的设计开发以及原材料的优选。 公司以武汉理工大学材料学院为技术支撑,以客户为导向,凭借着多年的设计、制造经验及专业人才优势,不断研发出市场前景可观、产品性能优越、客户满意度高的新型复合材料产品。 公司主要产品有:玻璃钢天线罩、玻璃钢浮体、玻璃钢造型、玻璃钢外壳、玻璃钢风管、玻璃钢汽车部件、碳纤维产品等。 “以人为本,诚信务实,创新致远”是公司不变的宗旨。 “严格的品质控制,完善的售后服务体系”是对顾客要求的保证。 武汉祥远复合材料有限公司热忱欢迎您

哈尼手游网

哈尼手游网为您提供各种最新最热的手机游戏攻略,手游副本攻略,手游关卡攻略,还提供了最全的新手攻略和各种手游技巧攻略,只为让您更懂手游!

洛克化工网

专业化工网站,洛克化工网提供化工化工产品数据库、化工供求信息、化工供应商、化工字典等栏目。

阳澄湖大闸蟹

阳澄湖莲花岛小宋蟹庄

甬悦儿童青少年心理诊疗基地

夜团是国内新兴的夜生活休闲服务平台,项目隶属于广东伊步网络科技有限公司,运营总部坐落于“中国硅谷”深圳。该项目起源于2020年,针对目前网约车/出租车司机经营成本过高的现状以及乘客夜间出行消费场景,为司机提供一个即能省钱又能赚钱的平台。夜团聚集了同城夜晚餐饮、娱乐、休闲等多个业态的商家,包括夜游、夜购、夜娱、夜食、夜享,让用户以优惠的价格享受夜生活,体会众多实惠和乐趣。

东楚网·黄石新闻网

湖北,黄石,东楚风云,在线新闻.东楚网是您了解黄石新闻最权威最专业的平台。全面整合黄石日报、东楚晚报、黄石电视台、黄石人民广播电台,大冶,阳新,下陆,开发铁山新闻资源.

一路狂奔科技(深圳)有限公司

一路狂奔科技(深圳)有限公司专注于无人机行业应用的技术研究,在环境监测、安监、消防领域有着广泛的经验。我们的产品服务包括无人机气体实时监测,气体采集,水体采集等,产品体积小,便携,一人就能完成工作任务。

全局底部横幅