Rustでディープラーニング

Transcript

1. Rustでディープラーニング ユニークビジョン株式会社 山本 一将
2. 自己紹介 2 ⚫ 名前：山本 一将（@kyamamoto9120） ⚫ 所属：ユニークビジョン株式会社 ⚫ 言語：C++, Rust, Python, Ruby ⚫ 趣味：将棋、キャンプ、DIY、野球観戦 ⚫ 実績：2015年 世界コンピュータ将棋選手権９位
3. はじめに 3 ① このスライドはconnpassにて公開します ② 実際の利用方法はQiitaに書きました
4. ディープラーニングの背景 4 ⚫ 学習・推論ともにPythonでの例がほとんど • ライブラリ・プラットフォームが充実している • 情報も多く、初学者にも優しい ⚫ 一方、C++は使われるケースがある • TensorFlow Servingのような推論環境 • 囲碁、将棋のようなゲームAI
5. Rustにおけるディープラーニング 5 ⚫ パフォーマンスが要求される場面で使いたい • C++は辛すぎる ⚫ 有名フレームワークのRustバインディングはある！ • 情報は皆無 • ドキュメントも乏しい
6. 有名フレームワークのRustバインディング 6 TensorFlow PyTorch tensorflow crate tch 公式 提供元 非公式 2.9k Star 942 ◦ GPUサポート ◦ ◦ モデル構築 ◦ ◦ 学習 ◦ ◦ Python製モデルで推論 ？
7. それぞれ使ってみた感想 7 ⚫ 環境構築 • CPUでちょっと使ってみるだけならtchが簡単 • 本格的に使う場合は差はない ⚫ モデル構築 • tchはPythonと比較的近いインタフェースで構築可能 • tensorflowは苦行 • 学習をRustでする場合でもモデル構築はPythonで行うべき
8. それぞれ使ってみた感想 8 ⚫ 学習 • tchはデータセットを扱う便利関数がある • tchには転移学習のサンプルもあって実用的 • tensorflowはサンプルだけでは何とも言えない ⚫ 事前学習済みモデルでの推論 • tensorflowはPythonで学習した重みを利用可能 • tchはPythonで学習した重みが使えるかは不明 • サンプルではOCaml版のモデルを使っている
9. まとめ 9 ⚫ Rustでもディープラーニングは可能 ⚫ ディープラーニング初学者が試しに使うならtch！ ⚫ 推論のみ高速化するならtensorflow！ ⚫ C++の代替として普及して欲しい
10. ありがとうございました