arXiv reaDer
動的機能集約によるロバストな表現
Robust Representation via Dynamic Feature Aggregation
ディープ畳み込みニューラルネットワーク(CNN)ベースのモデルは、敵対的な攻撃に対して脆弱です。考えられる理由の1つは、CNNベースのモデルの埋め込みスペースがまばらであり、敵対的なサンプルを生成するためのスペースが大きくなることです。この研究では、動的特徴集約と呼ばれる、新しい正則化を使用して埋め込みスペースを圧縮する方法を提案します。特に、2つのサンプル間の凸結合は、集約の要と見なされます。埋め込みスペースでは、選択されたサンプルがピボットの表現と同様になるようにガイドされます。一方、このような正則化の簡単な解決策を軽減するために、モデルの最後の完全に接続されたレイヤーは、異なるクラスの埋め込みコードが直交して別々に処理される直交分類器に置き換えられます。正則化と直交分類器を使用すると、よりコンパクトな埋め込みスペースを取得できます。これにより、敵対的な攻撃に対するモデルの堅牢性が向上します。 56.91%の平均精度は、さまざまな攻撃方法に対するCIFAR-10の方法によって達成され、37.31%のマージンで堅実なベースライン(Mixup)を大幅に上回っています。さらに驚くべきことに、経験的な結果は、提案された方法が、学習されたコンパクトな特徴空間のために、分布外(OOD)検出のための最先端の性能も達成できることを示しています。分布(ID)データセットとしてCIFAR-10を採用し、OODデータセットとしてLSUNを採用した場合、提案された方法によって0.937のF1スコアが達成されます。コードはhttps://github.com/HaozheLiu-ST/DynamicFeatureAggregationで入手できます。
Deep convolutional neural network (CNN) based models are vulnerable to the adversarial attacks. One of the possible reasons is that the embedding space of CNN based model is sparse, resulting in a large space for the generation of adversarial samples. In this study, we propose a method, denoted as Dynamic Feature Aggregation, to compress the embedding space with a novel regularization. Particularly, the convex combination between two samples are regarded as the pivot for aggregation. In the embedding space, the selected samples are guided to be similar to the representation of the pivot. On the other side, to mitigate the trivial solution of such regularization, the last fully-connected layer of the model is replaced by an orthogonal classifier, in which the embedding codes for different classes are processed orthogonally and separately. With the regularization and orthogonal classifier, a more compact embedding space can be obtained, which accordingly improves the model robustness against adversarial attacks. An averaging accuracy of 56.91% is achieved by our method on CIFAR-10 against various attack methods, which significantly surpasses a solid baseline (Mixup) by a margin of 37.31%. More surprisingly, empirical results show that, the proposed method can also achieve the state-of-the-art performance for out-of-distribution (OOD) detection, due to the learned compact feature space. An F1 score of 0.937 is achieved by the proposed method, when adopting CIFAR-10 as in-distribution (ID) dataset and LSUN as OOD dataset. Code is available at https://github.com/HaozheLiu-ST/DynamicFeatureAggregation.
updated: Mon May 16 2022 06:22:15 GMT+0000 (UTC)
published: Mon May 16 2022 06:22:15 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アソシエイト