★ FEATURED ARTICLE 1. 为什么线性回归是深度学习的必修第一课先抛个观点如果你真的想把深度学习搞明白线性回归是绕不过去的第一块基石。我见过太多人一上来就抱着《动手学深度学习》啃CNN、Transformer结果连损失函数下降曲线都看不懂更别提调参了。原因很简单——深度学习里那些看似高深的东西本质上大部分都是线性回归的“套娃”版本。线性回归要做的事情非常朴素给定一堆数据点找到一条直线或者一个超平面来拟合它们之间的关系。放在深度学习的语境里这条直线就变成了神经网络的第一层权重拟合的过程就变成了梯度下降更新参数的过程。换句话说线性回归是深度学习里最基础、最简单、最容易理解的一个完整闭环数据、模型、损失函数、优化器、训练、评估你能在这个简单模型里跑通整套流程后面再看神经网络就不会发怵。适合谁来学两类人。一类是刚入门机器学习的新手想找一个能动手跑的模型练手另一类是准备进入深度学习、但被复杂的框架和术语吓到的人先用线性回归把“训练”这个核心动作刻进脑子里。这篇文章我会直接用Python把线性回归从数学原理到代码实现、再到和深度学习的联系完整走一遍最后还会聊聊我实际踩过的坑。2. 线性回归的核心思路与设计拆解2.1 线性回归到底在求解什么我们先从最直觉的场景说起。假设你开了一家奶茶店想研究“温度”和“奶茶销量”之间的关系。你记录了10天的数据温度30度时卖了80杯温度25度时卖了65杯温度20度时卖了50杯……你心里大概有个感觉温度越高卖得越多但具体多多少能不能用一个公式表达线性回归就是干这个事的。它假设自变量x温度和因变量y销量之间是线性关系也就是y w * x b这里的w叫权重weightb叫偏置bias。在这个例子里w可以理解成“温度每升高1度销量增加多少杯”b可以理解成“温度是0度时销量是多少”。你可能觉得这太简单了但稍微扩展一下就有意思了。如果x不止一个比如温度、湿度、是否是周末三个因素都影响销量那就变成多元线性回归y w1 * x1 w2 * x2 w3 * x3 b写成向量形式就是 y X * W b。到了这一步它和神经网络里的“全连接层”就已经非常接近了——神经网络的一层本质上就是一个多元线性回归只不过后面再叠加一个非线性激活函数而已。2.2 怎么判断一条直线好不好损失函数有了模型表达式下一个问题就是w和b取什么值最好这时候就需要一个衡量标准也就是损失函数。最常用的损失函数叫均方误差Mean Squared ErrorMSE。它的计算方式特别直观把所有样本的真实值y_true和预测值y_pred做差把差值平方消除正负号的影响同时放大大误差的惩罚对所有样本求平均。公式如下MSE (1/n) * Σ(y_true - y_pred)^2生活化理解就像你扔飞镖每次扔完量一下飞镖离靶心多远把所有的误差距离平方加总再求平均。误差越小说明你的投掷水平越高。同理MSE越小说明这条直线对数据的拟合越好。那么为什么用平方而不是绝对值因为平方函数是可导的后续做梯度下降需要用到导数绝对值在0点处不可导优化起来会麻烦。另一个原因是平方对大误差的惩罚更重这既有好处也有坏处——好处是模型会更努力地去拟合那些偏离大的点坏处是如果数据里有异常值模型会被带偏这个我在后面排查部分会讲。2.3 优化器选型梯度下降是核心引擎有了损失函数理论上我们可以用“暴力试错”来找最好的w和b随机生成一堆(w, b)组合计算每个组合对应的MSE取最小的那个。但这样效率太低尤其是参数多了以后组合呈指数爆炸。正确的做法是梯度下降Gradient Descent。核心思路是损失函数J(w, b)是一个关于w和b的曲面我们要沿着“地势最陡峭的方向”往下走直到走到最低点。每次更新参数的规则是w w - learning_rate * ∂J/∂w b b - learning_rate * ∂J/∂b这里的learning_rate学习率是步长决定每次走多大一步。步子太大容易在山谷两边来回震荡甚至直接冲出去步子太小则要很多步才能走到终点训练时间拉长。在线性回归里头我们的损失函数是一个凸函数也就是说这个曲面只有一个全局最低点没有局部最优的困扰。这比深度学习的非凸优化问题要简单得多但正因为它简单非常适合用来理解“梯度下降是在干嘛”这件事。等你理解了手动计算梯度再去理解深度学习里的反向传播会发现原理其实是同一个。2.4 闭式解 vs 迭代优化为什么深度学习只用迭代线性回归其实还有一个不需要迭代的“直接解法”——最小二乘法的解析解公式W (X^T * X)^(-1) * X^T * y也就是说只要数据量不大直接套这个公式一步到位就能求出最优参数。那为什么深度学习中从来不用这种解法两个原因。第一矩阵求逆的复杂度是O(n^3)当数据维度很大深度学习里动辄上百万参数时这个计算量完全不可行。第二深度学习模型不是线性的是层层嵌套的非线性函数根本不存在闭式解。所以必须用梯度下降这种迭代方式一步一步逼近最优解。但在线性回归这个环节两种方法我都建议你亲手跑一遍。跑闭式解能让你理解“数学公式是怎么变成代码的”跑梯度下降能让你理解“深度学习训练的核心循环是怎么运转的”。两条腿走路基础才扎实。3. 完整实操从零手写线性回归3.1 环境准备与数据构造我用的环境非常普通Python 3.9、NumPy、Matplotlib这几个就够了。不依赖任何深度学习框架这样才能看清楚核心逻辑。我们先造一组带线性关系的数据作为实验样本import numpy as np import matplotlib.pyplot as plt # 生成100个随机点x在0到10之间均匀分布 np.random.seed(42) X np.linspace(0, 10, 100) # 真实关系是 y 2*x 5加上高斯噪声模拟真实数据 true_w 2.0 true_b 5.0 y true_w * X true_b np.random.randn(100) * 2 # 可视化 plt.scatter(X, y, alpha0.7) plt.xlabel(x) plt.ylabel(y) plt.show()这里有个细节要注意我用了np.random.seed(42)来固定随机种子。这样每次运行生成的数据是一样的方便你复现、排查问题。实际工作中做实验也建议固定随机种子不然模型的结果每次跑都不一样你根本没法判断改动是有效还是随机波动。3.2 手写梯度下降训练循环下面是核心代码。我故意不用任何封装好的库完全用NumPy手动实现def compute_loss(w, b, X, y): y_pred w * X b loss np.mean((y - y_pred) ** 2) return loss def compute_gradient(w, b, X, y): n len(X) # 损失函数对w求导 dw (2 / n) * np.sum(X * (w * X b - y)) # 损失函数对b求导 db (2 / n) * np.sum(w * X b - y) return dw, db def train(X, y, learning_rate0.01, epochs1000): w 0.0 b 0.0 for epoch in range(epochs): dw, db compute_gradient(w, b, X, y) w w - learning_rate * dw b b - learning_rate * db if epoch % 100 0: loss compute_loss(w, b, X, y) print(fEpoch {epoch}: loss{loss:.4f}, w{w:.4f}, b{b:.4f}) return w, b运行train(X, y)你会看到loss从几千一路下降最后w落在2.0附近b落在5.0附近。这个过程本质上就是深度学习训练的缩影前向计算算出预测值、计算损失MSE、反向求梯度这里因为模型简单梯度是显式推导出来的、更新参数梯度下降。等后面用PyTorch或TensorFlow的时候这个循环里的“反向求梯度”会被自动微分的框架替掉其他环节完全没有区别。3.3 闭式解验证再来看看闭式解法。把X扩展成矩阵形式在左边拼一列1对应偏置b然后直接套公式# 构造矩阵 [X, 1] X_b np.c_[X, np.ones(len(X))] # 最小二乘公式 theta np.linalg.inv(X_b.T X_b) X_b.T y w_closed, b_closed theta print(fClosed-form solution: w{w_closed:.4f}, b{b_closed:.4f})你跑出来会发现闭式解和梯度下降迭代出来的结果几乎一致。这说明两种路线最终收敛到同一个最优解——在凸优化问题里这是数学保证的。用一个小模型把这两个解法打通你对“优化”的理解会完全不一样。3.4 学习率调多大会出事我强烈建议你做一个实验把learning_rate改成0.5再跑一遍。你会看到loss不但不下降反而逐渐变大最后直接变成inf。这就是学习率过大导致的“梯度爆炸”效应。这个现象的原因可以用一个简单类比说明你在下山每一步都迈得太大直接跨过谷底跳到了对面的半山腰接着又跨回来越蹦越高最终彻底掉出山体。而学习率太小的时候loss下降得非常缓慢就像挪蚂蚁步跑几百个epoch还在山腰上磨蹭。实践中我总结出一个经验学习率的选择通常从0.01到0.001这个区间开始试然后观察loss曲线。如果loss振荡不降就调小学习率如果loss下降得非常慢就适当调大。这种“玄学”其实背后都是几何直觉。4. 从线性回归无缝过渡到深度学习4.1 把线性模型“翻译”成神经网络语言现在到了最关键的环节把线性回归和深度学习框架彻底打通。如果你去看PyTorch或TensorFlow里定义的全连接层Linear层你会发现它做的事情就是output input W.T b这不就是线性回归的向量形式吗对完全一样。区别只在于神经网络会在这个线性变换后面紧跟一个非线性的激活函数比如ReLU、Sigmoid让模型能够拟合非线性关系。如果没有激活函数无论叠多少层线性层整个网络仍然是线性的“深度”就失去了意义。所以你可以把线性回归理解成一个只有一层、没有激活函数的神经网络。这也是为什么很多教材的目录是“线性回归 - Softmax回归 - 多层感知机 - CNN - RNN”一层一层往上加复杂度。4.2 用PyTorch重写一遍学完手写版本我们用PyTorch把同一个模型实现一遍对比一下深度框架帮你做了什么import torch import torch.nn as nn import torch.optim as optim # 将数据转为Tensor X_t torch.tensor(X, dtypetorch.float32).view(-1, 1) y_t torch.tensor(y, dtypetorch.float32).view(-1, 1) # 定义模型一个线性层输入维度1输出维度1 model nn.Linear(1, 1) # 损失函数均方误差 criterion nn.MSELoss() # 优化器随机梯度下降SGD学习率0.01 optimizer optim.SGD(model.parameters(), lr0.01) # 训练循环 for epoch in range(1000): optimizer.zero_grad() y_pred model(X_t) loss criterion(y_pred, y_t) loss.backward() optimizer.step() if epoch % 100 0: print(fEpoch {epoch}: loss{loss.item():.4f})你没看错训练的核心就这几行前向计算y_pred、算loss、反向传播backward()、更新参数step()。至于梯度是怎么算的PyTorch的自动微分机制帮你做了。但如果你没有亲手手写过梯度推导你会觉得backward()是一个黑盒出了问题完全不知道从哪儿排查。这也是我坚持先带大家手写一遍的原因。4.3 损失函数曲线判断训练是否正常的核心工具训练深度学习模型最重要的指标之一就是loss曲线。线性回归里你打印每一轮的loss会看到一条平滑下降的曲线。但实际训练中我发现一个极其普遍的新手误区只看最终loss不看loss曲线形态。正常的loss曲线应该呈现“快速下降到平缓收敛”的形状。如果loss曲线像过山车一样剧烈震荡大概率是学习率太大如果loss曲线几乎是一条水平线且数值很大可能是学习率太小、特征没有归一化或者梯度消失了。举个例子在标准化的数据上做线性回归learning_rate0.01loss从几千下降到几十大概在300个epoch后趋于平缓但如果数据不做标准化比如x的取值是0到10000同样的学习率会导致loss爆裂。原因在于x取值范围大的时候X * w那部分的梯度会非常大放大了参数更新的步长。在线性回归里我们就应该培养这个习惯每次训练都画出loss曲线观察它的收敛形态而不是只看最终数字。5. 常见问题与排查技巧实录5.1 梯度爆炸从数值溢出到NaN这是我训练中遇到的第一个经典坑。当我把学习率从0.01改成0.1loss在前几个epoch直接变成NaN。排查思路非常简单把学习率调回0.01程序恢复正常说明是学习率过大打印每一轮更新后的w和b发现w和b在几十轮内膨胀到几百上千说明是参数更新步长太大数值溢出。解决方案也不难降低学习率、对输入特征做标准化、必要时使用梯度裁剪gradient clipping。在线性回归中梯度爆炸主要是因为数据没标准化和学习率过大但深度学习里梯度爆炸是家常便饭提前养成检查梯度的习惯会受益匪浅。5.2 特征工程的重要性x的取值范围决定了收敛速度在线性回归中x的取值范围直接影响收敛速度。同样是y 2x 5这个关系如果x在0到1之间learning_rate0.1也能很快收敛如果x在0到10000之间learning_rate0.1直接爆炸。原因在于损失函数对w的偏导里包含一个乘以x的项x越大梯度越大参数更新步长就会异常大。所以数据预处理的标准化标准化为均值0、方差1不是锦上添花而是优化能够正常进行的前提。这个经验在深度学习中更加重要图像数据归一化到0-1区间、文本特征做归一化都是为了让不同尺度的特征在梯度下降过程中“公平竞争”。5.3 为什么你的损失函数不再下降我见过不少人在线性回归实验里卡在loss不再下降这个现象上以为代码写错了。第一种可能是数据本身含有噪声loss下降到某个值后继续减小反而可能把噪声也拟合进去也就是过拟合第二种可能是学习率太小模型还在一个非常平缓的区域慢慢挪理论上可以继续下降但需要极长的时间第三种可能是模型结构不够比如真实关系是二次的你偏用线性模型来拟合loss自然会卡在一个较高的水平。排查方法很简单画出拟合直线和原始数据散点图。如果直线明显偏离数据的整体走势说明模型容量不够如果直线很好地穿过数据中心但每个点都有一定偏离那基本是数据噪声的极限了。顺嘴提一个非常有价值的实操技巧记录不同学习率下的loss曲线放在同一张图上对比。这会让你非常直观地看到学习率从0.0001、0.001、0.01、0.1的变化对收敛速度和最终loss的影响。我自己当年做完这个对比实验对梯度下降的理解直接上了一个台阶。5.4 模型评估训练损失低不等于模型好还有一个很容易被忽略的问题怎么评估你的线性回归模型好坏最简单的方法是划分训练集和测试集。用训练集拟合出w和b然后在测试集上计算MSE。如果测试集loss明显高于训练集loss说明模型在训练集上“死记硬背”了噪声这就是过拟合。对线性回归来说因为模型简单过拟合风险相对较小但当你加入多项式特征比如x^2之后过拟合很快就会出现。做个简单版本from sklearn.model_selection import train_test_split X_train, X_test, y_train, y_test train_test_split(X_b, y, test_size0.2, random_state42) # 在训练集上拟合 theta np.linalg.inv(X_train.T X_train) X_train.T y_train # 在测试集上评估 y_pred X_test theta test_loss np.mean((y_test - y_pred) ** 2) print(fTest loss: {test_loss:.4f})这个流程以后做深度学习也是一模一样的train set训练、validation set调参、test set最终验证三者不可混用。错误的做法是反复用测试集调参这相当于获得答案后再做练习题测试集的评估意义就被彻底摧毁了。6. 从线性回归向外拓展的几条路线6.1 逻辑回归从回归到分类的跳板线性回归解决的是“预测连续数值”的问题但现实里还有一堆“分类”问题邮件是不是垃圾邮件是/否、图片里是猫还是狗猫/狗。线性回归的输出是一个无界的实数没法直接当作概率使用。解决办法是在线性输出上套一个Sigmoid函数把结果压缩到0到1之间这就是逻辑回归Logistic Regression。逻辑回归是深度学习中二分类问题的原型也可以看成从线性回归到神经网络的一个自然桥梁。理解了线性回归的损失函数是MSE再去看逻辑回归的交叉熵损失你会发现它们都是在衡量“预测分布和真实分布的差异”只是换了一个更适合分类场景的函数。6.2 多项式回归线性模型的非线性扩展如果数据不是直线关系而是抛物线怎么办一个巧妙的办法是“把非线性变成线性”给模型增加一个x^2的特征。这样原来的y ax b就变成了y ax bx^2 c看起来还是线性回归但x换成了更高维度的特征。这就是特征工程Feature Engineering的雏形。这个方法显示了线性回归的局限性也反映了深度学习的一个核心思想特征表示决定了模型能力上限。深度学习之所以强大是因为它通过层层特征提取自动学习到合适的表示而不需要人工设计x^2、x^3这样的特征。6.3 深度神经网络线性层 激活函数 多层堆叠最关键的路线就是你已经猜到的将单个线性层换成多个线性层每层后面加激活函数。结构变成输入 - 线性层1 - ReLU - 线性层2 - ReLU - ... - 线性层N - 输出这个过程就是多层感知机MLP。虽然看起来只是做了线性变换加非线性变换的反复操作但它有理论上的保证只要层数足够多、宽度足够大神经网络可以以任意精度逼近任意连续函数。这就是通用近似定理。线性回归是这条大路的第0公里而你已经跑完了这一程。7. 写在最后的个人体会我带过很多完全零基础的朋友学习机器学习和深度学习发现每个人几乎都会经历同一个拐点手推完线性回归的梯度再用框架跑通第一个训练循环之后原本模糊的概念突然就清晰了。因为“训练”这件事本质上就是反复执行“计算预测 - 计算损失 - 求梯度 - 更新参数”这个循环线性回归里它是显式的、透明的到了深度学习中它被框架封装了但内核完全没有变。所以我一直坚信深度学习入门不要急着上GPU、云平台或者复杂框架先把线性回归在NumPy里手动实现一遍再看框架怎么封装效率远高于一上来就用高级API。基础不牢的时候你无法判断训练结果到底是模型问题、数据问题还是代码问题只能靠瞎猜。另外分享一个小技巧我每次调试代码习惯在训练循环里每隔一定epoch打印训练集和验证集的loss同时画出一条拟合曲线的动画。这样做能捕捉到训练过程中的每一个异常比如某个epoch之后loss突然反弹能快速定位到是数据扰动还是学习率半路需要调整。这个习惯看起来不起眼但确实帮我排查了无数次问题。线性回归只是一个开始但这“开始”的价值值得你花足够的时间去打磨。后面的路还很长但每一步都建立在同一块基石之上。
阅读完成 · 觉得有帮助？
