线性模型是机器学习中最基本的模型,既可以用来做回归任务,也可以用来做分类任务。这篇文章我们主要介绍用来做回归任务的线性回归。

      线性模型主要有三个优点:

      (1)形式简单,易于建模;

      (2)作为机器学习最基础的模型,许多功能强大的非线性模型都是在线性模型的基础上加入层级结构或高维映射演进而来;

      (3)具有良好的模型可解释性,权重w直观体现了各特征属性在预测中的重要性。

      线性回归,顾名思义,就是通过学习得到一个特征的线性组合模型来预测连续值。

      按特征(属性)数目,线性回归可以分为一元线性回归和多元线性回归:

 

一元线性回归模型: 

机器学习笔记(二)---- 线性回归_人工智能学习

      a和b学得之后,模型就确定了,这里,自变量只有一个,所以该模型是平面上的一条直线。

 

多元线性回归模型:

机器学习笔记(二)---- 线性回归_人工智能学习_02

      用向量形式改写为

机器学习笔记(二)---- 线性回归_人工智能学习_03

      w是各自变量(特征属性)的权重,wi绝对值越大,表明特征xi对于预测值影响越大,该模型自变量有多个,所以在空间上是一个平面。

 

学习策略及模型评估:

      如何求解自变量的权重w和b呢?通常采用极小化模型预测输出和真实值之间的距离,在回归任务中,采用基于均方误差最小化的“最小二乘法”来求解w和b。

      线性模型的评估主要使用均方误差、均方根误差、R-Square(被模型解释的信息比例)

      -- 均方误差(MSE)(使得均方误差最小,同时也可作为线性模型的损失函数)如下,求解w和b使得E最小化的的过程,就称为线性回归模型的“最小二乘参数估计”

机器学习笔记(二)---- 线性回归_人工智能学习_04

      均方误差的几何意义:试图找到一条直线或一个平面,使得所有样本到直线上的欧式距离之和最小。如下图所示,分别表示一元线性模型和多元线性模型的均方差几何表示。

机器学习笔记(二)---- 线性回归_机器学习_05

      -- 均方根误差(RMSE):实际上就是均方根(MSE)的平方根

机器学习笔记(二)---- 线性回归_机器学习_06

      -- R Squared:变量对于预测值的解释程度,可以简单理解为模型对于预测值解释能力的强弱,取值在[0,1]之间,类似于分类算法中的正确率。一般情况下,越大越好。

机器学习笔记(二)---- 线性回归_人工智能学习_07

      SSres为拟合数据和原始数据的误差平方和,SStot为原始数据和均值之差的平方和

 

多项式回归模型:

      现实问题中,直线或者平面并不能很好地拟合大部分数据,说明特征属性和预测值之间并没有很强的线性关系,我们需要采用多项式回归模型进行曲线或超平面拟合,这是一种特殊的线性模型,模型中自变量的指数大于1,那就是多项式回归模型,一元多项式模型公式如下:

机器学习笔记(二)---- 线性回归_机器学习_08

线性回归实战练习:

      下面通过一个小的实战来对线性回归有一个更深入的理解,实战中上面提到的知识都会用到。我采用sklearn中自带的数据集boston,该数据集包含对波士顿房价影响的多个特征属性值及对应的房价值,可以用来做回归任务的训练。

      -- 数据分析

      先看下数据的基本情况和有哪些特性:一共有506条数据,13个特征属性

boston_data = datasets.load_boston()
print (boston_data['data'].shape,"\n", boston_data['DESCR'])
------------------------------------------------------------
(506, 13) 
 Boston House Prices dataset
        - CRIM     per capita crime rate by town
        - ZN       proportion of residential land zoned for lots over 25,000 sq.ft.
        - INDUS    proportion of non-retail business acres per town
        - CHAS     Charles River dummy variable (= 1 if tract bounds river; 0 otherwise)
        - NOX      nitric oxides concentration (parts per 10 million)
        - RM       average number of rooms per dwelling
        - AGE      proportion of owner-occupied units built prior to 1940
        - DIS      weighted distances to five Boston employment centres
        - RAD      index of accessibility to radial highways
        - TAX      full-value property-tax rate per $10,000
        - PTRATIO  pupil-teacher ratio by town
        - B        1000(Bk - 0.63)^2 where Bk is the proportion of blacks by town
        - LSTAT    % lower status of the population

      通过变量关系图看下各特征与预测值(房价)的相关性如何:

# 通过双变量关系图查看变量与预测值相关性
data = pd.DataFrame(datasets.load_boston().data)
data.columns = boston_data['feature_names']
data['price'] = boston_data['target']
sns.pairplot(data, x_vars=data.columns, y_vars='price', kind='reg')
plt.show()

