arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2302.09565cs.CV

优化用于半导体缺陷检测的 YOLOv7

Optimizing YOLOv7 for Semiconductor Defect Detection

KU Leuven(荷语鲁汶大学);Interuniversity Microelectronics Centre (imec)(校际微电子中心(imec))

Enrique Dehaerne, Bappaditya Dey, Sandip Halder, Stefan De Gendt

AI总结:

本文通过超参数调优、数据增强与 WBF 模型集成优化 YOLOv7,用于半导体线距图案缺陷检测,最佳集成较默认模型 mAP 提升 10%。

摘要:

使用深度学习(DL)进行目标检测的领域正在不断发展,许多新技术和模型被提出。YOLOv7 是基于 YOLO 系列模型的最先进目标检测器,该系列模型已在工业应用中流行起来。一个可能的应用领域是半导体缺陷检测。任何机器学习模型的性能都取决于其超参数。此外,以不同方式组合一个或多个模型的预测也会影响性能。在本研究中,我们对最近提出的最先进目标检测器 YOLOv7 进行实验,通过使用不同超参数训练和评估模型,研究哪些超参数能够提高半导体线距图案缺陷的检测精度。使用默认超参数和非极大值抑制(NMS)预测组合的基础 YOLOv7 模型,在平均精度均值(mAP)方面优于以往工作中的所有 RetinaNet 模型。我们发现,在训练期间随机垂直翻转图像可使所有缺陷类别的平均 AP 提升 3%。与默认模型相比,其他超参数值仅改善了某些类别的 AP。研究发现,将在不同缺陷类别上取得最佳 AP 的模型组合起来是一种有效的集成策略。使用加权框融合(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%.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