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的"智能",就是通过反向传播,不断调整矩阵里的数字,让"内积"和"外积"的计算结果恰好符合预期。

相关推荐
可爱系程序猿7 小时前
ARM电脑安装打印机驱动失败:先辨认安装器、系统架构与打印路径
arm开发·程序人生·电脑
这个DBA有点耶8 小时前
时间序列数据库选型2026:5款主流产品深度对比与场景适配
大数据·数据库·程序人生·架构·时序数据库·dba·数据库管理员
ThornArmor1 天前
奔腾的呼吸:被驱逐者的灵感
开发语言·程序人生·架构
郝学胜-神的一滴2 天前
Qt 高级编程 045:坐标体系深度实战
开发语言·c++·windows·python·qt·程序人生
HugoStudio_SWAN2 天前
洛谷 P10719 \[GESP202406 五级] 黑白格——暴力美学与图像处理的最小外接矩形
c++·图像处理·人工智能·学习·程序人生·算法·目标跟踪
HugoStudio_SWAN3 天前
洛谷 B4450 / B3867 / B3923 智慧购物、储蓄与做题——从程序到生活
c++·学习·程序人生·算法·生活
职场的momo3 天前
字节生活服务海量内推,挑战亿级订单与AI交易中台
人工智能·程序人生·面试·职场和发展·跳槽·生活·业界资讯
CoderYanger3 天前
前端基础——JavaScript(基础语法)(下篇)
java·开发语言·前端·javascript·程序人生·面试·职场和发展
rust&python3 天前
课程表的归一化处理
python·程序人生·信息可视化
sxjk19873 天前
GPON带宽共享与忙时拥塞机制深度解析
程序人生