arXiv reaDer
バイナリ ニューラル ネットワークの再帰型双一次最適化
Recurrent Bilinear Optimization for Binary Neural Networks
バイナリ ニューラル ネットワーク (BNN) は、実世界の組み込みデバイスに大きな可能性を示しています。強力な BNN を実現するための重要なステップの 1 つとして、スケール ファクターの計算は、実数値の対応するものとのパフォーマンス ギャップを縮小する上で重要な役割を果たします。ただし、既存の BNN は、実数値の重みとスケール係数の固有の双一次関係を無視しているため、トレーニング プロセスが不十分なために最適ではないモデルが生成されます。この問題に対処するために、逆伝播プロセスで固有の双線形変数を関連付けることにより、BNN (RBONN) の学習プロセスを改善する Recurrent Bilinear Optimization が提案されています。私たちの仕事は、バイリニアの観点から BNN を最適化する最初の試みです。具体的には、反復最適化と Density-ReLU を使用して、スパースな実数値の重みフィルターを順次バックトラックします。これは十分にトレーニングされ、制御可能な学習プロセスに基づいてパフォーマンスの限界に達します。堅牢な RBONN を取得し、さまざまなモデルやデータセットで最先端の BNN よりも優れたパフォーマンスを示します。特に、オブジェクト検出のタスクでは、RBONN は優れた一般化パフォーマンスを発揮します。コードは https://github.com/SteveTsui/RBONN でオープンソース化されています。
Binary Neural Networks (BNNs) show great promise for real-world embedded devices. As one of the critical steps to achieve a powerful BNN, the scale factor calculation plays an essential role in reducing the performance gap to their real-valued counterparts. However, existing BNNs neglect the intrinsic bilinear relationship of real-valued weights and scale factors, resulting in a sub-optimal model caused by an insufficient training process. To address this issue, Recurrent Bilinear Optimization is proposed to improve the learning process of BNNs (RBONNs) by associating the intrinsic bilinear variables in the back propagation process. Our work is the first attempt to optimize BNNs from the bilinear perspective. Specifically, we employ a recurrent optimization and Density-ReLU to sequentially backtrack the sparse real-valued weight filters, which will be sufficiently trained and reach their performance limits based on a controllable learning process. We obtain robust RBONNs, which show impressive performance over state-of-the-art BNNs on various models and datasets. Particularly, on the task of object detection, RBONNs have great generalization performance. Our code is open-sourced on https://github.com/SteveTsui/RBONN .
updated: Sun Sep 04 2022 06:45:33 GMT+0000 (UTC)
published: Sun Sep 04 2022 06:45:33 GMT+0000 (UTC)
参考文献 (このサイトで利用可能なもの) / References (only if available on this site)
被参照文献 (このサイトで利用可能なものを新しい順に) / Citations (only if available on this site, in order of most recent)
Amazon.co.jpアソシエイト