AI中的数学一

把AI计算里最核心的数学概念串起来。

在AI(深度学习)中,"积" 这个概念贯穿始终。把AI中常见的"积"分成三大类 来梳理:基础代数积矩阵乘法积神经网络专用积


第一类:基础代数积(两个向量的运算)

这是最底层的定义,也是AI论文里最常提到的。

1. 内积(Inner Product / Dot Product)
  • 定义 :两个同长度的向量,对应位置元素相乘,再全部加起来。

  • 公式:A⃗⋅B⃗=∑iAi×BiA⋅B=∑i​Ai​×Bi​

  • 几何意义:衡量两个向量的"相似度"或"夹角余弦"(投影长度)。

  • AI中的应用无处不在 。全连接层(y = Wx)的计算、卷积的滑动窗口计算、Transformer中的注意力分数(Q·K)计算,本质上都是内积。

2. 外积(Outer Product)
  • 定义 :两个长度可以不同 的向量(假设长度m和n),A的每一个元素分别去乘B的每一个元素,生成一个 m行 × n列 的矩阵。

  • 公式:(A⃗⊗B⃗)ij=Ai×Bj(A⊗B)ij​=Ai​×Bj​

  • 几何意义:生成一个平面(秩1矩阵)。

  • AI中的应用

    • RNN/LSTM 中的某些门控计算。

    • 低秩分解(LoRA微调大模型):用两个小矩阵的外积来近似一个大矩阵的更新量,大幅减少参数量。

    • 在GPU底层(如Tensor Core),大量使用外积累加(Outer Product Accumulation)来并行计算矩阵乘法。


第二类:矩阵乘法积(涉及二维数组)

这是AI训练和推理时的计算主体

3. 矩阵乘积(Matrix Product / GEMM)
  • 定义:C=A×BC=A×B。要求A的列数等于B的行数。

  • 结果 :结果C中每个元素 CijCij​,是A的第i行B的第j列内积

  • AI中的应用 :全连接层、卷积层(通过im2col转化后)、Transformer的FFN(前馈网络)。AI 99%的计算量都在跑GEMM。

4. 逐元素乘积(Hadamard Product / Element-wise Product)
  • 定义 :两个形状完全一样的矩阵,对应位置的数字直接相乘,结果形状不变。

  • 公式:(A∘B)ij=Aij×Bij(A∘B)ij​=Aij​×Bij​

  • AI中的应用

    • 注意力机制中的Mask(掩码)操作。

    • 残差连接中的门控机制(Gating)。

    • Batch Normalization 中的缩放(乘以gamma)和平移。


第三类:神经网络专用积(特殊运算)

这些是AI独有的,不再局限于纯数学定义。

5. 卷积(Convolution / Cross-correlation)
  • 定义(AI版) :用一个小的卷积核(Kernel),在大的输入矩阵上滑动 ,每一步做对应位置的逐元素相乘 ,然后求和(即内积),得到一个输出值。

  • 本质 :带权重共享局部连接的滑动内积。

  • AI中的应用:处理图像(CNN)、音频、序列数据。

6. 克罗内克积(Kronecker Product)
  • 定义 :用矩阵B去替换矩阵A中的每一个元素。如果A是m×n,B是p×q,结果是一个 (mp) × (nq) 的大矩阵。

  • AI中的应用

    • 模型压缩:用克罗内克积构造大矩阵,减少参数量。

    • 神经正切核(NTK) 理论分析中常用。


总结一览表(帮你快速回忆)

名称 输入形状 计算方式 输出形状 AI典型场景
内积 (Dot) 向量 (n,) 和 (n,) 对应乘,再求和 一个数字(标量) 注意力分数、全连接单神经元
外积 (Outer) 向量 (m,) 和 (n,) 每个每个 矩阵 (m × n) LoRA微调、底层硬件加速
矩阵乘 (GEMM) 矩阵 (m×k) 和 (k×n) 行与列的内积 矩阵 (m × n) AI计算主力(全连接、FFN)
逐元素乘 (Hadamard) 矩阵 (m×n) 和 (m×n) 对应位直接乘 矩阵 (m × n) 注意力Mask、门控机制
卷积 (Conv) 矩阵和小的核 滑动窗口做内积 矩阵(特征图) 图像分类、目标检测

终极硬核总结(划重点)

如果去翻NVIDIA的CUDA手册或AI框架的底层源码,会发现:

  • 硬件只认识"乘加(MAC)" :无论是内积、外积、GEMM,底层都是 a*b + c 这条指令。

  • AI训练的反向传播(BP) :本质就是在疯狂计算雅可比矩阵(Jacobian),而雅可比矩阵的计算,又可以被拆解成上述各种"积"的组合。

所以,可以这样记:AI的"智能",就是通过反向传播,不断调整矩阵里的数字,让"内积"和"外积"的计算结果恰好符合预期。

相关推荐
郝学胜-神的一滴1 天前
Python 高级编程 026:序列内核深剖
开发语言·python·程序人生·软件工程
CoderYanger3 天前
A.每日一题:1979. 找出数组的最大公约数
java·程序人生·算法·leetcode·面试·职场和发展·学习方法
个 人 练 习 生3 天前
strcmp与strstr函数的模拟实现
c语言·开发语言·经验分享·学习·程序人生
xiaoyustudiowww4 天前
白盒测试和黑盒测试一点个人观点
程序人生
可触的未来,发芽的智生4 天前
发现-元认知技能,激起神经符号系统跃变
javascript·人工智能·python·程序人生·自然语言处理
星栈独行5 天前
翻完 Pi 源码:它和 Codex、Claude Code 有何不同
开发语言·javascript·人工智能·程序人生