论文阅读——RetNet

transformer的问题:计算量大,占用内存大,不好部署。

所以大家在找能解决办法,既能和transformer表现一样好,又能在推理阶段计算复杂度很低。

这些方法大概分类三类:一是代替transformer非线性注意力机制的线性注意力,二是牺牲并行训练,但是推理效率高的循环模型,三是寻找一种其他机制代替注意力机制。但是都不成功。

RetNet整体结构:

X是每层的输入序列,LN是LayerNorm

MSR:multi-scale retention

RetNet是L个单独模块堆叠,每个模块包含MSR和FFN两部分。

考虑循环模型序列建模问题,可以表示为:

其中,Sn是隐层,Vn是输入。

By absorbing A into WQ and WK,把方程写为:

γ简化为标量:

retention layer定义为:

相关推荐
夕小瑶几秒前
字节把 PS 做进了生图模型里,实测 Seedream 5.0 Pro 指哪改哪
人工智能
量化吞吐机4 分钟前
2026年AI量化工具评估,先分开学习开发和验证
人工智能·python
IvorySQL5 分钟前
PG 日报|新增 VFD 缓存监控视图,精细化数据库调优
大数据·数据库·人工智能·缓存·postgresql·区块链
小白狮ww6 分钟前
告别「机械念经」:TADA-1b 让模型「说人话」
人工智能·语音识别·语音生成·ai声音
咏方舟【长江支流】6 分钟前
【开源】跨语言·跨平台·跨数据库(3) —— 为用宝框架增加ORM,让AI执行
数据库·人工智能·开源·ai编程·orm·咏方舟-长江支流·用宝框架
AliCloudROS7 分钟前
一次真实录屏:我只说每月别超过 200 块,小程序后端就搭好了
运维·人工智能·云计算
code_pgf7 分钟前
具身智能的核心组成部分及其作用
人工智能·机器人
长风2308 分钟前
Day 16:大屏呈现——管理层视角 Dashboard 设计
人工智能·安全
User_芊芊君子12 分钟前
实测参赛|用飞算JavaAI快速落地企业项目招投标管理系统
人工智能·ai·amd
深海鱼在掘金13 分钟前
深入浅出RAG——第2章:文本嵌入 —— 让计算机理解语义的桥梁
人工智能