HyperAIHyperAI

Command Palette

Search for a command to run...

利用反馈记忆解决Transformer的一些局限性

Angela Fan Thibaut Lavril Edouard Grave Armand Joulin Sainbayar Sukhbaatar

摘要

尽管Transformer架构属于前馈网络,却已在序列化、自回归任务中取得了成功。与循环神经网络不同,Transformer通过注意力机制在并行处理输入标记的同时捕捉时间依赖关系。尽管这种并行化设计提升了计算效率,却限制了模型对输入序列固有顺序特性的充分挖掘:当前层的表示只能访问较低层的表示,而无法利用已生成的高层抽象表示。针对这一局限,本文提出了一种名为反馈Transformer(Feedback Transformer)的新型架构,该架构将所有先前时刻的表示显式地传递给所有后续时刻的表示,使得当前时间步的最低层表示能够基于过去时刻的高层抽象表示进行构建。我们在语言建模、机器翻译和强化学习等多个基准任务上进行了实验验证,结果表明,这种增强的表示能力能够使结构更小、层数更浅的模型在性能上显著超越传统Transformer。


用 AI 构建 AI

从创意到上线——通过免费 AI 协同编码、开箱即用的环境和最优惠的 GPU 价格,加速您的 AI 开发。

AI 协同编码
开箱即用的 GPU
最优定价

HyperAI Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
利用反馈记忆解决Transformer的一些局限性 | 论文 | HyperAI超神经