arXiv reaDer
表現の微調整による転移学習ガウス異常検出
Transfer Learning Gaussian Anomaly Detection by Fine-tuning Representations
現在の最先端の異常検出(AD)手法は、大規模なImageNetトレーニングによって生成された強力な表現を活用しています。ただし、壊滅的な忘却は、半教師あり設定の新しいデータセットで事前にトレーニングされた表現をうまく微調整することを妨げます。したがって、表現は一般的に修正されます。私たちの仕事では、壊滅的な忘却を克服し、したがって、転移学習設定でADの事前トレーニングされた表現をうまく微調整するための新しい方法を提案します。具体的には、生成モデリングと識別モデリングのリンクに基づいて、正規クラスの多変量ガウス分布を誘導し、正規画像から推定分布までのマハラノビス距離をトレーニングの目的として使用します。さらに、壊滅的な忘却の開始を検出するための検証スキームで、隣接リスクの最小化に一般的に使用される増強を使用することを提案します。公開されたMVTecデータセットの広範な評価により、ADタスクの方法によって新しい最先端が達成されると同時に、以前の最先端に匹敵する異常セグメンテーションパフォーマンスが達成されることが明らかになりました。さらに、アブレーション研究は、誘導されたガウス分布の重要性と、拡張の選択に関して提案された微調整スキームの堅牢性を示しています。
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-supervised setting, and representations are therefore commonly fixed. In our work, we propose a new method to overcome catastrophic forgetting and thus successfully fine-tune pre-trained representations for AD in the transfer learning setting. Specifically, we induce a multivariate Gaussian distribution for the normal class based on the linkage between generative and discriminative modeling, and use the Mahalanobis distance of normal images to the estimated 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 dataset reveal that a new state of the art is achieved by our method in the AD task while simultaneously achieving anomaly segmentation 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 Jun 13 2022 08:33:46 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アソシエイト