1. 程式人生 > >約束優化方法之拉格朗日乘子法與KKT條件

約束優化方法之拉格朗日乘子法與KKT條件

引言

本篇文章將詳解帶有約束條件的最優化問題,約束條件分為等式約束與不等式約束,對於等式約束的優化問題,可以直接應用拉格朗日乘子法去求取最優值;對於含有不等式約束的優化問題,可以轉化為在滿足 KKT 約束條件下應用拉格朗日乘子法求解。拉格朗日求得的並不一定是最優解,只有在凸優化的情況下,才能保證得到的是最優解,所以本文稱拉格朗日乘子法得到的為可行解,其實就是區域性極小值,接下來從無約束優化開始一一講解。

無約束優化

首先考慮一個不帶任何約束的優化問題,對於變數 x∈ℝNx∈RN 的函式 f(x)f(x) ,無約束優化問題如下:

minxf(x)minxf(x)

該問題很好解,根據 Fermat 定理,直接找到使目標函式得 0 的點即可 即 ∇xf(x)=0∇xf(x)=0 ,如果沒有解析解的話,可以使用梯度下降或牛頓方法等迭代的手段來使 xx 沿負梯度方向逐步逼近極小值點。

等式約束優化

當目標函式加上約束條件之後,問題就變成如下形式:

minx f(x)s.t.   hi(x)=0,i=1,2,...,mminx f(x)s.t.   hi(x)=0,i=1,2,...,m

約束條件會將解的範圍限定在一個可行域,此時不一定能找到使得 ∇xf(x)∇xf(x) 為 0 的點,只需找到在可行域內使得 f(x)f(x) 最小的值即可,常用的方法即為拉格朗日乘子法,該方法首先引入 Lagrange Multiplier α∈ℝmα∈Rm ,構建 Lagrangian 如下:

L(x,α)=f(x)+∑i=1mαihi(x)L(x,α)=f(x)+∑i=1mαihi(x)

求解方法如下:首先對 Lagrangian  關於 αα 與 xx 求 :

