arXiv reaDer
一度限りの敵対的トレーニング:堅牢性と精度の間のその場でのトレードオフを無料で
Once-for-All Adversarial Training: In-Situ Tradeoff between Robustness and Accuracy for Free
敵対的なトレーニングとその多くのバリエーションは、ネットワークの深い堅牢性を大幅に向上させますが、標準の精度を犠牲にします。さらに、トレーニングプロセスは重いため、精度と堅牢性の間のトレードオフを徹底的に調査することは実用的ではなくなります。このホワイトペーパーでは、この新しい質問をします。トレーニングされたモデルをその場ですばやく調整し、何度も(再)トレーニングせずに、標準精度と堅牢な精度の間で達成可能なトレードオフを調べる方法は?提案されたフレームワークであるOnce-for-allAdversarial Training(OAT)は、入力として制御ハイパーパラメーターを使用した革新的なモデル条件付きトレーニングフレームワークに基づいて構築されています。トレーニングされたモデルは、テスト時に「無料」でさまざまな標準精度と堅牢な精度の間で調整できます。重要なノブとして、デュアルバッチ正規化を利用して、標準機能と敵対機能の統計を分離し、パフォーマンスを低下させることなく1つのモデルでそれらを学習できるようにします。さらに、OATを1回限りの敵対的トレーニングおよびスリミング(OATS)フレームワークに拡張します。これにより、精度、堅牢性、および実行時効率の間の共同トレードオフが可能になります。実験によると、再トレーニングやアンサンブルを行わなくても、OAT / OATSは、さまざまな構成で専用にトレーニングされたモデルと比較して、同等またはさらに優れたパフォーマンスを実現します。コードと事前トレーニング済みモデルは、https://github.com/VITA-Group/Once-for-All-Adversarial-Trainingで入手できます。
Adversarial training and its many variants substantially improve deep network robustness, yet at the cost of compromising standard accuracy. Moreover, the training process is heavy and hence it becomes impractical to thoroughly explore the trade-off between accuracy and robustness. This paper asks this new question: how to quickly calibrate a trained model in-situ, to examine the achievable trade-offs between its standard and robust accuracies, without (re-)training it many times? Our proposed framework, Once-for-all Adversarial Training (OAT), is built on an innovative model-conditional training framework, with a controlling hyper-parameter as the input. The trained model could be adjusted among different standard and robust accuracies "for free" at testing time. As an important knob, we exploit dual batch normalization to separate standard and adversarial feature statistics, so that they can be learned in one model without degrading performance. We further extend OAT to a Once-for-all Adversarial Training and Slimming (OATS) framework, that allows for the joint trade-off among accuracy, robustness and runtime efficiency. Experiments show that, without any re-training nor ensembling, OAT/OATS achieve similar or even superior performance compared to dedicatedly trained models at various configurations. Our codes and pretrained models are available at: https://github.com/VITA-Group/Once-for-All-Adversarial-Training.
updated: Tue Nov 10 2020 08:18:58 GMT+0000 (UTC)
published: Thu Oct 22 2020 16:06:34 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アソシエイト