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。

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

论文地址:

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


本文地址: https://www.gpxz.com/article/be0e91d84d8569e6a9a1.html
全局中部横幅
全局中部横幅
UC浏览器下载2024安卓最新版

【豌豆荚】为你提供UC浏览器APP安卓版下载,历史版本、旧版下载,查看2024最新UC浏览器手机版介绍、应用截图、网友评论,方便快捷的将安卓版UC浏览器应用免费下载到手机。

乐享厨卫清洁的日式美学

花王(上海)产品服务有限公司(董事长 西口徹)宣布,花王(中国)于2020年7月下旬全新推出花王厨卫清洁系列——“花王厨房重油去污清洁泡沫(薄荷清香)...

NSK中空轴丝杠

搜索NSK中空轴丝杠?我们提供高精密、高速度的NSK中空轴丝杠解决方案,满足您运动控制需求。联系我们获取报价和技术支持。

义乌市敬域电子商务商行

辩论不休汽车网以服务全球华人汽车用户为己任,提供最新买车汽车报价、汽车图片、现车购买及互动论坛,第一时间发布汽车新车、试驾、评测、用车等丰富内容,是面向人群最广泛的专业门户汽车网站

铝单板生产加工厂家

是一家专业销售生产室外铝单板,铝幕墙,氟碳铝单板,冲孔铝单板,造型铝单板,双曲铝单板,石纹铝单板,木纹铝单板,等优质室内外幕墙装饰材料和室内金属铝质天花吊顶、金属吊顶、吸音吊顶、异型天花、轻钢龙骨、人字龙骨、铝压纹板、铝拉网、木纹吊顶、铝蜂窝板、铝型材、滚涂铝卷、集成吊顶、集成电器、工程筒灯,工程T5/T8格栅灯、LED灯等装饰工程配套材料的企业

嘉达鸭

嘉达鸭是一个深度探索手机科技的资讯门户,致力于苹果手机教学和资讯的平台,让每个用户第一时间能找到iPhone手机使用问题的图文教学和最新资讯。

钢丝网骨架管

公司始创于2016年6月22日,注册资金6000万人民币,专业从事钢丝网骨架塑胶(聚乙烯)复合管的研发、生产、销售、安装服务的现代化高科技术产业。主导产品钢丝网骨架塑胶(聚乙烯)复合管以其特有的防腐蚀、耐磨、耐高温、抗严寒、耐压、抗蠕变、稳定性好、安装方便等特点逐步取代各类金属管道和塑料管道,被广泛应用于石油、化工、燃气、供水、采矿、海水输送等领域。

夏季的风

一个默默无闻的主页

Unity中国官网

Unity是全球应用非常广泛的实时内容开发平台,为游戏、汽车、建筑工程、影视动画等广泛领域的开发者提供强大且易于上手的工具来创作、运营和变现3D、2DVR和AR可视化体验。

道法自然

收集分享无法计算的价值


全局底部横幅