Classification of steel based on laser-induced breakdown spectroscopy combined with restricted Boltzmann machine and support vector machine

Abstract

Abstract In recent years, a laser-induced breakdown spectrometer (LIBS) combined with machine learning has been widely developed for steel classification. However, the much redundant information of LIBS spectra increases the computation complexity for classification. In this work, restricted Boltzmann machines (RBM) and principal component analysis (PCA) were used for dimension reduction of datasets, respectively. Then, a support vector machine (SVM) was adopted to process feature information. Two models (RBM-SVM and PCA-SVM) are compared in terms of performance. After optimization, the accuracy of the RBM-SVM model can achieve 100%, and the maximum dimension reduction time is 33.18 s, which is nearly half of that of the PCA model (53.19 s). These results preliminarily indicate that LIBS combined with RBM-SVM has great potential in the real-time classification of steel.