1. 梯度下降法

YL發表於2022-06-18

1. 簡介

梯度下降法是一種函式極值的優化演算法。在機器學習中,主要用於尋找最小化損失函式的的最優解。是演算法更新模型引數的常用的方法之一。

2. 相關概念

1. 導數

  • 定義

設一元函式\(f(x)\)\(x_0\)的臨域內有定義,若極限

\[f^{`}(x_0)=\lim_{\Delta x\to0}\frac{f(x+\Delta x)-f(x)}{\Delta } \]

存在,則稱\(f^{`}(x_0)\)\(f(x)\)\(x=x_0\)處的導數。

  • 意義
    1. 導數的絕對值大小代表了當前函式的在該處的變化速度
    2. 導數的正負代表了在一定臨域內隨著自變數\(x\)的增加,函式值是增大還是減小

2. 偏導數

  • 定義

對於多元函式\(f(x),x \in R^p\)\(f(x)\)在對\(x_i\)的偏導數定義為

\[\frac{\partial f(x)}{\partial x_i}=\lim_{\Delta x \to 0}\frac{f(x_1,x_2,\cdots,x_i+\Delta x,\cdots,x_p)-f(x_1,x_2,\cdots,x_i,\cdots,x_p)}{\Delta x} \]

  • 意義

偏導數定義了多元函式在某個數軸方向上的變化情況。

3. 方向導數

  • 定義

函式的偏導數定義了在各個數軸上的變化率,方向導數則為函式在任意方向上的變化率。以二元函式\(f(x,y)\)為例:

\[\nabla \frac{\partial f(x)}{\partial l}|_{(x_0,y_0)}=\frac{\partial f(x)}{\partial x}\cos(\alpha)+\frac{\partial f(x)}{\partial y}\cos(\beta) \]

  • 意義

多元函式在某點處的方向導數有無數個,每一個方向導數的值代表了在該方向上的變化程度,我們要尋找在某點處函式變化最快的方向就可以轉化成尋找在該點處方向導數的絕對值最大時對應的那個方向

4. 梯度

  • 定義

梯度是一個向量,表示函式沿著該方向的變化率最大,記為

\[f(x)=(\frac{\partial f(x)}{\partial x_1},\frac{\partial f(x)}{\partial x_2},\cdots,\frac{\partial f(x)}{\partial x_p})^T \]

  • 為什麼該方向為變化最快的方向

根據方向導數定義,

\[\begin{align*} \frac{\partial f(x)}{\partial l}|_{(x_0,y_0)} &=\frac{\partial f(x)}{\partial x}\cos(\alpha)+\frac{\partial f(x)}{\partial y}\cos(\beta) \\ &= (\frac{\partial f(x)}{\partial x},\frac{\partial f(x)}{\partial y})(\cos(\alpha),\cos(\beta))^T \\ &= A\cdot I \quad\quad (A=(\frac{\partial f(x)}{\partial x},\frac{\partial f(x)}{\partial y}),I=(\cos(\alpha),\cos(\beta))^T ) \\ &= ||A||\times||I||\cos(\theta) \qquad (\theta為兩個向量的夾角) \end{align*} \]

當且僅當 \(\theta=0\),即\(A\)\(I\)通向時,方向導數取得最大值,因此梯度表示變化率最大的方向,此時方向導數為正。因此梯度指向函式增大的方向。

3 原理詳解

動圖

假設在一個類是凹函式的山中放一個小球,讓它自然的滾動到山谷(最小值點)處,那麼小球滾動每個地點滾動的方向都是梯度的負方向。

現在有一個凹函式,要找到它的最小值,在不考慮解析解的情況下,也可以利用類似的方法去求解。先隨機找一個初始點\(x_0\),然後求出該點的梯度,利用公式\(x_1=x_0-lr*\nabla f(x)\)模擬小球的滾動,其中\(lr\)為滾動的步長,也稱為學習率

動圖

通過迭代公式 \(x_n=x_{n-1}-lr* \nabla f(x)\)一步步去逼近函式的極小值點。通常迭代的結束條件有:

  • 指定迭代次數
  • 計算迭代前後函式值的差距,若在一個非常小的閾值以為就可以認為已經找到最小值

4. 程式碼實現

案例 :\(f(x)=(x_1-2)^2+(x_2-3)^2+(x_3-4)^4\)

import numpy as np
#定義函式
def func(x):
    return (x[0]-2)**2+(x[1]-3)**2+(x[2]-4)**2
#定義梯度
def gradFunc(x):
    return np.array([(x[0]-2)*2,(x[1]-3)*2,(x[2]-4)*2])
# 定義梯度下降法
def SGD(init_x,func,gradFunc,lr=0.01,maxIter=100000,error=1e-10):
    x=init_x
    for iter in range(0,maxIter):
        gd=gradFunc(x)
        x_new=x-lr*gd
        if(np.abs(func(x)-func(x_new))<error):
            return x_new
        x=x_new
    return x_new
SGD(np.array([1,1,1]),func,gradFunc) 

array([1.99998703, 2.99997406, 3.99996109])

SGD(np.array([10,10,10]),func,gradFunc)

array([2.00003215, 3.00002813, 4.00002411])

相關文章