arXiv reaDer
教師なし異常の検出とローカリゼーションのための自己監視マスキング
Self-Supervised Masking for Unsupervised Anomaly Detection and Localization
最近、マルチメディアデータの異常検出とローカリゼーションが、機械学習コミュニティの間で大きな注目を集めています。医療診断や産業欠陥検出などの実際のアプリケーションでは、異常は画像の一部にのみ存在します。再構成ベースの異常検出アーキテクチャをローカライズされた異常に拡張するために、教師なし異常の検出とローカリゼーションのためのSelf-Supervised Masking(SSM)という名前の、ランダムマスキングと復元による自己監視学習アプローチを提案します。 SSMは、修復ネットワークのトレーニングを強化するだけでなく、推論時のマスク予測の効率を大幅に向上させます。ランダムマスキングにより、各画像はさまざまなトレーニングトリプレットのセットに拡張され、オートエンコーダーはトレーニング中にさまざまなサイズと形状のマスクを使用して再構成することを学習できます。推論時の異常検出とローカリゼーションの効率と有効性を向上させるために、正常領域を徐々に明らかにし、最終的に異常領域を特定する、新しいプログレッシブマスクリファインメントアプローチを提案します。提案されたSSMメソッドは、異常検出と異常ローカリゼーションの両方でいくつかの最先端技術を上回り、Retinal-OCTでそれぞれ98.3%AUC、MVTec ADで93.9%AUCを達成します。
Recently, anomaly detection and localization in multimedia data have received significant attention among the machine learning community. In real-world applications such as medical diagnosis and industrial defect detection, anomalies only present in a fraction of the images. To extend the reconstruction-based anomaly detection architecture to the localized anomalies, we propose a self-supervised learning approach through random masking and then restoring, named Self-Supervised Masking (SSM) for unsupervised anomaly detection and localization. SSM not only enhances the training of the inpainting network but also leads to great improvement in the efficiency of mask prediction at inference. Through random masking, each image is augmented into a diverse set of training triplets, thus enabling the autoencoder to learn to reconstruct with masks of various sizes and shapes during training. To improve the efficiency and effectiveness of anomaly detection and localization at inference, we propose a novel progressive mask refinement approach that progressively uncovers the normal regions and finally locates the anomalous regions. The proposed SSM method outperforms several state-of-the-arts for both anomaly detection and anomaly localization, achieving 98.3% AUC on Retinal-OCT and 93.9% AUC on MVTec AD, respectively.
updated: Fri May 13 2022 11:42:06 GMT+0000 (UTC)
published: Fri May 13 2022 11:42:06 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アソシエイト