{∇xL(x,α)=0∇αL(x,α)=0{∇xL(x,α)=0∇αL(x,α)=0

令導數為 0 ,求得 xx 、αα  的值後,將 xx 帶入 f(x)f(x) 即為在約束條件 hi(x)hi(x) 下的可行解。這樣做的意義是什麼呢? 接下來看一個直觀的示例,對於二維情況下的目標函式是 f(x,y)f(x,y),在平面中畫出 f(x,y)f(x,y) 的等高線,如下圖的虛線所示, 並只給出一個約束等式 h(x,y)=0h(x,y)=0 ,如下圖的綠線所示,目標函式 f(x,y)f(x,y) 與約束 g(x,y)g(x,y) 只有三種情況,相交、相切或者沒有交集,沒交集肯定不是解,只有相交或者相切可能是解,但相交得到的一定不是最優值,因為相交意味著肯定還存在其它的等高線在該條等高線的內部或者外部,使得新的等高線與目標函式的交點的值更大或者更小,這就意味著只有等高線與目標函式的曲線相切的時候,才可能得到可行解.

1

因此給出結論:拉格朗日乘子法取得極值的必要條件是目標函式與約束函式相切,這時兩者的法向量是平行的,即

∇xf(x)–α∇xh(x)=0∇xf(x)–α∇xh(x)=0

所以只要滿足上述等式,且滿足之前的約束 hi(x)=0,i=1,2,…,mhi(x)=0,i=1,2,…,m ,即可得到解,聯立起來,正好得到就是拉格朗日乘子法。這裡只是直觀展示了一下拉格朗日乘子法的幾何推導 ,並沒有給出詳細的證明。

不等式約束優化

當約束加上不等式之後,情況變得更加複雜,首先來看一個簡單的情況,給定如下不等式約束問題:

minx f(x) s.t.  g(x)≤0minx f(x) s.t.  g(x)≤0

對應的 Lagrangian 與圖形分別如下所示:

L(x,λ)=f(x)+λg(x)L(x,λ)=f(x)+λg(x)

這時的可行解必須落在約束區域 g(x)g(x) 之內,下圖給出了目標函式的等高線與約束:

1

由圖可見可行解 xx 只能在 g(x)<0g(x)<0 或者 g(x)=0g(x)=0  的區域裡取得:

  • 當可行解 xx 落在 g(x)<0g(x)<0 的區域內,此時直接極小化 f(x)f(x) 即可;
  • 當可行解 xx 落在 g(x)=0g(x)=0 即邊界上,此時等價於等式約束優化問題.

當約束區域包含目標函式原有的的可行解時,此時加上約束可行解扔落在約束區域內部,對應 g(x)<0g(x)<0 的情況,這時約束條件不起作用;當約束區域不包含目標函式原有的可行解時,此時加上約束後可行解落在邊界 g(x)=0g(x)=0 上。下圖分別描述了兩種情況,右圖表示加上約束可行解會落在約束區域的邊界上。

743682-20160731123535997-311672087

以上兩種情況就是說,要麼可行解落在約束邊界上即得 g(x)=0g(x)=0 ,要麼可行解落在約束區域內部,此時約束不起作用,另 λ=0λ=0消去約束即可,所以無論哪種情況都會得到:

λg(x)=0λg(x)=0

還有一個問題是 λλ 的取值,在等式約束優化中,約束函式與目標函式的梯度只要滿足平行即可,而在不等式約束中則不然,若 λ≠0λ≠0,這便說明 可行解 xx 是落在約束區域的邊界上的,這時可行解應儘量靠近無約束時的解,所以在約束邊界上,目標函式的負梯度方向應該遠離約束區域朝向無約束時的解,此時正好可得約束函式的梯度方向與目標函式的負梯度方向應相同:

−∇xf(x)=λ∇xg(x)−∇xf(x)=λ∇xg(x)

上式需要滿足的要求是拉格朗日乘子 λ>0λ>0 ,這個問題可以舉一個形象的例子,假設你去爬山,目標是山頂,但有一個障礙擋住了通向山頂的路,所以只能沿著障礙爬到儘可能靠近山頂的位置,然後望著山頂嘆嘆氣,這裡山頂便是目標函式的可行解,障礙便是約束函式的邊界,此時的梯度方向一定是指向山頂的,與障礙的梯度同向,下圖描述了這種情況 :

2

可見對於不等式約束,只要滿足一定的條件,依然可以使用拉格朗日乘子法解決,這裡的條件便是 KKT 條件。接下來給出形式化的 KKT 條件 首先給出形式化的不等式約束優化問題:

minx f(x)s.t.   hi(x)=0, i=1,2,...,m           gj(x)≤0, j=1,2,...,nminx f(x)s.t.   hi(x)=0, i=1,2,...,m           gj(x)≤0, j=1,2,...,n

列出 Lagrangian 得到無約束優化問題:

L(x,α,β)=f(x)+∑i=1mαihi(x)+∑j=1nβigi(x)L(x,α,β)=f(x)+∑i=1mαihi(x)+∑j=1nβigi(x)

經過之前的分析,便得知加上不等式約束後可行解 xx 需要滿足的就是以下的 KKT 條件:

∇xL(x,α,β)βjgj(x)hi(x)gj(x)βj=0=0, j=1,2,...,n=0, i=1,2,...,m≤0, j=1,2,...,n≥0, j=1,2,...,n(1)(2)(3)(4)(5)(1)∇xL(x,α,β)=0(2)βjgj(x)=0, j=1,2,...,n(3)hi(x)=0, i=1,2,...,m(4)gj(x)≤0, j=1,2,...,n(5)βj≥0, j=1,2,...,n

滿足 KKT 條件後極小化 Lagrangian 即可得到在不等式約束條件下的可行解。 KKT 條件看起來很多,其實很好理解:

(1) :拉格朗日取得可行解的必要條件;

(2) :這就是以上分析的一個比較有意思的約束,稱作鬆弛互補條件;

(3) ∼∼ (4) :初始的約束條件;

(5) :不等式約束的 Lagrange Multiplier 需滿足的條件。

主要的KKT條件便是 (3) 和 (5) ,只要滿足這倆個條件便可直接用拉格朗日乘子法, SVM 中的支援向量便是來自於此,需要注意的是 KKT 條件與對偶問題也有很大的聯絡,下一篇文章就是拉格朗日對偶。

參考文獻: