看到一篇寫的非常好的關於 word2vec 文章《word2vec中的數學原理詳解》[1],本來想着看完之後按自己對知識的整理做一下記錄,但是覺得原文作者(peghoty)寫的太好了,這裏按作者寫作邏輯記錄一下個人的學習筆記吧。
目錄
三、基於Hierarchical Softmax的CBOW 模型
一、統計語言模型
(一)貝葉斯公式
貝葉斯公式應該都熟悉:
(二)統計語言模型
例:藉助於貝葉斯公式給出了在語音識別系統中,給定語音段 ,找到使得概率 最大的文本 ,公式如下:
這裏 是聲學模型, 是語言模型,這裏我們只關注於語言模型的計算。
簡單來說,統計語言模型是構建一個基於語料庫的用來計算句子概率的概率模型。假設 表示 個詞 按順序構成的句子,則 的聯合概率是:
利用貝葉斯公式上式可被鏈式分解爲:
用通用的白話說就是:長句的概率 第 1 個詞的概率 第 1 個詞出現的情況下第 2 個詞的概率 第 1 、 2 個詞出現的情況下第 3 個詞的概率 之前所有詞出現的情況下第 T 個詞的概率。
這種計算方式存在的問題是計算這些概率需要的成本太高。引用原作者的話:“假設語料庫對應詞典 的代銷(即詞彙量)爲 ,那麼,如果考慮長度爲 的任意句子,理論上就有 種可能,而每種可能都要計算 個概率,總共就需要計算 個概率”。針對該問題,常見的方法有 模型和神經網絡兩種方法。
二、n-gram 模型
(二)大數定理
不失一般性地,我們從上述利用貝葉斯公式鏈式分解的 中抽取一個概率 ,利用貝葉斯公式,有:
這裏, 和 分別是當語料庫足夠大時,根據大數定理表示的 和 在語料中出現的次數,當 很大時,計算 和 所耗的時間很長( 表示語料庫中同時包含詞 的語料數量,這意味着在計數過程中需要對每條語料進行判斷 次)。
(二)n-gram
首先得提一下馬爾科夫假設:下一個詞的出現僅依賴於它前面的 個詞。現在我們可以把概率 近似表示:
該方法可顯著降低計算複雜度,實際應用中最常採用的是 的三元模型(即依賴於該詞之前的兩個詞)。
三、基於Hierarchical Softmax的CBOW 模型
(一)網絡結構
論文《Exploiting Similarities among Languages for Machine Translation》[2]給出了 CBOW 模型的網絡結構,如下圖1 :
該網絡結構包括三層:輸入層、投影層和輸出層。模型是在已知當前詞 的上下文 的前提下預測當前詞 ,即計算 ,這裏 表示詞 的上下文集合。
- 輸入層:當前詞 的上下文 對應的詞向量 。這裏 表示詞向量的長度, 表示詞向量空間。
- 投影層:將輸入層的 4 個向量做求和累加,即 。
- 輸出層:輸出層對應一棵二叉樹,它是以語料中出現過的詞當葉子結點,以各詞在語料中出現的次數當權值構造出來的 Huffman 樹。在這棵 Huffman 樹中,葉子結點共 個,分別對應 中的詞,非葉子結點 個(葉子結點兩兩合併得到的新結點數)。(具體內容見另一篇博客[3])
下圖2 給出一個特殊的 CBOW 網絡結構:
(二)梯度計算
原作者給出了若干相關記號,這裏稍微修改一下,就上圖2 ,我們考慮詞 爲“夏天”的情況,記:
- :從根結點出發到達 對應葉子結點的路徑(38→23→9→4→3)。
- :路徑 中包含結點的個數(5 個:38,23,9,4,3)。
- :路徑 中的第 個節點,其中 表示根結點, 表示詞 對應的結點( 表示 38 對應的結點)。
- :詞 的 Huffman編碼,根結點不對應編碼,故沒有 ( )。
- :路徑 中非葉子結點對應的向量(以“夏天”爲例,共有 個非葉子結點)。
爲什麼用 Huffman 樹作爲 CBOW 的輸出呢?原作者的原話如下:
以圖2 中詞 是“夏天”爲例,從根結點出發到“夏天”這個葉子結點,中間共經歷了 4 詞分支(每天紅色的邊對於一次分支),而每次分支都可視爲進行了一次二分類。
既然是從二分類的角度來考慮問題,那麼對於每一個非葉子結點,就需要爲其左右子結點制定一個類別,即哪個是正類(標籤爲 1 ),哪個是負類(標籤爲 0 )。碰巧,除根結點以外,數中每個結點都對應了一個取值爲 0 或 1 的 Huffman 編碼。因此,一種最自然的做法就是將 Huffman 編碼爲 1 的結點定義爲正類,編碼爲 0 的結點定義爲負類。當然,這只是個約定而已,你也可以將編碼爲 1 的結點定義爲負類,而將編碼爲 0 的結點定義爲正類。事實上, word2vec 選用的就是後者,即約定:
簡而言之,將一個結點進行分類時,分到左邊就是負類,分到右邊就是正類。
使用 激活函數,一個結點被分爲正類的概率是:
被分爲負類的概率也就是:
最後,對於詞典 中的任意詞 , Huffman 樹中必存在一條從根結點到詞 對應結點的路徑 (且唯一)。 路徑 上存在 個分支,將每個分支看做一次二分類,每一次分類就產生一個概率,將這些概率乘起來,就是所需的 。
注意到這裏的 是路徑 中非葉子結點對應的向量,熟悉 BP 神經網絡(不懂的可以看我的另一篇博客[4])的讀者應該就知道使用隨機梯度下降法需要優化的參數了。
在 BP 神經網絡中,輸出層的結果與真實值計算均方誤差(MSE),並通過誤差的反向傳播方式來降低誤差,從而優化神經網絡中的權重矩陣;在 CBOW 模型中,輸出層的結果與真實值之間的均方誤差被替換爲概率 ,我們的目標變爲提高概率,故通過隨機梯度上升法來對非葉子結點向量集 進行優化。繼而對詞 的詞向量進行優化。
公式有:
[1] https://www.cnblogs.com/peghoty/p/3857839.html
[2] Mikolov T , Le Q V , Sutskever I . Exploiting Similarities among Languages for Machine Translation[J]. computer science, 2013.
[3] https://blog.csdn.net/qq_41297934/article/details/105169804
[4] https://blog.csdn.net/qq_41297934/article/details/104863418