计算机集成制造系统 ›› 2024, Vol. 30 ›› Issue (3): 1115-1126.DOI: 10.13196/j.cims.2023.0135

• • 上一篇    下一篇

近邻密度辅助模糊优化孪生支持向量机的钢板表面缺陷分类

侯政通1+,胡鹰1,乔磊明1,邓志飞2   

  1. 1.太原科技大学计算机科学与技术学院
    2.东北大学计算机科学与工程学院
  • 出版日期:2024-03-31 发布日期:2024-04-03
  • 基金资助:
    国家自然科学基金资助项目(52275357,52175354);山西重大专项课题项目(20181102016);山西省专利推广项目(20210524)。

Nearest neighbor density assisted fuzzy optimized TWSVM for surface defect classification of steel plates

HOU Zhengtong1+,HU Ying1,QIAO Leiming1,DENG Zhifei2   

  1. 1.School of Computer Science and Technology,Taiyuan University of Science and Technology
    2.School of Computer Science and Engineering,Northeastern University
  • Online:2024-03-31 Published:2024-04-03
  • Supported by:
    Project supported by the  National Natural Science Foundation,China(No.52275357,52175354),the Program of Major Special Projects of Shanxi Province,China(No.20181102016),and the Patent Promotion Project of Shanxi Province,China(No.20210524).

摘要: 为提升钢板表面缺陷分类精度,提出一种选择性弱化样本的分类模型。首先,在图像预处理阶段引入显著性检测算法来减少二值化后图像出现失真的影响;其次,为了降低不利的边缘样本点对模型的影响,同时又能提高有利的边缘样本点对模型的贡献,构造了一种新的密度模糊隶属度函数对样本进行权重赋值;最后,在孪生支持向量机(TWSVM)的基础上,将构造的密度模糊隶属度函数作为优化条件嵌入模型内,提出了近邻密度辅助模糊优化的TWSVM算法,以提高分类效果。在数据集NEU上的实验结果表明,引入显著性检测算法后,重新设计的特征在整体准确率上提高了1.66%,同时采用优化后的算法进行缺陷分类,准确率达到98.33%,进一步提高了分类性能。

关键词: 图像处理, 显著性检测, 缺陷分类, 孪生支持向量机, 密度函数, K近邻

Abstract: To enhance the classification accuracy of steel plate surface defects,a classification model with selectively weakened samples was proposed.The salient object detection algorithm was introduced in the image preprocessing step to reduce the effect of distortion in the image after binarization.To reduce the influence of unfavorable edge sample points on the model while increasing the contribution of favorable edge sample points to the model,a new density fuzzy membership function was constructed to assign weights to the samples.Based on Twin Support Vector Machine (TWSVM),the nearest neighbor density assisted fuzzy optimized TWSVM algorithm was proposed by embedding the constructed density fuzzy membership function within the model as an optimization condition to improve the classification effect.Experimental results on the dataset NEU showed that the redesigned features improved the overall accuracy by 1.66% after the introduction of the salient object detection algorithm,while the defect classification using the optimized algorithm achieved an accuracy of 98.33%,further improving the classification performance.

Key words: image processing, saliency detection, image processing, twin support vector machine, density function, K-nearest neighbor

中图分类号: