1. 程式人生 > >VAE變分自編碼器的一點理解

VAE變分自編碼器的一點理解

前言

網上很多關於VAE的教程都包含大量枯燥難懂的數學公式,如果有大佬特別喜歡推導這些公式,很高興能夠附上以下連結。然而,今天只是想簡單的談下自己的理解,只有圖片,沒有公式。

主要內容

如下圖所示,其實VAE的主要思想就是以均值方差計算模組來作為Encoder模組,在訓練階段,Encoder模組把n維訓練資料編碼成2xm維資料,其中包含m維均值以及m維方差,關於這點,可以看成是在學習訓練資料的分佈。然後從標準高斯分佈中取樣m維資料,並和上述編碼產生的均值和方差聯合計算,從而得出取樣變數Z。取樣變數Z繼續通過Decoder方式來解碼還原真實影象。並以此影象作為最終生成結果,和初始影象做損失計算。

通過上述方式,對編碼和解碼兩部分做訓練,當訓練完成之後,去掉編碼部分,這時m維均值和m維方差均已訓練結束,為特定值,只需要輸入標準高斯分佈資料,並和均值方差做運算,最終通過解碼器即可生成影象。

這裡寫圖片描述
這裡寫圖片描述
這裡寫圖片描述