梯度下降:神经网络如何学习

原标题Gradient Descent: How a Neural Network Learns

直观介绍如何最小化训练损失,并观察数字分类器学到了什么、没有学到什么。

LLLeisha Lee嘉宾
频道
3Blue1Brown
发布
2017年10月16日
视频
20:33
阅读
3 分钟
原文
英语
在 YouTube 打开
往下滚动阅读时,视频会缩到右下角继续播放。字幕跟随阅读语言。
阅读语言共 2 种语言可读
由 AI 从英语原文翻译。打开“对照原文”可以逐段核对。
摘要

30 秒看懂

Grant Sanderson 用手写数字网络解释梯度下降,然后检查其隐藏权重和局限。Leisha Lee 的访谈片段讨论记忆与泛化。

  1. 1损失函数示例对带标签训练图像的输出平方误差取平均。
  2. 2负梯度负梯度给出局部下降最快的方向。
  3. 3训练与泛化训练损失降低不保证每个样本或未见数据的表现都更好。
  4. 4对噪声也可能自信分类器可能对训练分布之外的输入给出自信预测。

要点建议

  1. 损失函数衡量网络输出与训练标签的差异。
  2. 梯度下降沿局部损失下降方向调整参数。
  3. 局部最小值不一定是全局最小值。
  4. 测试表现与内部表示需要分别检查。
  5. 网络可以拟合随机标签而无法将其泛化。
全文

按章节阅读整理后的全文

只看原始文稿
01

数字分类器

Grant Sanderson 用手写数字分类器介绍梯度下降,随后检查隐藏层实际学到了什么。示例使用 28×28 灰度像素的 MNIST 图像。

Grant Sanderson 用手写数字分类器介绍梯度下降,随后检查隐藏层实际学到了什么。示例使用 28×28 灰度像素的 MNIST 图像。

784 个输入神经元表示像素值。后续每个神经元用权重和偏置组合上一层的激活值,再应用激活函数。示例有两个各含 16 个神经元的隐藏层,约有 13,000 个可调整参数。

十个输出神经元对应十个数字,激活值最高者是预测结果。一种希望中的解释是早期层寻找边缘,后续层将边缘组合成环和笔画,但这个假设需要检验。

训练时提供带标签的数字图像,并调整权重与偏置。泛化能力则用训练时未见过的带标签图像单独评估。

02

定义损失

参数随机初始化时,最初输出通常很差。对于一个带标签的例子,Sanderson 将损失定义为输出激活值与目标值之差的平方和:正确数字的目标为 1,其他为 0。

参数随机初始化时,最初输出通常很差。对于一个带标签的例子,Sanderson 将损失定义为输出激活值与目标值之差的平方和:正确数字的目标为 1,其他为 0。

总体损失是训练集中各样本损失的平均值。它以网络权重和偏置为输入,输出一个衡量模型在这些样本上表现的数字。

03

寻找下降方向

为了理解如何降低损失,先想象只有一个变量的函数。在每个位置,斜率告诉你局部向下的方向;反复迈出小步可以接近局部最小值。

为了理解如何降低损失,先想象只有一个变量的函数。在每个位置,斜率告诉你局部向下的方向;反复迈出小步可以接近局部最小值。

起点很重要:复杂地形可能有许多谷底,最终到达的局部最小值不保证是全局最小值。步长也影响更新是否会越过谷底。

对于两个变量的函数,梯度指向局部上升最快的方向。负梯度指向下坡,其大小表示局部的陡峭程度。

04

参数空间中的梯度下降

同样的思想可以扩展到数千个参数:把权重与偏置排列成向量,计算损失梯度,再反复沿其相反方向移动一小步。这就是梯度下降。

同样的思想可以扩展到数千个参数:把权重与偏置排列成向量,计算损失梯度,再反复沿其相反方向移动一小步。这就是梯度下降。

