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

基于不确定性的深度学习铸造缺陷检测框架

An Uncertainty-Aware Deep Learning Framework for Defect Detection in Casting Products

Lian Sazeh group(联萨泽集团);University of Tehran(德黑兰大学);Deakin University(迪肯大学);Institute for Intelligent Systems Research and Innovation(智能系统研究与创新研究所)

Maryam Habibpour, Hassan Gharoun, AmirReza Tajally, Afshar Shamsi, Hamzeh Asgharnezhad, Abbas Khosravi, Saeid Nahavandi

摘要:

由于铸造工艺的复杂性,在铸造生产中不可避免地会产生缺陷。虽然传统的铸造产品人工视觉检测在大规模生产中速度缓慢且效率低下,但自动可靠的缺陷检测不仅可以增强质量控制过程,还可以积极提高生产率。然而,由于铸件缺陷外观的多样性和变化性,铸件缺陷检测是一项具有挑战性的任务。卷积神经网络(CNN)已广泛应用于图像分类和缺陷检测任务。然而,具有频率论推理的CNN需要大量的数据来进行训练,并且仍然无法报告其预测不确定性的有益估计。因此,利用迁移学习范式,我们首先在一个小数据集上应用四个强大的基于CNN的模型(VGG16,ResNet50,DenseNet121和InceptionResNetV2)来提取有意义的特征。提取的特征然后由各种机器学习算法处理以执行分类任务。仿真结果表明,线性支持向量机(SVM)和多层感知器(MLP)在铸件图像的缺陷检测中表现出最好的性能。其次,为了实现可靠的分类和测量认知不确定性,我们采用了不确定性量化(UQ)技术(MLP模型集成),使用从四个预先训练的CNN中提取的特征。UQ混淆矩阵和不确定性精度度量也被用来评估预测不确定性估计。综合比较表明,基于VGG16的UQ方法在提取不确定度方面优于其他方法。我们相信,具有不确定性意识的自动缺陷检测解决方案将加强铸件生产的质量保证。

英文摘要:

Defects are unavoidable in casting production owing to the complexity of the casting process. While conventional human-visual inspection of casting products is slow and unproductive in mass productions, an automatic and reliable defect detection not just enhances the quality control process but positively improves productivity. However, casting defect detection is a challenging task due to diversity and variation in defects' appearance. Convolutional neural networks (CNNs) have been widely applied in both image classification and defect detection tasks. Howbeit, CNNs with frequentist inference require a massive amount of data to train on and still fall short in reporting beneficial estimates of their predictive uncertainty. Accordingly, leveraging the transfer learning paradigm, we first apply four powerful CNN-based models (VGG16, ResNet50, DenseNet121, and InceptionResNetV2) on a small dataset to extract meaningful features. Extracted features are then processed by various machine learning algorithms to perform the classification task. Simulation results demonstrate that linear support vector machine (SVM) and multi-layer perceptron (MLP) show the finest performance in defect detection of casting images. Secondly, to achieve a reliable classification and to measure epistemic uncertainty, we employ an uncertainty quantification (UQ) technique (ensemble of MLP models) using features extracted from four pre-trained CNNs. UQ confusion matrix and uncertainty accuracy metric are also utilized to evaluate the predictive uncertainty estimates. Comprehensive comparisons reveal that UQ method based on VGG16 outperforms others to fetch uncertainty. We believe an uncertainty-aware automatic defect detection solution will reinforce casting productions quality assurance.

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

↑