掌握训练模型所需的核心基础概念

理解模型、训练、推理、过拟合、损失函数、优化器、学习率等关键术语。

基础概念与术语

什么是”模型”?

模型 = 一个数学函数,接收输入数据,输出预测结果。

  • 传统编程:我们写规则(if-else)→ 计算机执行
  • 机器学习:我们给数据+答案 → 计算机自己学习规则 → 新数据时自动预测
传统编程:  数据 + 规则 → 答案
机器学习:  数据 + 答案 → 规则(模型)

核心概念

🔹 训练(Training)

模型从数据中学习的过程。就像学生做题,不断纠正错误。

🔹 推理(Inference)

训练好的模型对新数据做预测。就像学生考试,用学到的知识解题。

🔹 数据集(Dataset)

  • 训练集 (Train Set):用来学习的题目
  • 验证集 (Validation Set):做题时模拟考试,调优用
  • 测试集 (Test Set):最终考试,评估真实水平
  • 过拟合 (Overfitting):题目背太熟,换道题就不会了
  • 欠拟合 (Underfitting):连题目都没学会

🔹 特征与标签

术语通俗理解举例(识别猫)
特征 (Feature)输入数据图片像素值
标签 (Label)正确答案”猫” 或 “不是猫”
样本 (Sample)一条数据一张图片

🔹 Epoch / Batch / Iteration

术语含义类比
Epoch完整看完一遍所有数据把一本书从头读到尾
Batch Size每次看多少页再总结一次看 32 页
Iteration看一个 batch 算一次每 32 页翻一次

公式Iterations per epoch = 总样本数 ÷ Batch Size

🔹 损失函数(Loss Function)

衡量模型预测得有多离谱。数值越小越好。

  • 分类任务 → Cross-Entropy Loss
  • 回归任务 → Mean Squared Error (MSE)

🔹 优化器(Optimizer)

根据 Loss 调整模型参数的算法。

  • SGD:最基础的梯度下降
  • Adam:最常用,自适应学习率
  • AdamW:Adam + 权重衰减改进版

🔹 学习率(Learning Rate)

每次更新参数时”迈的步子”大小:

  • 太大 → 训练震荡,不收敛
  • 太小 → 训练太慢,陷入局部最优

🔹 梯度消失 / 梯度爆炸

  • 梯度消失:深层网络前面层学不动了
  • 梯度爆炸:梯度值暴涨,参数直接崩掉

📌 一张图记住训练流程

数据准备 → 搭建模型 → 前向传播(预测) 
         → 计算 Loss → 反向传播(求梯度)
         → 优化器更新参数 → 回到前向传播循环
         → 直到 Loss 收敛 → 模型保存

🔗 下一篇

环境搭建与工具链