最小二乘法是用来做函数拟合或者求函数极值的方法。
最小二乘法的原理与要解决的问题
最小二乘法是由勒让德在 19 世纪发现的,原理的一般形式很简单,当然发现的过程是非常艰难的。形式如下式:
目标函数=∑(观测值−理论值)2
观测值就是我们的多组样本,理论值就是我们的假设拟合函数, 目标函数也就是在机器学习中常说的损失函数,我们的目标是得到使目标函数最小化时候的拟合函数的模型。举一个最简单的线性回归的简单例子,比如我们有 m 个只有一个特征的样本:
(x(1),y(1)),(x(2),y(2)),...(x(m),y(m))
样本采用下面的拟合函数:
hθ(x)=θ0+θ1x
这样我们的样本有一个特征 x,对应的拟合函数有两个参数θ0和θ1需要求出。
我们的目标函数为:
J(θ0,θ1)=i=1∑m(y(i)−hθ(x(i))2=i=1∑m(y(i)−θ0−θ1x(i))2
为了使J(θ0,θ1)的值最小,使用最小二乘法
最小二乘法的代数法解法
假设J(a,b)=∑i=1m(y(i)−ax(i)−b)2, 要获得该函数的极值,即对该函数的各个未知分量求导,使其导数为 0。即:
∂a∂J(a,b)=0,
∂b∂J(a,b)=0
对 b 求导:
∂b∂J(a,b)=i=1∑m2(y(i)−ax(i)−b)(−1)=0i=1∑m(y(i)−ax(i)−b)=0i=1∑my(i)−ai=1∑mx(i)−i=1∑mb=0i=1∑my(i)−ai=1∑mx(i)−mb=0i=1∑my(i)−ai=1∑mx(i)=mb⇒b=yˉ−axˉ(xˉ和yˉ分别为x和y的均值)
对 a 求导:
∂a∂J(a,b)=i=1∑m2(y(i)−ax(i)−b)(−x(i))=0i=1∑m(y(i)−ax(i)−b)x(i)=0i=1∑m(y(i)−ax(i)−yˉ+axˉ)x(i)=0i=1∑m(x(i)y(i)−a(x(i))2−x(i)yˉ+axˉx(i))=0i=1∑m(x(i)y(i)−x(i)yˉ−a(x(i))2+axˉx(i))=0i=1∑m(x(i)y(i)−x(i)yˉ)−i=1∑m(a(x(i))2−axˉx(i))=0i=1∑m(x(i)y(i)−x(i)yˉ)−ai=1∑m((x(i))2−xˉx(i))=0a=∑i=1m((x(i))2−xˉx(i))∑i=1m(x(i)y(i)−x(i)yˉ)
对于公式∑i=1mx(i)yˉ 可以进一步推导
i=1∑mx(i)yˉ=yˉi=1∑mx(i)=myˉxˉ=i=1∑mxˉyˉ=xˉi=1∑my(i)=i=1∑mxˉy(i)
带入 a 中可得到
a=∑i=1m((x(i))2−xˉx(i)−xˉx(i)+xˉ2)∑i=1m(x(i)y(i)−x(i)yˉ)−xˉy(i)+xˉyˉ)=∑i=1m(x(i)−xˉ)2∑i=1m(x(i)−xˉ)(y(i)−yˉ)
总结: 通过上面的推导可以得到 a 和 b 的值
a=∑i=1m(x(i)−xˉ)2∑i=1m(x(i)−xˉ)(y(i)−yˉ),b=yˉ−axˉ
这个方法很容易推广到多个样本特征的线性拟合。拟合函数表示为hθ(x1,x2,...xn)=θ0+θ1x1+...+θnxn, 其中θi (i = 0,1,2... n)为模型参数,xi (i = 0,1,2... n)为每个样本的 n 个特征值。这个表示可以简化,我们增加一个特征x0=1 ,这样拟合函数表示为:
J(θ0,θ1...,θn)=j=1∑m(hθ(x0(j)),x1(j),...xn(j)))−y(j)))2=j=1∑m(i=0∑nθixi(j)−y(j))2
利用损失函数分别对θi(i=0,1,...n)求导,并令导数为 0 可得:
j=0∑m(i=0∑n(θixi(j)−y(j))xi(j)=0(i=0,1,...n)
这样我们得到一个N+1
元一次方程组,这个方程组有N+1
个方程,求解这个方程,就可以得到所有的N+1
个未知的θ
最小二乘法的矩阵法解法
矩阵法比代数法要简洁,且矩阵运算可以取代循环,所以现在很多书和机器学习库都是用的矩阵法来做最小二乘法。这里用上面的多元线性回归例子来描述矩阵法解法。
假设函数hθ(x1,x2,...xn)=θ0+θ1x1+...+θn−1xn−1的矩阵表达方式为:
hθ(x)=Xθ
其中, 假设函数hθ(X)为 mx1 的向量,θ为 nx1 的向量,里面有 n 个代数法的模型参数。X为 mxn 维的矩阵。m 代表样本的个数,n 代表样本的特征数。
损失函数定义为:
J(θ)=21(Xθ−Y)T(Xθ−Y)
其中Y是样本的输出向量,维度为 mx1. 21在这主要是为了求导后系数为 1,方便计算。
根据最小二乘法的原理,我们要对这个损失函数对θ向量求导取 0。结果如下式:
∂θ∂J(θ)=XT(Xθ−Y)=0
这里面用到了矩阵求导链式法则,和两个矩阵求导的公式。
公式 1:∂x∂(xTx)=2xx为向量
公式 2: ∇Xf(AX+B)=AT∇Yf,Y=AX+B,f(Y)为标量
对上述求导等式整理后可得:
XTXθ=XTY
两边同时左乘(XTX)−1可得:
θ=(XTX)−1XTY
这样就求出了θ向量表达式的公式,免去了代数法一个个去求导的麻烦。只要给了数据,我们就可以用 θ=(XTX)−1XTY 算出θ
最小二乘法的局限性和适用场景
最小二乘法适用简洁高效,比梯度下降这样的迭代法似乎方便很多。但是这里我们就聊聊最小二乘法的局限性。
首先,最小二乘法需要计算XTX的逆矩阵,有可能它的逆矩阵不存在,这样就没有办法直接用最小二乘法了,此时梯度下降法仍然可以使用。当然,我们可以通过对样本数据进行整理,去掉冗余特征。让XTX的行列式不为 0,然后继续使用最小二乘法。
第二,当样本特征 n 非常的大的时候,计算XTX的逆矩阵是一个非常耗时的工作(nxn 的矩阵求逆),甚至不可行。此时以梯度下降为代表的迭代法仍然可以使用。那这个 n 到底多大就不适合最小二乘法呢?如果你没有很多的分布式大数据计算资源,建议超过 10000 个特征就用迭代法吧。或者通过主成分分析降低特征的维度后再用最小二乘法。
第三,如果拟合函数不是线性的,这时无法使用最小二乘法,需要通过一些技巧转化为线性才能使用,此时梯度下降仍然可以用。
第四,讲一些特殊情况。当样本量 m 很少,小于特征数 n 的时候,这时拟合方程是欠定的,常用的优化方法都无法去拟合数据。当样本量 m 等于特征数 n 的时候,用方程组求解就可以了。当 m 大于 n 时,拟合方程是超定的,也就是我们常用与最小二乘法的场景了。