当前位置：首页 > news >正文

机器学习 —— 深入剖析线性回归模型

news 来源：原创 2025/7/6 4:24:01

一、线性回归模型简介

线性回归是机器学习中最为基础的模型之一，主要用于解决回归问题，即预测一个连续的数值。其核心思想是构建线性方程，描述自变量（特征）和因变量（目标值）之间的关系。简单来说，若有一个自变量 $x$ 和一个因变量 $y$ ，简单线性回归模型可表示为： $\theta_0 + \theta_1x$ ，其中 $\theta_0$ 是截距， $\theta_1$ 是斜率，也被称为回归系数。通过这条直线，我们尝试让模型预测值尽可能接近真实值。

（一）多元线性回归

在实际应用中，数据往往具有多个特征，这就需要多元线性回归模型。假设我们有 $n$ 个自变量 $x_1, x_2, \cdots, x_n$ ，多元线性回归模型的表达式为： $\theta_0 + \theta_1x_1 + \theta_2x_2 + \cdots + \theta_nx_n$ 。从几何角度理解，简单线性回归是在二维平面上找一条最佳拟合直线；而多元线性回归则是在更高维度空间中寻找一个超平面，使得所有数据点到这个超平面的距离之和最小。

例如，在预测房价时，房屋价格可能受到面积、房龄、房间数量、周边配套设施等多个因素影响，多元线性回归模型能够综合考虑这些因素，从而做出更准确的预测。

（二）岭回归

岭回归是一种改进的线性回归算法，也被称为 Tikhonov 正则化。在普通线性回归中，当特征数量较多且存在多重共线性（即某些特征之间存在较强的线性关系）时，计算正规方程中的 $X^TX)^{-1}$ 可能会出现问题，导致模型不稳定，对训练数据的微小变化非常敏感，泛化能力差。

岭回归通过在损失函数中添加一个 L2 正则化项来解决这个问题。其损失函数变为： $J(\theta) = \sum_{i = 1}^{m}(y^{(i)} - \hat{y}^{(i)})^2 + \lambda\sum_{j = 1}^{n}\theta_j^2$ ，其中 $\lambda$ 是正则化参数，用来控制正则化的强度。当 $\lambda$ 越大时，对回归系数的约束越强，使得回归系数更倾向于收缩到 0，从而防止过拟合；当 $\lambda$ 为 0 时，岭回归就退化为普通的线性回归。

岭回归的优势在于，它不仅能在一定程度上解决多重共线性问题，还能提高模型的泛化能力，使得模型在面对新数据时表现更加稳定。

（三）Lasso 回归

Lasso 回归，即 Least Absolute Shrinkage and Selection Operator，同样是一种用于线性回归的正则化方法。与岭回归不同，Lasso 回归在损失函数中添加的是 L1 正则化项，其损失函数为： $J(\theta) = \sum_{i = 1}^{m}(y^{(i)} - \hat{y}^{(i)})^2 + \lambda\sum_{j = 1}^{n}|\theta_j|$ 。

L1 正则化的特点是它能够产生稀疏解，即可以自动筛选出对目标值影响较大的特征，将一些不重要的特征对应的系数直接压缩为 0，从而达到特征选择的目的。例如在基因数据分析中，数据维度极高，特征众多，Lasso 回归可以帮助我们从大量的基因特征中筛选出真正与疾病相关的基因，简化模型的同时提高解释性。

（四）弹性网络回归

弹性网络回归结合了岭回归和 Lasso 回归的优点，在损失函数中同时使用 L1 和 L2 正则化项，其损失函数表达式为： $J(\theta) = \sum_{i = 1}^{m}(y^{(i)} - \hat{y}^{(i)})^2 + \lambda_1\sum_{j = 1}^{n}|\theta_j| + \lambda_2\sum_{j = 1}^{n}\theta_j^2$ ，其中 $\lambda_1$ 和 $\lambda_2$ 分别是 L1 和 L2 正则化项的系数。

这种方法既可以像 Lasso 回归一样进行特征选择，又能像岭回归一样处理多重共线性问题。在一些复杂的数据场景中，比如图像识别中，数据既存在大量冗余特征，又有特征间的相关性，弹性网络回归能够发挥其综合优势，平衡模型的复杂度和性能。

二、线性回归模型的原理

