arXiv reaDer
Mean-Shifted Contrastive Loss for Anomaly Detection
詳細な異常検出方法は、正常なサンプルと異常なサンプルを区別する表現を学習します。非常に効果的な表現は、外部で訓練された強力な特徴抽出器 (たとえば、ImageNet で事前訓練された ResNets) が、正常なサンプルで異常のないトレーニング データで微調整された場合に得られます。ただし、これは、壊滅的な崩壊に見舞われる可能性のある困難な作業です。つまり、些細で非特定の機能を学習する傾向があります。この論文では、中心損失法と対照損失法の両方の故障モードを克服できる新しい損失関数を提案します。さらに、以前の作業で使用されたユークリッド距離を置き換える、予測の信頼度に敏感な信頼度不変の角度中心損失とそれを組み合わせます。私たちの改善により、Mean-Shifted Contrastive Loss に基づいた新しい異常検出アプローチが生まれました。これは、以前の方法よりも正確で、壊滅的な崩壊に対する感度が低くなります。私たちの方法は、CIFAR-10 データセットの 97.5% ROC-AUC を含む複数のベンチマークで最先端の異常検出パフォーマンスを実現します。
Deep anomaly detection methods learn representations that separate between normal and anomalous samples. Very effective representations are obtained when powerful externally trained feature extractors (e.g. ResNets pre-trained on ImageNet) are fine-tuned on the training data which consists of normal samples and no anomalies. However, this is a difficult task that can suffer from catastrophic collapse, i.e. it is prone to learning trivial and non-specific features. In this paper, we propose a new loss function which can overcome failure modes of both center-loss and contrastive-loss methods. Furthermore, we combine it with a confidence-invariant angular center loss, which replaces the Euclidean distance used in previous work, that was sensitive to prediction confidence. Our improvements yield a new anomaly detection approach, based on Mean-Shifted Contrastive Loss, which is both more accurate and less sensitive to catastrophic collapse than previous methods. Our method achieves state-of-the-art anomaly detection performance on multiple benchmarks including 97.5% ROC-AUC on the CIFAR-10 dataset.
updated: Mon Jun 07 2021 17:58:03 GMT+0000 (UTC)
published: Mon Jun 07 2021 17:58:03 GMT+0000 (UTC)
参考文献 (このサイトで利用可能なもの) / References (only if available on this site)
被参照文献 (このサイトで利用可能なものを新しい順に) / Citations (only if available on this site, in order of most recent)アソシエイト