线性回归 基本介绍

2018-02-01 ai

回归分析 (Regression Analysis) 是确定两种或两种以上变量间相互依赖的定量关系的一种统计分析方法,线性回归 (Linear Regression) 也是最基本的。

这里介绍其概念、公式推导以及基本的实现。

简介

线性回归是机器学习中一个最基本的类型,试图从数据集中找到一个线性模型,这个模型可以准确反映出输入 XiX_i 和输出 YiY_i 的对应关系。

假设每个数据样本有 dd 维,对应的线性函数如下。

f(x)=w1x1+w2x2++wdxd+bf(x) = w_1 x_1 + w_2 x_2 + \cdots + w_d x_d + b

首先从一个简单的示例介绍。

线性回归

简单来说,现在有一批的数据,示例如下。

linear regression example data

可以通过如下代码生成如上的测试数据。

import numpy as np
import matplotlib.pyplot as pylab

x = np.arange(0.0, 10.1, 0.1)
y = (2 * x - 5) + np.random.normal(scale=2.0**0.5, size=x.size)

pylab.scatter(x, y)
pylab.xlabel('x', fontsize=15)
pylab.ylabel('y', fontsize=15)
pylab.title('What our data looks like', fontsize=12)
pylab.show()

我们希望通过这些数据来拟合出一条直线。

y=w1x+w0y=w_1 x + w_0

也就是要确定 w1w_1w0w_0 的值,一般来说,最常用的计算方法是获取这些数据最小的 Root Mean Squared Error, RMSE ,也即。

w1,w0=arg minw1,w0i=1N(yi(w1xi+w0))2w_1, w_0=arg \ min_{w_1, w_0} \sum_{i=1}^{N}(y_i - (w_1 x_i + w_0))^2

直接使用现成的库,后面在介绍直接使用 numpy 的方式。

普通最小二乘法

如上所述,也就是找到一条参数对应的曲线,使得所有数据具有最小均方根误差 (RMSE),可以直接使用 Scikit-Learn 中的 LinearRegression 模块,完整的代码如下。

import numpy as np
import matplotlib.pyplot as pylab
from sklearn.linear_model import LinearRegression

x = np.arange(0.0, 10.1, 0.1)
y = (2 * x - 5) + np.random.normal(scale=2.0**0.5, size=x.size)

lr = LinearRegression()
lr.fit(x.reshape(-1, 1), y)

pylab.scatter(x,y)
pylab.plot(x, lr.coef_[0]*x + lr.intercept_, color='r')
pylab.xlabel('x',fontsize=18)
pylab.ylabel('y',fontsize=18)
pylab.show()

如上处理时,sklearn 要求输入的特征必须是二维数组的类型,但目前只有 1 个特征,需要用 reshape() 转行成二维数组的类型。

最终得到的直线如下所示。

linear regression example

实际上这里缺少一个度量,也就是 置信界限 (confidence bounds),当某个点的数据比较多时,认为这个范围内可信度大,而当数据较少时则不确定性增加。

这一问题涉及到了后面介绍的贝叶斯线性回归。

公式推导

一元推导

假设在一个样本集合 DD 中,有 nn 个样本 (x1,y1),(x2,y2),,(xn,yn){(x_1, y_1), (x_2, y_2), \cdots , (x_n, y_n)}

代价函数

一般通过均方误差 (欧式距离) 对结果进行评估,也就是代价函数 (Cost Function) ,表示为如下。

$$(w^, b^)=arg \ min_{(w,b)} \sum_{i=1}^{m} (f(x_i) - y_i)^2$$

arg minarg \ min 是指后面表达式值最小时,对应 (w,b)(w, b) 的取值,实际上就是最小二乘法。接着就是要看如何求解参数,使得代价函数最小。

E(w,b)=i=1m(yiwxib)2E(w, b) = \sum_{i=1}^m(y_i - wx_i - b)^2

其中,函数 EE 是关于参数 (w,b)(w, b) 的凸函数,那么这也就变成了一个凸优化问题,其中 wwbb 是未知参数,使其对应的偏导为 00 后,就可以得到最优解。

公式推导

首先分别对 wwbb 求偏导,注意,这里使用的是一元。

$$ \begin{align} \frac{\partial{E(w,b)}}{\partial{w}}&=\frac{\partial{\sum_{i=1}^m(y_i-wx_i-b)^2}}{\partial{w}} \ &=\frac{\partial{\sum_{i=1}^m((y_i-b)^2+w^2x_i^2-2w(y_i-b)x_i)}}{\partial{w}} \ &= \sum_{i=1}^m(2wx_i^2-2(y_i-b)x_i) \ &= 2(w\sum_{i=1}^m{x_i^2}-\sum_{i=1}^m(y_i-b)x_i)\

\frac{\partial{E(w,b)}}{\partial{b}}&=\frac{\partial{\sum_{i=1}^m(y_i-wx_i-b)^2}}{\partial{b}} \ &=\frac{\partial{\sum_{i=1}^m((y_i-wx_i)^2+b^2-2b(y_i-wx_i))}}{\partial{b}} \ &= \sum_{i=1}^m(2b-2(y_i-wx_i)) \ &= 2(mb-\sum_{i=1}^m{(y_i-wx_i)}) \end{align} $$

然后令偏导 (也就是上述的结果) 为 00 最终求解出结果。

w=mxiyixiyimxi2(xi)2 b=1mi=1m(yiwxi)=xi2yixixiyimxi2(xi)2 \begin{align} w&=\frac{m\sum {x_i y_i}-\sum {x_i}\sum {y_i}}{m\sum {x_i^2}-(\sum x_i)^2} \ b&=\frac{1}{m}\sum_{i=1}^{m}(y_i-wx_i)=\frac{\sum {x_i^2} \sum{ y_i}-\sum {x_i}\sum {x_i y_i}}{m\sum {x_i^2}-(\sum x_i)^2} \end{align}

高维推导

假设在一个样本集合 DD 中,有 nn 个样本 (X1,Y1),(X2,Y2),,(Xn,Yn){(X_1, Y_1), (X_2, Y_2), \cdots , (X_n, Y_n)} ,其中每个样本由 dd 个属性,可以表示为输入特征向量为 X=(x1,x2,,xd)TX = (x_1, x_2, \cdots , x_d)^T ,对应模型的预测输出 f(x)f(x) 可以表示为输入 XX 的线性函数,也就是。

f(x)=w1x1+w2x2++wdxd+bf(x) = w_1 x_1 + w_2 x_2 + \cdots + w_d x_d + b

对应的矩阵形式为。

Y=f(x)=wTx+bY=f(x)=w^T x + b

其中 xxww 均为 d×1d \times 1 的列向量,其中 ww 代表权重 (Weight) 的意思。接着将 wwbb 合并为 θ\theta

梯度下降

上面讨论的都是解析解,也就是直接通过公式进行计算,接着讨论的是通过梯度下降进行求解。