视频封面

视频元信息

  • 视频标题:4.3 正确的矩阵维数(Getting Matrix Dimensions Right)
  • 所属课程:吴恩达深度学习专项课程 - 神经网络和深度学习(Neural Networks and Deep Learning)
  • 章节:第四周 - 深层神经网络(Deep Neural Networks)
  • 时长:约 11 分钟
  • 语言:英文讲解(中英字幕)

内容摘要(TL;DR)

本节讲解如何通过推导矩阵维度来验证深层神经网络前向传播代码的正确性。核心公式:W^[l] 的维度为 (n^[l], n^[l-1]),b^[l] 的维度为 (n^[l], 1)。在向量化实现中,Z 和 A 的维度从 (n^[l], 1) 变为 (n^[l], m),X 变为 (n^[0], m)。掌握维度规律可有效排查代码 bug。


一、回顾:前向传播的基本公式 [00:28]

前向传播的第一步是计算第一层的线性输出:

$$Z^{[1]} = W^{[1]} \cdot X + b^{[1]}$$

本节先忽略偏置项 b,重点分析参数矩阵 W 的维度。

理解:这是深层神经网络中每一层都要重复执行的核心运算,理解其维度关系是正确实现网络的前提。


二、示例网络结构定义 [00:48]

以一个 5 层神经网络为例:

层 类型 单元数 符号
第 0 层 输入层 2 n^[0] = n_x = 2
第 1 层 隐藏层 3 n^[1] = 3
第 2 层 隐藏层 5 n^[2] = 5
第 3 层 隐藏层 4 n^[3] = 4
第 4 层 隐藏层 2 n^[4] = 2
第 5 层 输出层 1 n^[5] = 1

注意:目前讨论的网络只有一个输出单元,后续课程会涉及多输出单元的情况。

思考:用上标 [l] 表示层号,n^[l] 表示第 l 层的单元数,这是吴恩达课程的标准记号体系,务必牢记。


三、单样本情况下各矩阵的维度推导 [01:40]

3.1 Z 的维度

Z^[1] 是第 1 层的激活向量(线性计算结果),维度为:

$$Z^{[1]} : (n^{[1]}, 1) = (3, 1)$$

3.2 X 的维度

X 是输入特征向量,维度为:

$$X : (n^{[0]}, 1) = (2, 1)$$

3.3 W^[1] 的维度推导

核心逻辑:需要找到一个矩阵 W^[1],使得它乘以 (n^[0], 1) 的向量后,得到 (n^[1], 1) 的向量。

由矩阵乘法规则:

$$(3, 2) \times (2, 1) = (3, 1)$$

因此:

$$W^{[1]} : (n^{[1]}, n^{[0]}) = (3, 2)$$

3.4 通用公式(重点)

考点:W^[l] 的维度 = (n^[l], n^[l-1]) 即:当前层的单元数 × 前一层的单元数

3.5 各层 W 的维度汇总

参数 维度 推导
W^[1] (3, 2) (n^[1], n^[0])
W^[2] (5, 3) (n^[2], n^[1])
W^[3] (4, 5) (n^[3], n^[2])
W^[4] (2, 4) (n^[4], n^[3])
W^[5] (1, 2) (n^[5], n^[4])

记忆技巧:W^[l] 的维度就是「下一层的维度,上一层的维度」——行数对应本层输出维度,列数对应本层输入维度。


四、偏置向量 b 的维度 [04:48]

4.1 推导逻辑

Z^[1] = W^[1]·X + b^[1],其中 Z^[1] 是 (3, 1),W^[1]·X 也是 (3, 1)。

要与 (3, 1) 的向量相加得到 (3, 1) 的结果,b^[1] 必须是 (3, 1)。

4.2 通用公式

考点:b^[l] 的维度 = (n^[l], 1) 即:当前层的单元数 × 1

参数 维度
b^[1] (3, 1)
b^[2] (5, 1)
b^[3] (4, 1)
b^[4] (2, 1)
b^[5] (1, 1)

五、反向传播中导数的维度一致性 [06:02]

易错点:dW 的维度必须与 W 相同,db 的维度必须与 b 相同。

  • dW^[l] : (n^[l], n^[l-1]) —— 与 W^[l] 一致
  • db^[l] : (n^[l], 1) —— 与 b^[l] 一致

