清晨好,您是今天最早来到科研通的研友!由于当前在线用户较少,发布求助请尽量完整的填写文献信息,科研通机器人24小时在线,伴您科研之路漫漫前行!

Token Statistics Transformer: Linear-Time Attention via Variational Rate Reduction

安全性令牌 统计 还原(数学) 变压器 数学 计算机科学 计量经济学 电气工程 计算机网络 工程类 几何学 电压
作者
Ziyang Wu,Tianjiao Ding,Yifu Lu,Druv Pai,Jingyuan Zhang,Weida Wang,Yaodong Yu,Yi Ma,Benjamin D. Haeffele
出处
期刊:Cornell University - arXiv
标识
DOI:10.48550/arxiv.2412.17810
摘要

The attention operator is arguably the key distinguishing factor of transformer architectures, which have demonstrated state-of-the-art performance on a variety of tasks. However, transformer attention operators often impose a significant computational burden, with the computational complexity scaling quadratically with the number of tokens. In this work, we propose a novel transformer attention operator whose computational complexity scales linearly with the number of tokens. We derive our network architecture by extending prior work which has shown that a transformer style architecture naturally arises by "white-box" architecture design, where each layer of the network is designed to implement an incremental optimization step of a maximal coding rate reduction objective (MCR$^2$). Specifically, we derive a novel variational form of the MCR$^2$ objective and show that the architecture that results from unrolled gradient descent of this variational objective leads to a new attention module called Token Statistics Self-Attention (TSSA). TSSA has linear computational and memory complexity and radically departs from the typical attention architecture that computes pairwise similarities between tokens. Experiments on vision, language, and long sequence tasks show that simply swapping TSSA for standard self-attention, which we refer to as the Token Statistics Transformer (ToST), achieves competitive performance with conventional transformers while being significantly more computationally efficient and interpretable. Our results also somewhat call into question the conventional wisdom that pairwise similarity style attention mechanisms are critical to the success of transformer architectures. Code will be available at https://github.com/RobinWu218/ToST.
最长约 10秒,即可获得该文献文件

科研通智能强力驱动
Strongly Powered by AbleSci AI
科研通是完全免费的文献互助平台,具备全网最快的应助速度,最高的求助完成率。 对每一个文献求助,科研通都将尽心尽力,给求助人一个满意的交代。
实时播报
gmc完成签到 ,获得积分10
4秒前
搬砖的化学男完成签到 ,获得积分0
19秒前
27秒前
32秒前
zw完成签到,获得积分10
48秒前
刘辰完成签到 ,获得积分10
1分钟前
wxyinhefeng完成签到 ,获得积分10
1分钟前
打打应助舒心的南珍采纳,获得10
1分钟前
MOF完成签到 ,获得积分10
1分钟前
糖宝完成签到 ,获得积分10
1分钟前
我睡觉的时候不困完成签到 ,获得积分10
1分钟前
小马甲应助xun采纳,获得10
1分钟前
1分钟前
有魅力荟发布了新的文献求助10
1分钟前
缥缈的闭月完成签到,获得积分10
1分钟前
echo完成签到 ,获得积分10
2分钟前
回首不再是少年完成签到,获得积分0
2分钟前
jasmine完成签到 ,获得积分10
2分钟前
吉祥高趙完成签到 ,获得积分10
3分钟前
小小果妈完成签到 ,获得积分10
3分钟前
奶糖喵完成签到 ,获得积分10
3分钟前
hbuhfl完成签到 ,获得积分10
3分钟前
一彤展翅完成签到,获得积分10
3分钟前
3分钟前
xun发布了新的文献求助10
3分钟前
ally完成签到,获得积分10
3分钟前
li完成签到 ,获得积分10
4分钟前
kenchilie完成签到 ,获得积分10
4分钟前
属实有点拉胯完成签到 ,获得积分10
4分钟前
apckkk完成签到 ,获得积分10
4分钟前
4分钟前
SciKid524完成签到 ,获得积分10
4分钟前
chemlixy完成签到 ,获得积分10
5分钟前
cstp完成签到,获得积分10
5分钟前
mark33442完成签到,获得积分10
5分钟前
小美酱完成签到 ,获得积分10
5分钟前
淡然藏花完成签到 ,获得积分10
5分钟前
苏su完成签到 ,获得积分10
5分钟前
完美梨愁完成签到 ,获得积分10
5分钟前
苗条的小蜜蜂完成签到 ,获得积分10
5分钟前
高分求助中
Continuum Thermodynamics and Material Modelling 3000
Production Logging: Theoretical and Interpretive Elements 2700
Les Mantodea de Guyane Insecta, Polyneoptera 1000
Structural Load Modelling and Combination for Performance and Safety Evaluation 1000
Conference Record, IAS Annual Meeting 1977 820
電気学会論文誌D(産業応用部門誌), 141 巻, 11 号 510
Typology of Conditional Constructions 500
热门求助领域 (近24小时)
化学 材料科学 生物 医学 工程类 有机化学 生物化学 物理 纳米技术 计算机科学 内科学 化学工程 复合材料 基因 遗传学 物理化学 催化作用 量子力学 光电子学 冶金
热门帖子
关注 科研通微信公众号,转发送积分 3571317
求助须知:如何正确求助?哪些是违规求助? 3141926
关于积分的说明 9444838
捐赠科研通 2843331
什么是DOI,文献DOI怎么找? 1562830
邀请新用户注册赠送积分活动 731326
科研通“疑难数据库(出版商)”最低求助积分说明 718524