eduzhai > Applied Sciences > Engineering >

Unsupervised machine learning via transfer learning and k-means clustering to classify materials image data

  • king
  • (0) Download
  • 20210506
  • Save

... pages left unread,continue reading

Document pages: 19 pages

Abstract: Unsupervised machine learning offers significant opportunities for extractingknowledge from unlabeled data sets and for achieving maximum machine learningperformance. This paper demonstrates how to construct, use, and evaluate a highperformance unsupervised machine learning system for classifying images in apopular microstructural dataset. The Northeastern University Steel SurfaceDefects Database includes micrographs of six different defects observed onhot-rolled steel in a format that is convenient for training and evaluatingmodels for image classification. We use the VGG16 convolutional neural networkpre-trained on the ImageNet dataset of natural images to extract featurerepresentations for each micrograph. After applying principal componentanalysis to extract signal from the feature descriptors, we use k-meansclustering to classify the images without needing labeled training data. Theapproach achieves $99.4 pm 0.16 $ accuracy, and the resulting model can beused to classify new images without retraining This approach demonstrates animprovement in both performance and utility compared to a previous study. Asensitivity analysis is conducted to better understand the influence of eachstep on the classification performance. The results provide insight towardapplying unsupervised machine learning techniques to problems of interest inmaterials science.

Please select stars to rate!

         

0 comments Sign in to leave a comment.

    Data loading, please wait...
×