理解:梯度下降更新参数时 W := W - α·dW,只有维度相同才能逐元素相减,这是维度检查的自然推论。


六、Z 与 A 的维度关系 [06:31]

因为激活函数 g 是逐元素(element-wise)应用的:

$$Z^{[l]} = g(A^{[l]})$$

所以 Z^[l] 和 A^[l] 必须具有相同的维度。

考点:Z^[l] 与 A^[l] 维度相同


七、向量化实现中的维度变化 [06:47]

7.1 单样本 vs 向量化的对比

量 单样本 向量化(m 个样本)
Z^[1] (n^[1], 1) (n^[1], m)
X (n^[0], 1) (n^[0], m)
W^[1] (n^[1], n^[0]) (n^[1], n^[0]) 不变
b^[1] (n^[1], 1) (n^[1], 1) 不变

7.2 向量化中的 Z^[1]

将每个样本的 z^[1](i) 水平堆叠:

$$Z^{[1]} = [z^{[1](1)}, z^{[1](2)}, \ldots, z^{[1](m)}]$$

维度变为 (n^[1], m)。

7.3 维度验证

$$(n^{[1]}, n^{[0]}) \times (n^{[0]}, m) = (n^{[1]}, m)$$

维度完全吻合。

7.4 Python 广播机制(Broadcasting)

易错点:b^[1] 在向量化中仍是 (n^[1], 1),但通过 Python 广播会自动复制为 (n^[1], m) 后逐元素相加。

理解:这是 NumPy 的广播特性——当 (n^[1], m) 与 (n^[1], 1) 相加时,后者沿水平方向自动扩展,无需手动复制。


八、向量化后 Z、A、X 的通用维度 [09:15]

量 维度
Z^[l] (n^[l], m)
A^[l] (n^[l], m)
A^[0] = X (n^[0], m)

特殊情形:当 l = 0 时,A^[0] 就是训练集输入特征 X,维度为 (n^[0], m)。


九、反向传播中的 dZ 和 dA [09:57]

与 Z 和 A 维度相同:

  • dZ^[l] : (n^[l], m)
  • dA^[l] : (n^[l], m)

理解:反向传播的梯度维度与前向传播的激活值维度一一对应,这是链式法则在矩阵形式下的自然结果。


十、本节核心价值与总结 [10:19]

核心提醒:实现深层神经网络时,始终保持对矩阵维度的追踪,可以帮你消除大量潜在 bug。

吴恩达强调,这是他自己写代码时常用的调试方法:

  1. 前向传播维度检查:W^[l] = (n^[l], n^[l-1]),b^[l] = (n^[l], 1)
  2. 反向传播维度检查:dW 和 W 同维度,db 和 b 同维度
  3. 向量化维度检查:Z、A、X 的第二维变为 m(训练样本数)
  4. Z 与 A 同维度:因为激活函数是逐元素运算

【拓展】维度检查的实战价值

  • 调试利器:在 NumPy 中可用 assert(W.shape == (n_l, n_prev)) 断言维度,快速定位 bug。
  • 常见错误:混淆 W 的行列顺序(写成 (n^[l-1], n^[l]))会导致矩阵乘法报错或结果维度错误。
  • 广播陷阱:Python 广播虽方便,但有时会"悄悄"修正维度不匹配的错误(如 (m,) 与 (m,1) 的差异),反而掩盖 bug。建议显式使用 reshape 确保维度正确。
  • 多输出网络:当输出层有多个单元时,n^[L] > 1,公式依然成立。

十一、下节预告 [10:56]

本节完成了前向传播机制的维度分析。下节将讨论:为什么深度神经网络如此有效?为什么深层表示比浅层表示更好?


关键标签 / 关键词

深度学习 #吴恩达 #神经网络 #矩阵维度 #前向传播 #反向传播 #向量化 #Python广播 #维度检查 #调试技巧 #DeepLearning #MatrixDimensions #ForwardPropagation #Vectorization #Broadcasting #Debugging


生成时间:2026-10-10 21:50:28 生成平台:VideoNote www.xygy.top