arXiv reaDer
微調整表現による転移学習ガウス異常検出
Transfer Learning Gaussian Anomaly Detection by Fine-Tuning Representations
現在の最先端の異常検出(AD)メソッドは、大規模なImageNetトレーニングによって生成された強力な表現を活用します。ただし、壊滅的な忘却は、半教師あり/監視なしの設定で新しいデータセットの事前トレーニングされた表現の微調整を成功させることを妨げます。したがって、表現は一般的に修正されます。私たちの仕事では、転移学習環境でADの学習表現を微調整する新しい方法を提案します。生成モデリングと識別モデリングの間のリンクに基づいて、正規クラスの多変量ガウス分布を誘導し、正規画像の分布までのマハラノビス距離をトレーニングの目的として使用します。さらに、壊滅的な忘却の開始を検出するための検証スキームで、ビシナルリスクの最小化に一般的に使用される増強を使用することを提案します。公開されているMVTecADデータセットの広範な評価により、ADタスクの方法によって新しい最先端が達成されると同時に、以前の最先端に匹敵するASパフォーマンスが達成されることが明らかになりました。さらに、アブレーション研究は、誘導されたガウス分布の重要性と、拡張の選択に関して提案された微調整スキームの堅牢性を示しています。
Current state-of-the-art Anomaly Detection (AD) methods exploit the powerful representations yielded by large-scale ImageNet training. However, catastrophic forgetting prevents the successful fine-tuning of pre-trained representations on new datasets in the semi/unsupervised setting, and representations are therefore commonly fixed. In our work, we propose a new method to fine-tune learned representations for AD in a transfer learning setting. Based on the linkage between generative and discriminative modeling, we induce a multivariate Gaussian distribution for the normal class, and use the Mahalanobis distance of normal images to the distribution as training objective. We additionally propose to use augmentations commonly employed for vicinal risk minimization in a validation scheme to detect onset of catastrophic forgetting. Extensive evaluations on the public MVTec AD dataset reveal that a new state of the art is achieved by our method in the AD task while simultaneously achieving AS performance comparable to prior state of the art. Further, ablation studies demonstrate the importance of the induced Gaussian distribution as well as the robustness of the proposed fine-tuning scheme with respect to the choice of augmentations.
updated: Mon Aug 09 2021 15:29:04 GMT+0000 (UTC)
published: Mon Aug 09 2021 15:29: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アソシエイト