线性回归模型的目标是找到一组最优的回归系数 $\theta = [\theta_0, \theta_1, \cdots, \theta_n]$ ，使得模型预测值与真实值之间的误差最小。通常，我们使用最小二乘法来衡量这种误差。最小二乘法的目标函数（也称为损失函数）为： $J(\theta) = \sum_{i = 1}^{m}(y^{(i)} - \hat{y}^{(i)})^2$ ，其中 $m$ 是样本数量， $y^{(i)}$ 是第 $i$ 个样本的真实值， $\hat{y}^{(i)}$ 是第 $i$ 个样本的预测值， $\hat{y}^{(i)} = \theta_0 + \theta_1x_1^{(i)} + \theta_2x_2^{(i)} + \cdots + \theta_nx_n^{(i)}$ 。

为了找到使损失函数最小的 $\theta$ ，我们可以对 $J(\theta)$ 求关于 $\theta$ 的导数，并令导数为零，从而得到正规方程： $\theta = (X^TX)^{-1}X^Ty$ ，其中 $X$ 是特征矩阵，每一行代表一个样本，每一列代表一个特征， $y$ 是目标值向量。但正如前面提到的，当 $X^TX$ 接近奇异矩阵（即不可逆）时，求解正规方程会出现问题，这也是岭回归、Lasso 回归和弹性网络回归等方法出现的原因之一。

三、线性回归模型的优化方法

除了使用正规方程求解回归系数外，我们还可以使用梯度下降法来优化损失函数。梯度下降法是一种迭代的优化算法，它通过不断地沿着损失函数的负梯度方向更新回归系数，来逐步减小损失函数的值。

具体来说，对于损失函数 $J(\theta)$ ，其梯度为： $\nabla J(\theta) = \frac{2}{m}X^T(X\theta - y)$ 。在每次迭代中，我们按照以下公式更新回归系数： $\theta = \theta - \alpha\nabla J(\theta)$ ，其中 $\alpha$ 是学习率，它控制着每次更新的步长。学习率的选择非常关键，如果学习率过大，可能会导致模型无法收敛，甚至发散；如果学习率过小，模型收敛速度会非常慢，需要更多的迭代次数。

四、Python 代码实现

下面我们使用 Python 来实现一个简单的线性回归模型，包括普通线性回归、多元线性回归、岭回归、Lasso 回归和弹性网络回归，并对比它们的效果。首先，我们需要导入必要的库，如numpy、matplotlib和sklearn中的相关模块。

import numpy as np
import matplotlib.pyplot as plt
from sklearn.linear_model import LinearRegression, Ridge, Lasso, ElasticNet
from sklearn.preprocessing import PolynomialFeatures
from sklearn.model_selection import GridSearchCV# 生成一些随机数据
np.random.seed(0)
X = 2 * np.random.rand(100, 1)
y = 4 + 3 * X + np.random.randn(100, 1)# 普通线性回归
lin_reg = LinearRegression()
lin_reg.fit(X, y)
y_lin_pred = lin_reg.predict(X)# 多元线性回归（添加一个多项式特征）
poly_features = PolynomialFeatures(degree=3, include_bias=False)  # 修改多项式次数为3
X_poly = poly_features.fit_transform(X)
lin_reg_2 = LinearRegression()
lin_reg_2.fit(X_poly, y)
y_poly_pred = lin_reg_2.predict(X_poly)# 岭回归
ridge_reg = Ridge(alpha=0.1)
ridge_reg.fit(X, y)
y_ridge_pred = ridge_reg.predict(X)# Lasso回归
lasso_reg = Lasso(alpha=0.1)
lasso_reg.fit(X, y)
y_lasso_pred = lasso_reg.predict(X)# 弹性网络回归
elastic_net_reg = ElasticNet(alpha=0.1, l1_ratio=0.5)
elastic_net_reg.fit(X, y)
y_elastic_pred = elastic_net_reg.predict(X)# 使用网格搜索优化岭回归和Lasso回归的超参数
ridge_grid = GridSearchCV(Ridge(), param_grid={'alpha': [0.01, 0.1, 1, 10, 100]})
ridge_grid.fit(X, y)
best_ridge = ridge_grid.best_estimator_
y_ridge_best_pred = best_ridge.predict(X)lasso_grid = GridSearchCV(Lasso(), param_grid={'alpha': [0.01, 0.1, 1, 10, 100]})
lasso_grid.fit(X, y)
best_lasso = lasso_grid.best_estimator_
y_lasso_best_pred = best_lasso.predict(X)# 绘制数据和拟合直线
plt.figure(figsize=(15, 8))plt.subplot(2, 3, 1)
plt.plot(X, y, "b.")
plt.plot(X, y_lin_pred, "r-", linewidth=2, label='Linear Regression')
plt.title('Linear Regression')
plt.xlabel("$x_1$", fontsize=18)
plt.ylabel("$y$", rotation=0, fontsize=18)
plt.legend()plt.subplot(2, 3, 2)
plt.plot(X, y, "b.")
X_sorted = np.sort(X, axis=0)
X_poly_sorted = poly_features.fit_transform(X_sorted)
plt.plot(X_sorted, lin_reg_2.predict(X_poly_sorted), "g-", linewidth=2, label='Polynomial Linear Regression (Degree=3)')
plt.title('Polynomial Linear Regression')
plt.xlabel("$x_1$", fontsize=18)
plt.ylabel("$y$", rotation=0, fontsize=18)
plt.legend()plt.subplot(2, 3, 3)
plt.plot(X, y, "b.")
plt.plot(X, y_ridge_pred, "m-", linewidth=2, label='Ridge Regression (alpha=0.1)')
plt.title('Ridge Regression')
plt.xlabel("$x_1$", fontsize=18)
plt.ylabel("$y$", rotation=0, fontsize=18)
plt.legend()plt.subplot(2, 3, 4)
plt.plot(X, y, "b.")
plt.plot(X, y_lasso_pred, "c-", linewidth=2, label='Lasso Regression (alpha=0.1)')
plt.title('Lasso Regression')
plt.xlabel("$x_1$", fontsize=18)
plt.ylabel("$y$", rotation=0, fontsize=18)
plt.legend()plt.subplot(2, 3, 5)
plt.plot(X, y, "b.")
plt.plot(X, y_elastic_pred, "y", linewidth=2, label='Elastic Net Regression (alpha=0.1, l1_ratio=0.5)')
plt.title('Elastic Net Regression')
plt.xlabel("$x_1$", fontsize=18)
plt.ylabel("$y$", rotation=0, fontsize=18)
plt.legend()plt.subplot(2, 3, 6)
plt.plot(X, y, "b.")
plt.plot(X, y_ridge_best_pred, "k", linewidth=2, label='Optimized Ridge Regression')
plt.plot(X, y_lasso_best_pred, "saddlebrown", linewidth=2, label='Optimized Lasso Regression') 
plt.title('Optimized Ridge and Lasso Regression')
plt.xlabel("$x_1$", fontsize=18)
plt.ylabel("$y$", rotation=0, fontsize=18)
plt.legend()plt.tight_layout()
plt.show()

