LoginSignup
14
21

深層学習のFPGA移植の3通りのアプローチ

Last updated at Posted at 2019-10-06

深層学習をFPGAに移植しようとするアプローチは大きく言って3つある。
その3つのアプローチについての紹介をする。

1:深層学習の行列演算の一部に対するアクセラレータを作成する。

 メインはC/C++のプログラムであり、その行列演算の一部をハードウェア実装を利用するC/C++のコードとして実現する。
 このアプローチでは、プログラムをビルドする必要があり、ネットワーク構造を変えた時には、ソースコードの違いがあるため、ビルドし直さなければならない可能性が高い。

長所:

  • 浮動小数点演算として実装している範囲だと、学習のやり直しが不要である。
    • そのため、推論部分の最適化だけに集中できる。

短所:

  • 浮動小数点演算に対するアクセラレータである場合、浮動小数点演算の計算コストは高い。
  • メイン計算部分と行列演算の一部を行うアクセラレータ間とでデータ転送を行う必要があり、その部分を含めたパフォーマンスの向上が必要になる。

情報源の1例:

2:深層学習の学習済みの重みファイルを他の環境に移植するツールの移植先の一つとしてFPGAの実装がある。

FPGAの実装部分は汎用ツールとして実装してあり、利用する重みファイルを変えることで実現するCNNの種類を変える。
浮動小数点モデルで訓練されたpre-trained モデルをint8などのモデルに変換することは、いくつかのベンダが自社のプラットフォーム上に提供している。それによって、各種model zoo を利用できる。顔照合の目的で、顔を検出して顔のランドマークを求めて、顔の属性推定をする。そういった枠組みの機械学習は、限られたリソースではつくり上げることができない。
だからこそ、既存のpre-trained モデルを利用できることは、重要な要素になっていて、浮動小数点モデルを計算量の少ないモデルに変換して、各プラットフォームで実行できることは重要だ。

長所

ツールがそのように作ってあるので、学習済みの重みファイルを元に移植先で実行可能になる。
FPGAで動作させるためのFPGA部分の設定は既にFPGA用にコンパイルしてあってbitstream をFPGAにダウンロードすればよいだけになっている場合もある。

少ない手数で、FPGAへの移植ができる場合がある(そのツールが提供している範囲で)。

CPUでの処理からFPGAに処理を外だしできるので、ARM系のCPUの場合利点がある。

### 短所

再学習なしに、浮動小数点での重みファイルをFPGA用に変えることができるということは、再学習をすれば、bit幅が減らせる可能性をあきらめるということでもある。そのために、再学習をして重みのbit幅を減らして実装するのに比べて計算量が多くなってしまう。

ただし、再学習できるほどのデータセットとbit幅が少ない学習を実行できる環境を持つことは極めて困難なので、この短所が、短所になりうる人は少ないだろう。

情報源の1例

OpenVino

FP11(Sign、指数5ビット、仮数5ビット) という選択肢がOpenVinoに登場してきている。

FP11 はINT8/9 よりもメリットがある•再トレーニング不要、より良いパフォーマンス、精度の損失が少ない
https://eventmarketing.blob.core.windows.net/mstechsummit2018-after/DA22_PDF_TS18.pdf

OpenVINO™ Toolkit / FPGA Plugin

Vitis (Xilinx)

  • AI オプティマイザー
  • AI クオンタイザー
  • AI コンパイラ

NVidia

Optimize Model, Build Engine for Inference
でも次のようにしてINT8にまでモデルを縮小できるにしている。

Path to UFF file for our model
Precision for inference engine (FP32, FP16, or INT8)
Calibration dataset (only needed if you’re running in INT8)
Batch size used during inference

3:量子化CNN、2値化CNN

長所:

計算量が少なくできる。任意のbit幅を実現しやすいFPGAならではの利用ができる。そのため、消費電力や回路規模を削減しやすい

短所:

既存の学習済みの結果が少ない。浮動小数点での学習結果をそのまま転用することはほぼできない。そのため、学習を実行する必要がある。

情報源の1例:

github https://github.com/HirokiNakahara/GUINNESS

Slideshare 2値化CNN on FPGAでGPUとガチンコバトル(公開版)

Slideshare 2値ディープニューラルネットワークと組込み機器への応用: 開発中のツール紹介

Slideshare ディープラーニングの2値化(Binarized Neural Network)

ニューラルネットワークの量子化についての最近の研究の進展と、その重要性

2値化CNNの1例

例題:MNISTのデータの学習
https://github.com/hillbig/binary_net

Xilinxでの例

https://github.com/Xilinx/BNN-PYNQ
以下のjupyter notebook を読むと、どのような分類・検出ができているのかがわかる。
https://github.com/Xilinx/BNN-PYNQ/tree/master/notebooks
https://github.com/Xilinx/BNN-PYNQ/blob/master/notebooks/CNV-BNN_Road-Signs.ipynb


未分類

深層学習の推論を実行する回路のハードウェア記述を生成するツール

NNgen


GUINESS


Ultra96ボードでYOLOを高速化


CNNのモデル削減に関する記事


第4回AIエッジコンテスト(実装コンテスト②) FPGAを使った自動車走行画像認識が開催されている。
このようなコンテストでどのように実装しているのかを調査したい。


追記 2023.10
githubなどにあるソースは、最後に改変してあるのがいつなのか、
いまメンテナンスされているのかどうかを確認しながら、方式の選択を考えよう。

14
21
4

Register as a new user and use Qiita more conveniently

  1. You get articles that match your needs
  2. You can efficiently read back useful information
  3. You can use dark theme
What you can do with signing up
14
21