RBM学習結果の可視化

MNIST文字画像データを与えて、エッジの重みを可視化するというのをやってみました。

自分で書いたjuliaのコードは学習がうまくいかなかったので、「ゆるふわ Restricted Boltzmann Machine」のコードでやってみました。

実行結果

(step1)

(step2)

(step5)

(step10)

(step50)

気になること

• それっぽい出力ではあるけど、deep learning tutorialにあるこの画像と違う
• 「ゆるふわ」さんのコードを実行すると僕の書いたJuliaより数十倍実行速度が速い

何が原因なのか、引き続き調べてみます。