LLM学习笔记(7)Scaled Dot-product Attention

虽然 Attention 有许多种实现方式,但是最常见的还是 Scaled Dot-product Attention。

ChatGPT 页面 来类比说明

Query 类比:

  • 在当前 ChatGPT 页面中,Query 就是你输入的问题或请求。
  • 例如,当你输入一句话 "Query 和 Key 是什么?",这相当于提出了一个查询(Query),它表示你希望从模型中获取关于 "Query 和 Key" 的相关信息。

Key 类比:

  • Key 就像是 ChatGPT 的知识索引,存储着大量可能的回答的"入口"或"标签"。
  • 比如,模型内部已经存储了与 "Query" 和 "Key" 概念相关的所有信息,它们以一种组织良好的方式被表示为一组"键"(Key)。这些键是用于索引知识点的。

Value 类比:

  • Value 是与这些 Key 索引相关联的实际内容或信息,也就是答案的实际内容。
  • 在模型中,每个 Key 都对应具体的 Value,例如,Key "Query 和 Key" 对应了一个解释它们概念的文本内容。

类比流程

  1. 你输入问题:

    • 当你输入 "Query 和 Key 是什么?" ,这一输入作为 Query 被发送到模型。
  2. 模型检索 Key:

    • ChatGPT 会将你的 Query 与它的知识库中的所有 Key 进行比较,找到最相关的 Key。例如,"Query" 和 "Key" 可能匹配到模型内部的一些知识点(Key):
      • Key1: "Attention 机制的 Query 和 Key 定义"
      • Key2: "Transformer 中的 Query 和 Key 应用"
  3. 计算相关性:

    • 模型会通过点积计算你的 Query 和每个 Key 的相似度,决定这些 Key 与 Query 的相关程度。相似度越高,模型就越认为这个 Key 相关。
  4. 生成答案:

    • 根据 Key 的相关性分配权重,从相应的 Value(知识内容)中提取和整合信息,生成一个最终的答案作为回复。

Transformer 模型中,QueryKey 是用于计算注意力权重的两个重要概念。它们的设计来源于信息检索系统的思想,其中 Query 表示检索请求,Key 表示被检索的索引。通过比较 QueryKey 的相似性,模型决定哪些信息(由 Value 提供)更相关。

Query 和 Key 的工作原理

  1. 相似性计算:

    • QueryKey 的点积表示它们的相似性程度。
    • 相似性越大,表示当前 Query 更应该关注对应 Key 位置的 Value
  2. 注意力权重计算:

    • 点积结果经过缩放和 Softmax 归一化后,生成注意力权重 wijw_{ij}wij,表示 Query_iKey_j 的关注程度。
  3. 结合 Value:

    • 最终注意力权重 wijw_{ij}wij 会作用于 Value,将所有位置的 Value 加权求和,生成新的上下文表示。
相关推荐
OAFD.5 分钟前
机器学习之线性回归:原理、实现与实践
人工智能·机器学习·线性回归
SHIPKING3932 小时前
【机器学习&深度学习】LMDeploy的分布式推理实现
人工智能·深度学习
mit6.8242 小时前
[RestGPT] docs | RestBench评估 | 配置与环境
人工智能·python
CareyWYR3 小时前
每周AI论文速递(250818-250822)
人工智能
门思科技3 小时前
LoRaWAN 的网络拓扑全解析:架构、原理与应用实践
服务器·网络·人工智能·科技·物联网·架构
兔子的倔强3 小时前
Transformer在文本、图像和点云数据中的应用——经典工作梳理
人工智能·深度学习·transformer
焊锡与代码齐飞4 小时前
嵌入式第三十五课!!Linux下的网络编程
linux·运维·服务器·开发语言·网络·学习·算法
省四收割者4 小时前
Go语言入门(10)-数组
数据结构·经验分享·笔记·vscode·算法·golang
lxmyzzs5 小时前
【图像算法 - 21】慧眼识虫:基于深度学习与OpenCV的农田害虫智能识别系统
人工智能·深度学习·opencv·算法·yolo·目标检测·计算机视觉
Gloria_niki5 小时前
机器学习之K 均值聚类算法
人工智能·机器学习