arXiv reaDer
半導体欠陥検出のためのYOLOv7の最適化
Optimizing YOLOv7 for Semiconductor Defect Detection
ディープ ラーニング (DL) を使用したオブジェクト検出の分野は、多くの新しい手法やモデルが提案され、常に進化しています。 YOLOv7 は、産業用アプリケーションで一般的になったモデルの YOLO ファミリーに基づく最先端のオブジェクト検出器です。そのような可能性のある応用分野の 1 つとして、半導体の欠陥検査があります。機械学習モデルのパフォーマンスは、そのハイパーパラメーターに依存します。さらに、1 つ以上のモデルの予測をさまざまな方法で組み合わせることも、パフォーマンスに影響を与える可能性があります。この研究では、最近提案された最先端のオブジェクト検出器である YOLOv7 を使用して、さまざまなハイパーパラメーターを使用してモデルをトレーニングおよび評価し、半導体ライン スペース パターン欠陥の検出精度に関してどのモデルがパフォーマンスを向上させるかを調査します。デフォルトのハイパーパラメーターと非最大抑制 (NMS) 予測を組み合わせたベース YOLOv7 モデルは、平均平均精度 (mAP) に関して、以前の研究のすべての RetinaNet モデルよりも優れています。トレーニング中に画像をランダムに上下反転させると、すべての欠陥クラスの平均 AP が 3% 改善されることがわかりました。他のハイパーパラメータ値は、デフォルト モデルと比較して特定のクラスでのみ AP を改善しました。さまざまな欠陥クラスに対して最適な AP を達成するモデルを組み合わせることは、効果的なアンサンブル戦略であることがわかりました。 Weighted Box Fusion (WBF) 予測を使用してアンサンブルからの予測を組み合わせると、最高のパフォーマンスが得られました。 WBF を使用した最適なアンサンブルは、デフォルト モデルの mAP を 10% 改善しました。
The field of object detection using Deep Learning (DL) is constantly evolving with many new techniques and models being proposed. YOLOv7 is a state-of-the-art object detector based on the YOLO family of models which have become popular for industrial applications. One such possible application domain can be semiconductor defect inspection. The performance of any machine learning model depends on its hyperparameters. Furthermore, combining predictions of one or more models in different ways can also affect performance. In this research, we experiment with YOLOv7, a recently proposed, state-of-the-art object detector, by training and evaluating models with different hyperparameters to investigate which ones improve performance in terms of detection precision for semiconductor line space pattern defects. The base YOLOv7 model with default hyperparameters and Non Maximum Suppression (NMS) prediction combining outperforms all RetinaNet models from previous work in terms of mean Average Precision (mAP). We find that vertically flipping images randomly during training yields a 3% improvement in the mean AP of all defect classes. Other hyperparameter values improved AP only for certain classes compared to the default model. Combining models that achieve the best AP for different defect classes was found to be an effective ensembling strategy. Combining predictions from ensembles using Weighted Box Fusion (WBF) prediction gave the best performance. The best ensemble with WBF improved on the mAP of the default model by 10%.
updated: Sun Feb 19 2023 12:51:07 GMT+0000 (UTC)
published: Sun Feb 19 2023 12:51:07 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アソシエイト