A weighted fuzzy C-means clustering method for hardness prediction  

作　　者：Yuan Liu Shi-zhong Wei 

机构地区：[1]School of Materials Science and Engineering,Henan University of Science and Technology,Luoyang 471003,Henan,China [2]School of Aerospace Engineering,Zhengzhou University of Aeronautics,Zhengzhou 450015,Henan,China [3]National Joint Engineering Research Center for Abrasion Control and Molding of Metal Materials,Henan University of Science and Technology,Luoyang 471003,Henan,China

出　　处：《Journal of Iron and Steel Research International》2023年第1期176-191,共16页

基　　金：supported by the National Research and Development Project of China (2020YFB2008400).

摘　　要：The hardness prediction model was established by support vector regression(SVR).In order to avoid exaggerating the contribution of very tiny alloying elements,a weighted fuzzy C-means(WFCM)algorithm was proposed for data clustering using improved Mahalanobis distance based on random forest importance values,which could play a full role of important features and avoid clustering center overlap.The samples were divided into two classes.The top 10 features of each class were selected to form two feature subsets for better performance of the model.The dimension and dispersion of features decreased in such feature subsets.Comparing four machine learning algorithms,SVR had the best performance and was chosen to modeling.The hyper-parameters of the SVR model were optimized by particle swarm optimization.The samples in validation set were classified according to minimum distance of sample to clustering centers,and then the SVR model trained by feature subset of corresponding class was used for prediction.Compared with the feature subset of original data set,the predicted values of model trained by feature subsets of classified samples by WFCM had higher correlation coefficient and lower root mean square error.It indicated that WFCM was an effective method to reduce the dispersion of features and improve the accuracy of model.

关 键 词：Hardness prediction Weighted fuzzy C-means algorithm Feature selection Particle swarm optimization Support vector regression Dispersion reduction 

分 类 号：TP311.13[自动化与计算机技术—计算机软件与理论]