オブジェクト指向の申し子!?JavaでDeep Learningをやってみた(お試し編) #機械学習

FUJITSU Advent Calendar 2019 23日目の記事です。
オブジェクト指向の申し子的言語と言えばJavaなのではないでしょうか？歴史的にはC++やSmalltalkが先に世に出た様ですが、Cとの互換性を考慮して開発されたC++はオブジェクト指向による実装を強制しません。Smalltalkに至っては筆者は寡聞にしてその存在を知りませんでした。。。以前ある書籍で「再利用性を考慮したプログラムを比較的簡単に実装できる開発思想＝オブジェクト指向」を代表する言語はJavaと読んだ記憶が有りますが、あながち間違っていないのではないかと思います。人気プログラミング言語の変遷、1965～2019年（Python視点）からも分かりますが、ソフトウェア開発でも広く利用されて来た言語です。因みに社内のソフトウェア開発では現在でも圧倒的なシェア！？を握っている様です。(俗説で統計データは有りません)但し、近年はPythonの様な生粋のオブジェクト指向言語では無い？言語の台頭も有り、今後勢力図が書き換わるやもしれません。。。

やりたいこと

前置きがとても長くなりましたが、要はJavaでDeep Learningをやってみると言うのが本記事の内容です。

なぜJavaなのか？

競技プログラミングが三度の飯より好きな友人が、以前はJavaでプログラミングをやっていたのが、最近Rustに乗り換えたと言っていました。何でも流行の言語をやってみたかったとのこと。前掲の人気プログラミング言語の変遷からも分かる様にどうも他の言語に押され気味のJava。。。古い言語という様な表現する人も居たりと散々の扱い。。。オブジェクト指向の申し子がまだまだ現役で戦えるということを証明するには昨今流行のDeep Learningをやってのけるぐらいのことをやらなければと思いやってみました。但し、今回は表題にも有る通りお試し編です。

The Last Java Samurai¹

因みに筆者は非情報系の出身で、学生時代にJavaを勉強しようとして挫折し、入社後勉強してJavaの良さが初めて分かりました。

Java ∩ Deep Learning ∩ OSSが実は存在する

Deep Learningを含む機械学習の世界ではPythonが圧倒的なシェアを握り、大半のOSS機会学習用ライブラリもPythonで実装されていますが、以前からJavaでDeep Learningのモデルを用いて学習や推論が可能なOSS²は有りました。しかし、今回は2019年12月3日に新しくリリースされたOSS「Deep Java Library (以下DJL)³」を使ってみようと思います。

Deep Java Library (DJL) でDeep Learning (漸く本題)

記事が長くなってしまっていますが漸く本題です。現在の所、頼れる情報は公式サイトや参考記事だけの様です。

環境構築

参考記事ではGradleを用いた方法でAWS上で環境構築を行っていましたが、今回はローカル環境でDocker + Jupyter Notebookを用いてコンテナ上で実行する方法を採用したいと思います。Jupyter NotebookはPython用の可視化ツールというイメージが有りますが、最近は多言語対応が進んでおり、益々便利になっています。

実行環境

macOS Mojave 10.14.6(CPU:Corei5-8210Y 1.6GHz, DRAM:16GB)⁴
Docker 19.03.5

ソースのクローン

GitHubからリポジトリをクローンします。

git clone https://github.com/awslabs/djl.git

`djl`ディレクトリ配下の`jupyter`ディレクトリに移動

cd djl/jupyter/

コンテナの起動

docker container run -itd -p 127.0.0.1:8888:8888 -v $PWD:/home/jupyter deepjavalibrary/jupyter

Dockerfileを編集してイメージをビルドすることも出来ますが、今回はデフォルトのまま起動しました。

コンテナの起動確認

$ docker container ls
CONTAINER ID        IMAGE                     COMMAND                  CREATED             STATUS              PORTS                      NAMES
03791a7b1641        deepjavalibrary/jupyter   "jupyter notebook --…"   4 minutes ago       Up 4 minutes        127.0.0.1:8888->8888/tcp   suspicious_brattain

Jupyterを起動

http://localhost:8888にアクセスして以下のページが表示されれば成功です。

よく見るとチュートリアルが各々のノートブックとして既に格納済です。Javaでやると豪語しつつJupyter Notebookのお手軽さに屈してしまいました。。。

Multi Layer Perceptron (MLP) の実行

train_your_first_model.ipynbを起動します。MLPを用いた手書き文字認識(MNIST)のサンプルです。学習の実行はShit + Enterで順にセル上のコマンドを実行していくだけです。とてもお手軽に実行出来ます。main関数やclassの記述も不要なのには最初驚きました。

モデル保存時に以下のメッセージが出力されました。正答率が約97%ですが、これはあくまで学習時の評価用データでの数値で、今回テストデータによる正答率は測定していません。。。。

Model (
    Name: mlp
    Model location: /home/jupyter/build/mlp
    Data Type: float32
    Accuracy: 0.96991664
    Epoch: 2
)

Model Zoo(Deep Learning)で推論を実行

今回はオブジェクト認識のサンプルを実行してみます。object_detection_with_model_zoo.ipynbを起動します。前掲の参考記事でModel Zooを試していたため、別のサンプルを試そうと思ったのですが、ライブラリの詳細な使い方をまだ理解出来ていないのと、学習には矢張りGPU環境が必須であるため(CPUではMLPの学習でも時間が。。。)今回は学習済モデル(SSD:Single Shot MultiBox Detector)によるオブジェクト認識(推論)を試しました。

オブジェクト認識を以下の画像で試します。犬、自転車、車が写っている写真です。

推論結果

実行結果を以下に示します。正しくオブジェクト(犬、自転車、車)を認識出来ています。

推論結果(数値)を以下に示します。各々のオブジェクトが高い確度で認識されていることが分かります。

[
    class: "car", probability: 0.99991, bounds: [x=0.612, y=0.137, width=0.293, height=0.160]
    class: "bicycle", probability: 0.95365, bounds: [x=0.162, y=0.207, width=0.594, height=0.590]
    class: "dog", probability: 0.93471, bounds: [x=0.168, y=0.350, width=0.274, height=0.594]
]

まとめと今後の課題

JavaでDeep Learningが実行可能なJDLを試しました。AWS上での実行を想定している様ですが、ローカル環境上のDockerでも問題無く動作しました。今回はお試し編でしたが、今後調査を進めてもう少し高度なことに挑戦したいと思います。最後に今後の課題を以下に記載します。

GPU環境への導入(まずはGPUマシーンの構築から)
各種APIやモジュールの使い方を調査して自分で実装したモデルの学習、評価等まで実行

以上、オブジェクト指向の申し子によるDeep Learningの実行お試し編でした。来年も頑張って記事の投稿を続けていきたいと思います。

https://blog.heroku.com/samurai-duke-and-the-legend-of-openjdk ↩
Deeplearning4jです。日本語の書籍として「Deep Learning Javaプログラミング深層学習の理論と実装 (巣籠悠輔著; インプレス; 2016年)」が出版されています。 ↩
AWSでの利用を想定して開発された様です。(参考情報) ↩
今回はDJLを試してみることを目標としているため、GPU＋CUDAが動作する環境ではなく、CPU上で実行します。故に、今回の検証はMulti Layer Perceptronの学習やDeep Learningの学習済モデルを用いた推論を試すことに留めます。GPU上での動作は後日試してみることにします。 ↩

オブジェクト指向の申し子!?JavaでDeep Learningをやってみた(お試し編)