梯度基于平均训练损失;降低平均值并不保证每个单独样本或未见数据上的表现都改善。反向传播高效计算这些更新所需的导数。

每个梯度分量指出参数应增大还是减小,以及局部损失对该参数有多敏感。若梯度为 (3, 1),在当前单位下第一坐标的局部导数是第二坐标的三倍。

05

测试与解释网络

Sanderson 报告说,这个简单的双隐藏层网络能正确分类约 96% 的未见测试图像,调整架构和其他设置还能改善结果。这是所述实验的结果,不是普遍保证。

Sanderson 报告说,这个简单的双隐藏层网络能正确分类约 96% 的未见测试图像,调整架构和其他设置还能改善结果。这是所述实验的结果,不是普遍保证。

随后他检验先前关于边缘的直觉。可视化学到的输入权重后,图案看起来大多像随机噪声,而不是整齐的边缘检测器。有效的分类器不一定学到设计者预想的表示。

面对随机噪声时,网络仍可能自信地预测某个数字。它的训练世界由居中的数字图像构成,而目标函数并未教它识别任意输入是否处于这个世界之外。

这个简单网络是理解后续模型的起点,并不能完整解释视觉智能。Sanderson 鼓励学习者尝试修改系统,并推荐 Michael Nielsen 免费提供的神经网络教材。

06

记忆与泛化

在访谈片段中,Leisha Lee 讨论一项研究:足够大的网络可以拟合随机打乱的训练标签,尽管这些标签无法支持对新样本的有效泛化。

在访谈片段中,Leisha Lee 讨论一项研究:足够大的网络可以拟合随机打乱的训练标签,尽管这些标签无法支持对新样本的有效泛化。

Lee 描述后续工作对随机标签与结构化标签的优化进行比较:在所引用的实验中,有意义的标签学得快得多。她还提到关于局部最小值质量的简化理论研究;这些结果并不意味着所有神经网络的最小值都一样好。

同频道的其他文章

20 张幻灯片40:44EN → 简体中文
3Blue1Brown· 2026年10月16日Grant Sanderson:现在是投身数学的最好时机Grant Sanderson, Constantin Kogler · 3Blue1Brown学生担心 AI 已经解决了数学。Grant Sanderson 却认为恰恰相反:即使机器证明定理的能力超过人类,数学这套制度依然有价值,而新人正迎来一个难得的机会,可以参与重塑它。MathematicsAI for scienceHuman-AI Collaboration+2
29 张幻灯片51:55EN → 简体中文
3Blue1Brown· 2026年9月18日AI 解不出的最后一道 IMO 题Grant Sanderson · 3Blue1BrownGrant Sanderson 带你走一遍 2025 年 IMO 第 6 题——前沿模型没做出来的铺砖谜题——以及要证明一种构造最优,需要怎样的视觉耐心。MathematicsAI researchReasoning & test-time compute+2
0:55EN → 简体中文
3Blue1Brown· 2026年7月24日64 块方糖谜题:六组相等距离一个 4×4×4 的方糖块有三种颜色;证明总能找到六对同色方糖,且每对距离相等。MathematicsEducation
33:51EN → 简体中文
3Blue1Brown· 2026年7月16日交叉熵连接了什么:压缩、语言模型与 KL 散度Grant Sanderson · 3Blue1BrownGrant Sanderson 从码长推导交叉熵,并用它解释下一个 token 训练和模型蒸馏。MathematicsMachine LearningFoundation Models+1
幻灯片2:25EN → 简体中文
3Blue1Brown· 2026年6月12日香农如何测量英语的信息熵一场猜字母游戏帮助克劳德·香农估算英语文本的信息量,也揭示了预测与压缩之间的联系。MathematicsAI History
幻灯片1:02EN → 简体中文
3Blue1Brown· 2026年6月10日月球机器人与最优二进制编码谜题四个方向出现的频率不同,每条指令最少需要多少比特?MathematicsScience Communication