论文阅读——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定义为:

相关推荐
这就是佬们吗1 分钟前
Function Calling 还是 MCP?先分清它们根本不是同一层的东西
人工智能·prompt·ai编程
AIGCmagic社区3 分钟前
KITTI AbsRel从6.5压到5.4,Marigold V2用一张32GB卡把编辑DiT收成单步深度估计
人工智能·算法·aigc·ai多模态
Theo_xx4 分钟前
声学感知基础:7.Multipath 与 CFR
人工智能·无线感知·声学感知
论文复现现场5 分钟前
课程作业要跑 PyTorch 训练,学校机房不够用去哪租?云 GPU 选型、环境迁移与防丢数据指南
人工智能·pytorch·深度学习·云计算·gpu·cuda
A小码哥5 分钟前
开发转型AI Agent研发:拆解 Agent 的记忆系统、推理链路与决策机制
人工智能
艾莉丝努力练剑6 分钟前
【Git:综合复盘】Git 原理与使用
大数据·人工智能·git·elasticsearch·面试
HRaitest9 分钟前
AI招聘系统架构深度拆解:传统外挂式AI vs AI原生基座的本质差异与潜能边界
人工智能·ai·系统架构·视觉检测·求职招聘
hhzz9 分钟前
OpenCV 入门到精通 09】特征检测与匹配:从 Harris 到 ORB 图像配准
人工智能·opencv·计算机视觉·开源·交互
小淮AI10 分钟前
企业文件管理方案选型观察:从部署模式、协作权限与集成能力看三个方向
人工智能
TMT星球14 分钟前
曹操出行AI打车接入荣耀YOYO,AI出行服务首次登陆主流手机智能体
人工智能·科技