1. 程式人生 > >EM演算法之求解三硬幣模型

EM演算法之求解三硬幣模型

  看了好幾天的EM演算法,還是看的一頭霧水。藉由三硬幣模型,嘗試使用EM演算法。

1、EM演算法流程

  1. E步:對完全資料的對數似然函式log(P(Y,Z|θ))求關於P(Z|Y,θ(i))的數學期望。
     EZ|Y,θ(i)[log(P(Y,Z|θ))]
     其中θ(i)是第i次迭代時,θ的估計值
  2. M步:對E步的結果求極值

2、案例:三硬幣模型

  假設有3枚硬幣,分別記作A,B,C。這些硬幣正面出現的概率分別是π,p,

q,進行如下擲硬幣實驗:先擲硬幣A,根據其結果選出硬幣B或C,正面選B,反面選C,然後擲選出的硬幣,擲硬幣的結果,出現正面記作1,反面記作0;獨立重複n次實驗,觀測記為Y=y1,y2,...,yn

3、EM演算法求解

3.1)符號標記:

    yj為第j次實驗的觀測
    Z為隱變數,表示擲硬幣A出現的結果。該變數只有兩個取值0,1
    zj為第j次實驗時,擲硬幣A出現的結果,同樣的,zj=1表示硬幣A擲出正面
    θ表示引數集合π,p,q
    θ(i)為第i次迭代時,π

,p,q的估計值

3.2)E-Step

  完全資料的對數似然函式為:
    

log(P(Y,Z|θ))=log(j=1np(yj,zj|θ))=j=1nlog(p(yj,zj|θ))
    期望為: