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

相关推荐
雷帝木木24 分钟前
数据湖与数据仓库:从理论到实践
人工智能·python·深度学习·机器学习
小小测试开发28 分钟前
AI Agent 评测确定性:snapshot → fork → act → assert → diff 的状态孪生测试世界
人工智能
咖啡星人k43 分钟前
2025 AI编程进入“自动驾驶“时代:我用MonkeyCode把Agent、MCP和AI原生工作流跑通了
人工智能·自动驾驶·prompt·aigc·ai编程·ai-native
jike_20261 小时前
4款会议翻译转写工具对比:多语言、方言和线下会议怎么选?
人工智能·语音识别·iphone
闻道且行之1 小时前
图片处理助手|泊松融合原理 + C++ 工程实现,seamlessClone 三模式一次讲透
数据库·c++·人工智能·opencv
ages_1231 小时前
AI销售手机技术架构深度解析:从MDM终端管控到LLM业务赋能的完整闭环
人工智能·智能手机·ai销售手机·ai拓客·ai员工手机·剪流ai员工手机
正经教主1 小时前
AI提示词工程(专家级)第26课:对抗性提示与模型安全
人工智能
码士集团小青1 小时前
YOLO:将AI Agents嵌入到IntelliJ IDEA
人工智能
yanghuashuiyue2 小时前
RNN结构记录
人工智能·rnn·深度学习
mennekes2 小时前
数据中心安全配电设备如何选择?
运维·人工智能·科技·安全·制造