TensorFlow.jsに保存機能が実装された件

最近話題のTensorFlow.jsですが、ついに待ちに待ったモデルの保存機能が実装されました。これでWebGLを使ったGPU計算が捗りますね！ということで、とりあえず使ってみました。

Transcript

1. TensorFlow.jsにモデルの保 存機能が実装された件 Meguro.es #15 @ Drecom 2018/5/22 @niisan-tokyo
2. ワタシ • 右の写真の顔をしたおっさん • エンジニア歴 ６年くらい • PHPer歴 ≒ エンジニア歴 • 現在の所属会社、ニジボックス歴は2.5年くら い • 最近のブームは機械学習で音楽生成するって いうテーマ • PCを買い替えてしまった
3. PHPerだけど、ちゃんとJavaScript の話ししますね
4. TensorFlow.js
5. TensorFlow.jsとは • WebGLを使って、間接的にGPUを使えるようにした、ブラウザで動かすための機 械学習ライブラリ • 記法がTensorFlowに寄せられているため、TensorFlowに慣れている人だったら、 簡単に習熟できる ( かもしれない ) • Kerasのレイヤーも取り込んでいるので、Kerasに慣れている人でも簡単に習熟で きる ( かもしれない ) • 元deeplearn.js
6. 改名してTensorFlowのファミリーになりました https://speakerdeck.com/niisantokyo/deeplearnjsfalseshao-jie 発表後１ヶ月程度でout of dateと なったワタシの発表
7. 余波 https://github.com/wan2land/php-tensorflow ( 意訳 )このプロジェクトはWebでTensorflowするためのものだったんだけど、 tensorflow.jsっていうもっと良いものができたんで、そっちつかうといいよ！このプロジェク トはここまでだぜ！ マジかよ。。。
8. TensorFlow.jsになってできるようになったこと • TensorFlow ( python )で出力された学習済みモデルを直接イ ンポートできるようになった ( pip instal tensorflojs ) • Kerasの出力したモデルもインポートできるようになった • モデルを保存することができるようになった ( 現時点から4日 ほど前 )
9. やったぜ！ セーブ機能が入りました！！ セーブ先は • local storage • indexedDB • download • HTTP/HTTPS request
10. 使ってみる
11. save_test.js 今回はlocal storageに データをしまってみる
12. load_test.js modelをその ままロードし ている図
13. JSで機械学習のナニモカもができてしまう 時代が来てしまった！！
14. ちょっとあれな点 • model.fit 中はプログレスバーのような途中経過を出す機能が まだ実装されていない ( not implement?) • KerasやTensorFlowの学習済みモデルを一旦別のライブラリ で変換する必要がある ( これは仕方ない ) • local storage には容量制限がある • Conv1D でsave and loadしたら、バグった
15. エラー 時系列データを１次元畳み込みフィルターをかけて解析しようとしているやつ 生成した時は普通に動く
16. エラー 普通にモデルを呼んで学習を再開しようとしているシーン
17. エラー んー？ 最後の層の形状が [1600, 1] になるべきって書いてある？ これ以外にもエラーが発生している箇所があるらしい。 まだ畳み込みは早かったかも
18. まとめ • deeplearn.jsがTensorFlow.jsになってました。 • TensorFlow.jsになったら、Kerasの書き方が使えるようになり ました。 • TensorFlow, Kerasの学習済みモデルをロードすることができ るようになりました ( convertの必要あり ) • モデルのセーブ機能が実装されました。 • セーブ機能はまだ発展途上に思います
19. 以上、お疲れ様でした