机器学习笔记(二)---- 线性回归_人工智能学习_09

      从双变量关系图中可以看出,RM、LSTAT这两个特征和房价的线性关系比较明显,其他特征的线性关系较弱。

      -- baseline模型
      先采用最简单的线性模型对数据进行预测,看效果怎么样。

-Python 代码

def Evaluate_model(true_data_X, true_data_y, pred_data, model):
    # 计算MSE(均方差)
    print ("MSE:",metrics.mean_squared_error(true_data_y, pred_data))
    # 计算RMSE(均方根差)
    print ("RMSE:",np.sqrt(metrics.mean_squared_error(true_data_y, pred_data)))
    # 模型R方
    print ("R^2:", model.score(true_data_X, test_y))

data_np = boston_data['data']
target_np = boston_data['target']

# 划分训练集和测试集
train_X,test_X, train_y, test_y = train_test_split(data_np,target_np,test_size = 0.4,random_state = 0)
model_linear = LinearRegression()
model_linear.fit(train_X, train_y)
# 预测
y_pred_linear = model_linear.predict(test_X)
# 模型评估
Evaluate_model(test_X, test_y, y_pred_linear, model_linear)
--------------------------------------------------------------
MSE: 25.7971648592073
RMSE: 5.079090948113382
R^2: 0.6881784869675758

      -- 多项式回归模型

      从baseline模型的预测结果来看,效果并不是特别好,R-Squared只有0.68,对测试数据的拟合一般,从前面数据分析的变量关系图也可以看出,大部分特征和预测值之间并不是明显的线性关系,我们可以提高自变量的维度,也就是提高x的阶数,将模型变换为多项式回归模型。

-Python 代码

# 多项式回归,最高阶设为2,阶数太高容易导致过拟合
poly = PolynomialFeatures(degree=2, interaction_only=True,include_bias=True)
train_X_poly = poly.fit_transform(train_X)
test_X_poly = poly.transform(test_X)
model_Polynomial = LinearRegression()
model_Polynomial.fit(train_X_poly, train_y)
y_pred_Polynomial = model_Polynomial.predict(test_X_poly)

# 模型评估
Evaluate_model(test_X_poly, test_y, y_pred_Polynomial, model_Polynomial)
---------------------------------------------------------------------
MSE: 17.989393685174065
RMSE: 4.241390536743117
R^2: 0.782554401304885

      看来预测房价,多项式回归模型比单纯的线性回归模型更适合,MSE和RMSE都减少,并且R-Squared提升到0.78,模型性能有较大程度的提升。

      -- 正则化

      选定多项式回归模型后,还有没有方法继续优化提升模型性能呢?答案是正则化,正则化在数学推导上比较复杂,我们可以把它简单理解为一种特征选择方法,在模型中添加“惩罚系数”使得一些不那么重要的特征降低在模型中的影响度,常用的正则化方法有Lasso和Ridge。

      Lasso方法倾向于压缩一部分特征的相关系数为0,保留一小部分特征,通俗的理解就是如果两个特征变量如果强相关的话,Lasso方法会将其中一个不那么重要特征的相关系数变为0。而Ridge方法的做法则是尽量保留特征信息,只是把相关变量的系数同时缩小。

      我尝试把两种正则化方法应用到模型中,看看效果如何

-Python 代码

# 引入Lasso正则化
model_lasso = LassoLarsCV()
model_lasso.fit(train_X_poly, train_y)
y_pred_lasso = model_lasso.predict(test_X_poly)
# 模型评估
Evaluate_model(test_X_poly, test_y, y_pred_lasso, model_lasso)
----------------------------------------------------------
MSE: 14.18038792061974
RMSE: 3.765685584408202
R^2: 0.8285955049352569
---------------------------------------------------------
# 引入Ridge正则化
model_ridge = RidgeCV(alphas=[0.1, 1.0, 10.0])
model_ridge.fit(train_X_poly, train_y)
y_pred_ridge = model_ridge.predict(test_X_poly)
# 模型评估
Evaluate_model(test_X_poly, test_y, y_pred_ridge, model_ridge)
-------------------------------------------------------------
MSE: 15.737364846211017
RMSE: 3.967034767456799
R^2: 0.8097756499882447

      从预测的结果来看,正则化对模型预测率有一定帮助,如果特征属性非常多,而训练样本又比较少的情况下,正则化对应模型性能提升会有比较好的效果,最后,通过图形比较看看baseline模型和优化后引入Lasso正则化的模型,图中表示的是预测值y和真实值x的变化关系,离直线y=x越近的点表示预测准确度越高。从图上可以发现,引入Lasso方法的图形中,点明显更靠近直线,说明预测偏差更小,MSE、RMSE和R^Squared等模型评估参数也说明了这一点。

作者:华为云专家周捷