1.1 注意力机制的数学基础 注意力机制作为现代深度学习的核心组件,其理论基础建立在数学的多个分支之上。本节将深入探讨注意力机制的数学原理,从线性代数、概率论到信息论,为后续的内容奠定坚实的理论基础。 1.1.1 注意力的数学定义 从数学角度来看,注意力机制本质上是一种加权求和的操作。给定查询向量Q、键向量K和值向量V,注意力输出的计算公式为: 其中,$dk$表示键向量的维度,这个缩放因子是为了防止点积过大导致softmax函数梯度消失。
注意力机制作为现代深度学习的核心组件,其理论基础建立在数学的多个分支之上。本节将深入探讨注意力机制的数学原理,从线性代数、概率论到信息论,为后续的内容奠定坚实的理论基础。
从数学角度来看,注意力机制本质上是一种加权求和的操作。给定查询向量Q、键向量K和值向量V,注意力输出的计算公式为:
\text{Attention}(Q, K, V) = \text{softmax}\left(\frac{QK^T}{\sqrt{d_k}}\right)V
其中,d_k表示键向量的维度,这个缩放因子是为了防止点积过大导致softmax函数梯度消失。