またーり くおんつ ノート

お勉強した初歩的なことを書くチラシの裏

Predictive State Recurrent Neural Networksを読んだ

NIPS 4本目.

今日もタイトルに惹かれて読んでみた.

[1705.09353] Predictive State Recurrent Neural Networks

概要

状態空間モデルをRNN系に拡張したというもの.
その中でPSRNNというLayerを提案して、多層構造の学習手法についても提案.

LSTMやGRU対比PTBやMotionのデータセットでOutperformした.

雰囲気

雰囲気のイメージ図はこんな感じだけど、数式無いとなんの説明にもなってないな.
(数式あると説明になるとは言っていない. てかすごい分かりにくかった.)

f:id:aki-don:20171102085933p:plain

Factorized PSRNNも提案している.
BPTTで出来るのがウリっぽい?けど普通の方が精度がいいらしくて、うーん...

f:id:aki-don:20171102090504p:plain

Penn Tree Bankに対して適応してみた例.
確かに実験としてはいい結果に見える.

f:id:aki-don:20171102090203p:plain

OpenAIのSwimmerやUCIリポジトリのHandwritingに対しての実験もしている.
良好な結果.

f:id:aki-don:20171102090354p:plain

所感

確かに実験としてはいい結果が出てるけど、追試してみないと何ともという感じ.

状態空間モデルマニアにはたまらないかもしれん.

著者実装公開してくれないかな.
(軽くググッたけど見つからず...)