arXiv reaDer
生徒と教師のピラミッドマッチングに注意を払った復興生徒
Reconstruction Student with Attention for Student-Teacher Pyramid Matching
異常の検出とローカリゼーションは、コンピュータービジョンの重要な問題です。最近、畳み込みニューラルネットワーク(CNN)が目視検査に使用されています。特に、異常なサンプルが不足しているため、このタスクの難易度が高くなり、教師なし学習ベースの方法が注目されています。エポック数の少ない通常の画像からのみトレーニングできるStudent-TeacherFeature Pyramid Matching(STPM)に焦点を当てています。ここでは、STPMの欠点を補う強力な方法を提案しました。提案された方法は、2人の学生と2人の教師で構成され、学生と教師のネットワークのペアはSTPMと同じです。他の学生と教師のネットワークには、通常の製品の機能を再構築する役割があります。異常画像から正常な製品の特徴を再構築することで、それらの差をとることで異常をより正確に検出することができます。新しい学生-教師ネットワークは、注意モジュールと元のSTPMとは異なる教師ネットワークを使用します。アテンションメカニズムは、入力画像の正常な領域を正常に再構築するように機能します。教師ネットワークが異なると、元のSTPMと同じ地域を見ることができなくなります。 2つの生徒と教師のネットワークから取得した6つの異常マップを使用して、最終的な異常マップを計算します。機能を再構築するための学生と教師のネットワークは、元のSTPMと比較して、ピクセルレベルと画像レベルのAUCスコアを改善しました。
Anomaly detection and localization are important problems in computer vision. Recently, Convolutional Neural Network (CNN) has been used for visual inspection. In particular, the scarcity of anomalous samples increases the difficulty of this task, and unsupervised leaning based methods are attracting attention. We focus on Student-Teacher Feature Pyramid Matching (STPM) which can be trained from only normal images with small number of epochs. Here we proposed a powerful method which compensates for the shortcomings of STPM. Proposed method consists of two students and two teachers that a pair of student-teacher network is the same as STPM. The other student-teacher network has a role to reconstruct the features of normal products. By reconstructing the features of normal products from an abnormal image, it is possible to detect abnormalities with higher accuracy by taking the difference between them. The new student-teacher network uses attention modules and different teacher network from the original STPM. Attention mechanism acts to successfully reconstruct the normal regions in an input image. Different teacher network prevents looking at the same regions as the original STPM. Six anomaly maps obtained from the two student-teacher networks are used to calculate the final anomaly map. Student-teacher network for reconstructing features improved AUC scores for pixel level and image level in comparison with the original STPM.
updated: Tue Nov 30 2021 13:24:10 GMT+0000 (UTC)
published: Tue Nov 30 2021 13:24:10 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アソシエイト