EM算法原理詳解


1.引言

以前我們討論的概率模型都是只含觀測變量(observable variable), 即這些變量都是可以觀測出來的,那么給定數據,可以直接使用極大似然估計的方法或者貝葉斯估計的方法;但是當模型含有隱變量(latent variable)的時候, 就不能簡單地使用這些估計方法。

如在高斯混合和EM算法中討論的高斯混合就是典型的含有隱變量的例子,已經給出EM算法在高斯混合模型中的運用,下面我們來討論一些原理性的東西。


 

2.Jensen 不等式

是值域為實數的函數,那么如果,則就是一個凸函數,如果自變量 x 是向量, 那么當函數的海森矩陣 半正定時(), 是凸函數,這是函數為凸函數的條件在向量輸入時的泛化。

如果,則稱嚴格凸函數,對應的向量輸入時的泛化是.

定理  令是一個凸函數,令是一個隨機變量,那么

時嚴格凸函數的時,當且僅當 以概率 1 成立的時,. 即當時常量時,上面不等式的等號成立。

注意上面 E 是表示期望的意思,習慣上,在寫變量期望的時候,會把緊跟括號略去,即.

用下面的圖對上面的定理作一個解釋:

這個圖中的實線代表凸函數, 隨機變量有 0.5 的概率取 a, 同樣以 0.5 的概率取 b, 所以的期望位於a,b的正中間,即a,b的均值.

從圖中可以看出,在 y 軸上, 位於之間,因為是凸函數,則必如上圖所示,

所以很多情況下,許多人並去記憶這個不等式,而是記住上面的圖,這樣更容易理解。

注意:如果是(嚴格)凹函數,即使(嚴格)凸函數(即,),那么Jensen不等式照樣成立,只不過不等號方向相反:


 

3.EM算法

假設在一個估計問題中有m個獨立樣本,根據這些數據,希望擬合出模型的參數,那么對數似然函數:

這里,是隱變量,如果能夠被觀測出來,最大似然估計就會變得很容易,但是現在觀測不出來,是隱變量。

在這種情況下,EM算法給出了一種很有效的最大似然估計的方法:重復地構造的下界(E步),然后最大化這個下界(M步)

 

對於每個,令表示隱變量的分布,即,考慮:

由(2)到(3)的推導用到了上面的Jensen不等式,此時是一個凹函數,因為,考慮上面關於的分布

正好是數量的期望,由Jensen不等式可以得到:

由此可以從(2)推出(3).


 

但是由於隱變量的存在,直接最大化很困難!試想如果能讓直接與它的下界相等,那么任何可以使的下界增大的,也可以使增大,所以自然就是選擇出使的下界達到極大的參數.

怎么樣才能使得取得下界呢,即上面不等式取等號,關鍵在於隱變量如何處理,下面就此討論。

現在,對於任意的分布,(3)給出了似然函數的下界. 對於分布到底是什么分布,可以有很多種選擇,到底該選擇哪一種呢?

 

在上面討論Jensen不等式的時候可以看出,不等式中等號成立的條件是隨機變量變成“常量”,對於要想取得下界值,必須要求

其中常數 c 與變量 無關,這很容易做到,我們選擇分布的時候,滿足下面的條件即可:

由於,於是我們可以知道:

注意理解上面這個等式式子是如何得出來的!!

於是就可以把分布設定為:在參數下,給定后,的后驗分布。

這樣設定好隱變量的分布之后,就直接取其下界,原來最大化似然函數的問題轉換為最大化其下界,這就是E步!

在M步中,就是去調整參數最大化上面提到的式子(3).

不斷重復E步和M步就是EM算法:

重復迭代直至收斂{

  

}


 

我們如何知道算法收斂呢?

假如是兩次連續迭代后的參數,需要證明.

正如上面所述,由於我們再選擇分布時,選擇:,於是:

參數就是通過極大化上面右邊的式子得出,因此:

注意第不等式(4)來自於:

 

這個式子對於任意的都成立,當然對於也成立。對於不等式(5),因為是通過如下極大化過程選出來的:

所以在處,式子的值要比在處式子的值要大!

式子(6)是通過上面討論過的方法選擇出合適的使得Jensen不等式取等號!

因此,EM算法使得似然函數單調收斂。在上面描述EM算法的時候,說是“重復迭代直至收斂”,一個常用的檢查收斂的方法是:如果兩次連續迭代之后,似然函數的值變化很小(在某個可容忍的范圍內),就EM算法中的變化已經很慢,可以停止迭代了。

 

注意:如果定義:

從之前的推導,我們知道. EM算法看作是關於函數 J 的梯度上升:E步是關於參數Q,M步是關於參數.

 


 

4.高斯混合的修正

在 高斯混合和EM算法 中,我們將EM算法用於優化求解高斯混合模型,擬合參數.

E步:

這里表示的是在分布下,的概率。

M步:考慮參數,最大化數值:

 

最大化求,對上面的式子關於求偏導數:

令這個偏導數為0,求出的更新方式:

這是在 高斯混合和EM算法 中已經得出的結論。

再考慮如何更新參數,把只與有關的項寫出來,發現只需要最大化:

因為,,所有的和為1,所以這是一個約束優化問題,參考簡易解說拉格朗日對偶(Lagrange duality),構造拉格朗日函數:

 

其中 β 是拉格朗日乘子. 求偏導數:

令偏導數為0,得到:

即:利用約束條件:,得到:(注意這里用到:).

於是可以得到參數的更新規則:

關於參數的更新規則,以及整個EM算法如何運用到高斯混合模型的優化,請參考:高斯混合和EM算法

5.總結

所謂EM算法就是在含有隱變量的時候,把隱變量的分布設定為一個以觀測變量為前提條件的后驗分布,使得參數的似然函數與其下界相等,通過極大化這個下界來極大化似然函數,從避免直接極大化似然函數過程中因為隱變量未知而帶來的困難EM算法主要是兩步,E步選擇出合適的隱變量分布(一個以觀測變量為前提條件的后驗分布),使得參數的似然函數與其下界相等;M步:極大化似然函數的下界,擬合出參數.


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM