MXNetのBinary Neural Netwok

Vengineerの戯言 : Twitter、Slideshare
SystemVerilogの世界へようこそ、すべては、SystemC v0.9公開から始まった

Research Spotlight: BMXNet – An Open Source Binary Neural Network Implementation Based On MXNet

MXNetのBinary Neural Network版。

基本的には、MXNetの各レイヤーが使えるが、一部、Binary対応のレイヤーに置き換える必要があるようですよ。

例えば、

　・Convolution    => QActivation + QConvolution

　・FullyConnected => QActivation + QFullyConnected

など、

Binaryなので、xnor や popcount を使うと。。。x86では、SSE4.2、ARMでは NEON でこれらはサポートしていると。

認識精度とモデルサイズは、

　・MNIST    : Lenet     ==  0.99/4.6MB  => 0.97/206KB
　・CIFAR-10 : ResNet-18 ==  0.90/44.7MB => 0.86/1.5MB

となると。

フレームワークでもBinary Neural Network が使えるのは、いいですね。