arXiv reaDer
鋳造製品の欠陥検出のための不確実性を意識した深層学習フレームワーク
An Uncertainty-Aware Deep Learning Framework for Defect Detection in Casting Products
鋳造工程が複雑なため、鋳造生産において欠陥は避けられません。鋳造製品の従来の人間による目視検査は、大量生産では遅く、非生産的ですが、自動で信頼性の高い欠陥検出は、品質管理プロセスを強化するだけでなく、生産性を積極的に向上させます。ただし、鋳造欠陥の検出は、欠陥の外観の多様性と多様性のために困難な作業です。畳み込みニューラルネットワーク(CNN)は、画像分類と欠陥検出タスクの両方に広く適用されています。ただし、頻度論的推定を使用するCNNは、トレーニングに大量のデータを必要としますが、予測の不確実性の有益な推定値を報告するにはまだ不十分です。したがって、転移学習パラダイムを活用して、最初に4つの強力なCNNベースのモデル(VGG16、ResNet50、DenseNet121、およびInceptionResNetV2)を小さなデータセットに適用して、意味のある特徴を抽出します。抽出された特徴は、分類タスクを実行するためにさまざまな機械学習アルゴリズムによって処理されます。シミュレーション結果は、線形サポートベクターマシン(SVM)と多層パーセプトロン(MLP)が鋳造画像の欠陥検出で最高のパフォーマンスを示すことを示しています。次に、信頼性の高い分類を実現し、認識論的不確実性を測定するために、事前にトレーニングされた4つのCNNから抽出された特徴を使用して、不確実性定量化(UQ)手法(MLPモデルのアンサンブル)を採用します。 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.
updated: Sat Jul 24 2021 16:17:20 GMT+0000 (UTC)
published: Sat Jul 24 2021 16:17:20 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アソシエイト