arXiv reaDer
AGKD-BML:注意誘導知識蒸留と双方向メトリック学習による敵対的攻撃に対する防御
AGKD-BML: Defense Against Adversarial Attack by Attention Guided Knowledge Distillation and Bi-directional Metric Learning
ディープニューラルネットワークは多くのタスクで優れたパフォーマンスを示していますが、慎重に設計された敵対攻撃に対して脆弱です。注意誘導知識蒸留と双方向メトリック学習(AGKD-BML)による新しい敵対的トレーニングベースのモデルを提案します。注意の知識は、教師モデルと呼ばれるクリーンなデータセットでトレーニングされた重み固定モデルから取得され、学生モデルと呼ばれる敵対的な例(AE)でトレーニング中のモデルに転送されます。このようにして、学生モデルは正しい領域に焦点を合わせることができ、AEによって破損した中間機能を修正して、最終的にモデルの精度を向上させることができます。さらに、特徴空間での表現を効率的に正則化するために、双方向メトリック学習を提案します。具体的には、クリーンなイメージが与えられると、最初に最も混乱しているクラスに攻撃されて、フォワードAEを取得します。次に、最も混乱しているクラスのクリーンな画像がランダムに選択され、元のクラスに攻撃されて、後方AEが取得されます。次に、トリプレット損失を使用して、元の画像とそのAEの間の表現距離を短くし、前方と後方のAEの間の表現距離を拡大します。攻撃が異なる2つの広く使用されているデータセットに対して、広範な敵対的堅牢性実験を実施します。私たちが提案するAGKD-BMLモデルは、常に最先端のアプローチを上回っています。 AGKD-BMLのコードは、https://github.com/hongw579/AGKD-BMLで入手できます。
While deep neural networks have shown impressive performance in many tasks, they are fragile to carefully designed adversarial attacks. We propose a novel adversarial training-based model by Attention Guided Knowledge Distillation and Bi-directional Metric Learning (AGKD-BML). The attention knowledge is obtained from a weight-fixed model trained on a clean dataset, referred to as a teacher model, and transferred to a model that is under training on adversarial examples (AEs), referred to as a student model. In this way, the student model is able to focus on the correct region, as well as correcting the intermediate features corrupted by AEs to eventually improve the model accuracy. Moreover, to efficiently regularize the representation in feature space, we propose a bidirectional metric learning. Specifically, given a clean image, it is first attacked to its most confusing class to get the forward AE. A clean image in the most confusing class is then randomly picked and attacked back to the original class to get the backward AE. A triplet loss is then used to shorten the representation distance between original image and its AE, while enlarge that between the forward and backward AEs. We conduct extensive adversarial robustness experiments on two widely used datasets with different attacks. Our proposed AGKD-BML model consistently outperforms the state-of-the-art approaches. The code of AGKD-BML will be available at: https://github.com/hongw579/AGKD-BML.
updated: Fri Aug 13 2021 01:25:04 GMT+0000 (UTC)
published: Fri Aug 13 2021 01:25: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アソシエイト