arXiv reaDer
敵対的特徴摂動を介した画像スタイルへのロバストネスのエンコード
Encoding Robustness to Image Style via Adversarial Feature Perturbations
敵対的訓練は、小さな敵対的摂動に対してロバストなモデルを作成するための業界標準です。ただし、機械学習の実践者は、入力画像のスタイルや照明の変更など、自然に発生する他の種類の変更に対して堅牢なモデルを必要としています。入力分布のこのような変化は、深い画像の特徴の平均と分散の変化として効果的にモデル化されています。画像ピクセルではなく、特徴統計を直接摂動させることによって敵対的トレーニングを適応させ、さまざまな目に見えない分布シフトに対してロバストなモデルを生成します。敵対的な特徴を視覚化することにより、これらの摂動と分布シフトの関係を調査します。提案された方法であるAdversarialBatch Normalization(AdvBN)は、トレーニング中に最悪の場合の特徴の摂動を生成する単一のネットワーク層です。敵対的な特徴分布でニューラルネットワークを微調整することにより、スタイルの変化や画像の破損など、さまざまな目に見えない分布の変化に対するネットワークの堅牢性が向上することを確認します。さらに、提案された敵対的特徴の摂動は、既存の画像空間データ拡張方法を補完し、パフォーマンスの向上につながる可能性があることを示します。ソースコードと事前トレーニング済みモデルは、https://github.com/azshue/AdvBNでリリースされています。
Adversarial training is the industry standard for producing models that are robust to small adversarial perturbations. However, machine learning practitioners need models that are robust to other kinds of changes that occur naturally, such as changes in the style or illumination of input images. Such changes in input distribution have been effectively modeled as shifts in the mean and variance of deep image features. We adapt adversarial training by directly perturbing feature statistics, rather than image pixels, to produce models that are robust to various unseen distributional shifts. We explore the relationship between these perturbations and distributional shifts by visualizing adversarial features. Our proposed method, Adversarial Batch Normalization (AdvBN), is a single network layer that generates worst-case feature perturbations during training. By fine-tuning neural networks on adversarial feature distributions, we observe improved robustness of networks to various unseen distributional shifts, including style variations and image corruptions. In addition, we show that our proposed adversarial feature perturbation can be complementary to existing image space data augmentation methods, leading to improved performance. The source code and pre-trained models are released at https://github.com/azshue/AdvBN.
updated: Sun Oct 31 2021 22:59:17 GMT+0000 (UTC)
published: Fri Sep 18 2020 17:52: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アソシエイト