核心概念

注意力机制

依据查询与键的匹配权重,对值向量作数据相关加权汇聚的神经网络操作。

是什么

注意力(attention)用查询QQ与键KK计算匹配,再以行归一化权重汇聚值VV。若Q,K∈Rn×dkQ,K\in\mathbb R^{n\times d_k}、V∈Rn×dvV\in\mathbb R^{n\times d_v},则

softmax⁡row(QKT/dk)V∈Rn×dv.\operatorname{softmax}_{\rm row}(QK^{\mathsf T}/\sqrt{d_k})V\in\mathbb R^{n\times d_v}.

QKTQK^{\mathsf T}为n×nn\times n;缩放用于控制点积随dkd_k增长的尺度。[1, §3.2, eq. (1)]

边界

掩码决定可访问位置。权重反映这一运算怎样混合值,不自动等于人类注意、因果贡献或事实来源。Transformer包含多头注意力,也包含位置表示、前馈层、残差与归一化,所以二者不是同义词。

在本书中

见第14章14.6。本页为编辑初稿,独立事实复核待完成。

本页参考来源

  1. Vaswani A, Shazeer N, Parmar N, et al.. Attention Is All You Need [C]. NeurIPS 2017, 2017. arXiv:1706.03762v7. 定位:§3.2, especially eq. (1). 稳定来源(访问 )。

相关概念