arXiv reaDer
リプシッツ連続性保持バイナリニューラルネットワーク
Lipschitz Continuity Retained Binary Neural Network
バイナリニューラルネットワークのパフォーマンスは、完全精度の重みベクトルとそれに対応するバイナリベクトル間の量子化誤差を排除することで大幅に復元できるという前提に基づいて、ネットワークの2値化の既存の作業では、前述の目的を達成するためにモデルの堅牢性の概念を採用することがよくあります。ただし、堅牢性は、確固たる理論的サポートがなければ、明確に定義されていない概念のままです。この作業では、BNNのモデルの堅牢性を定義するための厳密な基準として、明確に定義された機能プロパティであるリプシッツ連続性を紹介します。次に、モデルのロバスト性を改善するための正則化項としてリプシッツ連続性を保持することを提案します。特に、人気のあるリプシッツ関連の正則化法は、その極端な希薄性のためにBNNで崩壊することがよくありますが、保持行列は、ターゲットの重み行列のスペクトルノルムを近似するように設計されています。これは、BNNのリプシッツ定数の近似として展開できます。正確なリプシッツ定数計算(NPハード)。私たちの実験は、私たちのBNN固有の正則化手法が、BNN(ImageNet-Cでテスト済み)の堅牢性を効果的に強化し、CIFARおよびImageNetで最先端のパフォーマンスを実現できることを証明しています。
Relying on the premise that the performance of a binary neural network can be largely restored with eliminated quantization error between full-precision weight vectors and their corresponding binary vectors, existing works of network binarization frequently adopt the idea of model robustness to reach the aforementioned objective. However, robustness remains to be an ill-defined concept without solid theoretical support. In this work, we introduce the Lipschitz continuity, a well-defined functional property, as the rigorous criteria to define the model robustness for BNN. We then propose to retain the Lipschitz continuity as a regularization term to improve the model robustness. Particularly, while the popular Lipschitz-involved regularization methods often collapse in BNN due to its extreme sparsity, we design the Retention Matrices to approximate spectral norms of the targeted weight matrices, which can be deployed as the approximation for the Lipschitz constant of BNNs without the exact Lipschitz constant computation (NP-hard). Our experiments prove that our BNN-specific regularization method can effectively strengthen the robustness of BNN (testified on ImageNet-C), achieving state-of-the-art performance on CIFAR and ImageNet.
updated: Wed Jul 13 2022 22:55:04 GMT+0000 (UTC)
published: Wed Jul 13 2022 22:55:04 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アソシエイト