arXiv reaDer
表面欠陥検出のための混合教師あり:弱教師あり学習から完全教師あり学習まで
Mixed supervision for surface-defect detection: from weakly to fully supervised learning
深層学習法は、最近、産業品質管理における表面欠陥検出の問題に対処するために採用され始めています。ただし、学習に大量のデータが必要であり、高精度のラベルが必要になることが多いため、多くの産業上の問題を簡単に解決できないか、注釈要件のためにソリューションのコストが大幅に増加します。この作業では、完全に教師あり学習方法の重い要件を緩和し、非常に詳細な注釈の必要性を減らします。ディープラーニングアーキテクチャを提案することにより、表面欠陥検出のタスクで、弱い(画像レベル)ラベルから混合監視、完全(ピクセルレベル)注釈まで、さまざまな詳細の注釈の使用を検討します。提案されたエンドツーエンドアーキテクチャは、欠陥のセグメンテーションと分類の結果をもたらす2つのサブネットワークで構成されています。提案された方法は、産業品質検査のためのいくつかのデータセットで評価されます:KolektorSDD、DAGMおよびSeverstal SteelDefect。また、実際の産業問題に対処しながら取得した、いくつかのタイプの欠陥を含む3000を超える画像を含むKolektorSDD2と呼ばれる新しいデータセットを紹介します。 4つのデータセットすべてで最先端の結果を示します。提案された方法は、完全に監視された設定で関連するすべてのアプローチよりも優れており、画像レベルのラベルのみが使用可能な場合は、弱く監視された方法よりも優れています。また、弱くラベル付けされたトレーニング画像に追加されたほんの一握りの完全に注釈付きのサンプルを使用した混合教師ありは、完全に教師ありモデルのパフォーマンスに匹敵するパフォーマンスをもたらすことができますが、注釈コストは大幅に低くなります。
Deep-learning methods have recently started being employed for addressing surface-defect detection problems in industrial quality control. However, with a large amount of data needed for learning, often requiring high-precision labels, many industrial problems cannot be easily solved, or the cost of the solutions would significantly increase due to the annotation requirements. In this work, we relax heavy requirements of fully supervised learning methods and reduce the need for highly detailed annotations. By proposing a deep-learning architecture, we explore the use of annotations of different details ranging from weak (image-level) labels through mixed supervision to full (pixel-level) annotations on the task of surface-defect detection. The proposed end-to-end architecture is composed of two sub-networks yielding defect segmentation and classification results. The proposed method is evaluated on several datasets for industrial quality inspection: KolektorSDD, DAGM and Severstal Steel Defect. We also present a new dataset termed KolektorSDD2 with over 3000 images containing several types of defects, obtained while addressing a real-world industrial problem. We demonstrate state-of-the-art results on all four datasets. The proposed method outperforms all related approaches in fully supervised settings and also outperforms weakly-supervised methods when only image-level labels are available. We also show that mixed supervision with only a handful of fully annotated samples added to weakly labelled training images can result in performance comparable to the fully supervised model's performance but at a significantly lower annotation cost.
updated: Mon Apr 19 2021 08:44:32 GMT+0000 (UTC)
published: Tue Apr 13 2021 10:00: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アソシエイト