arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2310.01082cs.LGcs.AImath.OC
收藏

线性注意力(也许)就是你所需要的(理解Transformer优化)

Linear attention is (maybe) all you need (to understand transformer optimization)

MIT EECS(麻省理工学院电气工程与计算机科学系);MIT LIDS(麻省理工学院信息与决策系统实验室);KAIST(韩国科学技术院);KAIST ISysE(韩国科学技术院工业系统工程系);KAIST AI(韩国科学技术院人工智能学院);MIT CEE(麻省理工学院土木工程与环境工程系)

Kwangjun Ahn, Xiang Cheng, Minhak Song, Chulhee Yun, Ali Jadbabaie, Suvrit Sra

AI总结:

本研究通过线性化浅层Transformer模型重现训练动态,证明其可作为理解Transformer优化的有效抽象。

摘要:

Transformer训练以困难著称,需要精心设计优化器并使用各种启发式方法。我们通过仔细研究一个简单但规范的线性化浅层Transformer模型,在理解训练Transformer的微妙之处方面取得了进展。具体来说,我们训练线性Transformer来解决回归任务,灵感来自J. von Oswald等人(ICML 2023)和K. Ahn等人(NeurIPS 2023)的工作。最重要的是,我们观察到我们提出的线性化模型能够重现Transformer训练动态的几个显著方面。因此,本文获得的结果表明,一个简单的线性化Transformer模型实际上可能是理解Transformer优化的一个有价值的、现实的抽象。

英文摘要:

Transformer training is notoriously difficult, requiring a careful design of optimizers and use of various heuristics. We make progress towards understanding the subtleties of training Transformers by carefully studying a simple yet canonical linearized shallow Transformer model. Specifically, we train linear Transformers to solve regression tasks, inspired by J.~von Oswald et al.~(ICML 2023), and K.~Ahn et al.~(NeurIPS 2023). Most importantly, we observe that our proposed linearized models can reproduce several prominent aspects of Transformer training dynamics. Consequently, the results obtained in this paper suggest that a simple linearized Transformer model could actually be a valuable, realistic abstraction for understanding Transformer optimization.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