在上述代码中，我们首先生成了一些随机数据。然后分别使用LinearRegression类实现普通线性回归和多元线性回归（通过添加多项式特征实现），使用Ridge类实现岭回归，使用Lasso类实现 Lasso 回归，使用ElasticNet类实现弹性网络回归。最后绘制出数据点和各个模型的拟合直线，以便直观对比它们的效果。

五、总结与模型选用建议

不同的线性回归模型各有特点，在实际应用中需要根据具体情况选择合适的模型。

✨简单线性回归模型形式最为简单，仅包含一个自变量和一个因变量，适用于特征与目标值之间呈现明显线性关系，且数据特征单一的场景，比如根据时间预测某一产品的销量变化趋势。

🎈多元线性回归在简单线性回归基础上拓展到多个自变量，能处理更复杂的数据关系，像预测房价时综合考虑多个影响因素。但当数据存在多重共线性时，普通的多元线性回归可能导致模型不稳定。

🎨岭回归通过 L2 正则化项，在一定程度上缓解多重共线性问题，同时提升模型泛化能力。若数据特征众多且存在共线性，又希望保留所有特征，岭回归是不错的选择，如金融风险评估中，众多经济指标相互关联，岭回归可有效处理。

🍫Lasso 回归利用 L1 正则化产生稀疏解，自动筛选重要特征，实现特征选择，在高维数据场景优势明显，如基因数据分析，能从海量基因特征中找出关键特征。

🧆弹性网络回归结合了 L1 和 L2 正则化，兼具特征选择和处理共线性的能力，当数据既存在大量冗余特征，又有特征间相关性时，弹性网络回归能平衡模型复杂度与性能，例如图像识别领域。

在选择线性回归模型时，首先要分析数据特征，判断是否存在多重共线性、数据维度高低等。若数据简单且特征少，普通线性回归即可；若特征多且存在共线性，可考虑岭回归；若需特征选择，Lasso 回归或弹性网络回归更合适。还可以通过交叉验证等方法，比较不同模型在训练集和验证集上的性能指标，如均方误差（MSE）、决定系数（R²）等，最终选择性能最优的模型。不断实践和尝试不同模型，才能在实际应用中发挥线性回归模型的最大价值。