更改
跳到导航
跳到搜索
←上一编辑
下一编辑→
RWKV模型
(查看源代码)
2025年6月8日 (日) 21:50的版本
添加835字节
、
2025年6月8日 (日) 21:50
无编辑摘要
第919行:
第919行:
核心思想: Transformer 架构的核心是自注意力机制(Self-Attention),它允许模型在处理序列中的每个元素时,同时考虑序列中的所有其他元素,并根据它们之间的相关性分配不同的权重。
核心思想: Transformer 架构的核心是自注意力机制(Self-Attention),它允许模型在处理序列中的每个元素时,同时考虑序列中的所有其他元素,并根据它们之间的相关性分配不同的权重。
+
Transformer 中的注意力机制可表示为:
+
:<math>\text{Attention}(Q, K, V) = \text{softmax}\left( \frac{QK^\top}{\sqrt{d_k}} \right) V</math>
+
其中:
+
* <math>Q</math>:查询(Query)
+
* <math>K</math>:键(Key)
+
* <math>V</math>:值(Value)
+
* <math>d_k</math>:Key 的维度
主要特点:
主要特点:
第928行:
第935行:
==== RWKV 架构 ====
==== RWKV 架构 ====
核心思想: RWKV 旨在结合RNN的效率和Transformer的性能,它将Transformer的并行训练能力与RNN的常数时间推理能力相结合,实现了一种“线性的注意力机制”。
核心思想: RWKV 旨在结合RNN的效率和Transformer的性能,它将Transformer的并行训练能力与RNN的常数时间推理能力相结合,实现了一种“线性的注意力机制”。
+
+
RWKV 模拟注意力计算的一种核心结构是 WKV(Receptance Weighted Key Value),WKV 计算可简化为如下形式:
+
:<math>
+
x_{t+1} = \sigma(Rx_t) \cdot \frac{\exp(Kx_t) \cdot (Vx_t) + \exp(W) \cdot a_t}{\exp(Kx_t) + \exp(W) \cdot b_t}
+
</math>
+
其中:
+
* <math>x_t</math>:当前输入向量
+
* <math>R</math>、<math>K</math>、<math>V</math>:线性投影矩阵
+
* <math>\sigma</math>:激活函数(如 sigmoid)
+
* <math>W</math>:时间权重参数
+
* <math>a_t</math>、<math>b_t</math>:递归累加的历史值
主要特点:
主要特点:
RWKV
201
个编辑
导航菜单
个人工具
登录
名字空间
页面
讨论
变种
视图
阅读
查看源代码
查看历史
更多
搜索
导航
集智百科
集智主页
集智斑图
集智学园
最近更改
所有页面
捐赠我们
帮助
工具
特殊页面
可打印版本