多头注意力机制

前言

上期说到了自注意力机制,也叫做单头注意力机制。单头注意力只有一套,它只能产生一种注意力分布,如果所有信息都挤在一个注意力空间里,不同类型的关系容易互相干扰。

举个例子:

假设有这样一句话:

这个动物过不了马路,因为它太累了

这里的it指的是谁?

  • 可能关注 animal ↔ tired (理解代词指代)
  • 也可能关注 cross ↔ street (理解动作关系)

在单头注意力机制中:

它一次只能"重点看一种关系"

如果所有信息都挤在

多头注意力怎么做

假设有8个头(8 heads)。

输入 X 后,不再只学习一组头投影矩阵,而是学习8组

表示从不同角度观察这些词的关系,例如:代词关系,动作关系等...

...

每个头独立计算:

最后拼接:

每个头学什么?

训练后,不同头往往会自动分工。

例如:

The little boy who was wearing a red hat kicked the ball.

可能出现:

  • 头1:关注主谓关系
    • boy -> kicked
  • 头2:关注修饰关系
    • boy -> little
  • 头3:关注局部邻近词
    • wearing -> a red hat
  • 头4:关注长距离依赖
    • boy -> hat

总结

多头注意力的核心目的是:

让模型在不同表示子空间中并行的关注不同类型的信息关系,从而获得比单头注意力更强的表达能力

如果把一句话理解成观察一个人:

  • 单头注意力:只有一个摄影师,只能选择一个角度拍照。
  • 多头注意力:有很多摄影师同时拍照:
    • 有人拍正面;
    • 有人拍侧面
    • 有人拍全身;
    • 有人拍细节

最后这些照片综合起来,你会对这个人理解更加完整。

相关推荐
shijuejianceliu28 分钟前
传统规则视觉与嵌入式AI智能相机的技术差异
人工智能·嵌入式硬件·视觉检测·相机
goujunwe1 小时前
GEO 内容的“可摘录性”打磨技巧:段落、表格、小标题怎么写,方便大模型提取独立答案块
大数据·人工智能
羲云网络1 小时前
GPT Image 2.6 接入指南:Flare、Sunburst 怎么选,编辑失败如何排查
人工智能·python·aigc·图像生成·acedatacloud
张张123y1 小时前
DSH 插件调试避坑:client.js 设置面板、热重载与高频报错排查表
开发语言·javascript·人工智能·开源·ecmascript·后端开发·ai agent
Navigator_Z3 小时前
LeetCode //C - 1317. Convert Integer to the Sum of Two No-Zero Integers
c语言·算法·leetcode
tellmewhoisi3 小时前
深度学习0:数学知识:正态分布4(坐标变换的缩放比例(面积微元变换))
人工智能·深度学习
D e re k ZH3 小时前
C++ 标准库算法全景指南
数据结构·c++·算法
布吉岛的石头3 小时前
Java 程序员第 50 阶段14:灾难性遗忘与回放
人工智能·深度学习·transformer
不吃鱼的羊3 小时前
诊断27解锁dll文件制作(Visual Stdio无法安装版)
人工智能