arXiv reaDer
TL-SDD:少数のサンプルで表面欠陥を検出するための転移学習ベースの方法
TL-SDD: A Transfer Learning-Based Method for Surface Defect Detection with Few Samples
表面欠陥の検出は、製品の品質を保証するために製造業でますます重要な役割を果たしています。多くの深層学習手法は、表面の欠陥検出タスクで広く使用されており、欠陥の分類と場所でうまく機能することが証明されています。ただし、深層学習ベースの検出方法では、トレーニングに大量のデータが必要になることが多く、欠陥カテゴリの分布が不均衡であることが多いため、実際の産業シナリオには適用できません。言い換えれば、一般的な欠陥クラスには多くのサンプルがありますが、まれな欠陥クラスには非常に少ないサンプルがあり、これらのメソッドがまれな欠陥クラスを適切に検出することは困難です。不均衡な分布の問題を解決するために、この論文では、TL-SDDを提案します。これは、表面欠陥検出のための新しい伝達学習ベースの方法です。まず、一般的な欠陥クラスからまれな欠陥クラスに知識を転送するために、2フェーズのトレーニングスキームを採用します。次に、新しいメトリックベースの表面欠陥検出(M-SDD)モデルを提案します。このモデル用に3つのモジュールを設計します。(1)特徴抽出モジュール:高レベルのセマンティック情報と低レベルの構造情報を組み合わせた特徴融合を含みます。 (2)機能の再重み付けモジュール:例を機能の重要性を示す再重み付けベクトルに変換します。 (3)距離計量モジュール:各カテゴリーの表現までの距離を計算することによって欠陥が分類される距離空間を学習します。最後に、アルミニウムプロファイルの表面欠陥を含む実際のデータセットで提案された方法のパフォーマンスを検証します。ベースラインメソッドと比較して、提案されたメソッドのパフォーマンスは、まれな欠陥クラスで最大11.98%向上しています。
Surface defect detection plays an increasingly important role in manufacturing industry to guarantee the product quality. Many deep learning methods have been widely used in surface defect detection tasks, and have been proven to perform well in defects classification and location. However, deep learning-based detection methods often require plenty of data for training, which fail to apply to the real industrial scenarios since the distribution of defect categories is often imbalanced. In other words, common defect classes have many samples but rare defect classes have extremely few samples, and it is difficult for these methods to well detect rare defect classes. To solve the imbalanced distribution problem, in this paper we propose TL-SDD: a novel Transfer Learning-based method for Surface Defect Detection. First, we adopt a two-phase training scheme to transfer the knowledge from common defect classes to rare defect classes. Second, we propose a novel Metric-based Surface Defect Detection (M-SDD) model. We design three modules for this model: (1) feature extraction module: containing feature fusion which combines high-level semantic information with low-level structural information. (2) feature reweighting module: transforming examples to a reweighting vector that indicates the importance of features. (3) distance metric module: learning a metric space in which defects are classified by computing distances to representations of each category. Finally, we validate the performance of our proposed method on a real dataset including surface defects of aluminum profiles. Compared to the baseline methods, the performance of our proposed method has improved by up to 11.98% for rare defect classes.
updated: Mon Aug 16 2021 07:24:00 GMT+0000 (UTC)
published: Mon Aug 16 2021 07:24:00 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アソシエイト