Explainable efficient and optimized feature fusion network for surface defect detection

被引:9
|
作者
Sundarrajan, Kavitha [1 ]
Rajendran, Baskaran Kuttuva [2 ]
机构
[1] Kumaraguru Coll Technol, Dept Informat Technol, Coimbatore 641049, India
[2] Kumaraguru Coll Technol, Dept Comp Sci & Engn, Coimbatore 641049, India
关键词
Hot-rolled strip steel; Transfer learning; Deep learning model; Feature fusion network (FFN); Vgg16; Inceptionv3; Resnet50; Feature extraction; Image classification; Explainable artificial intelligence (XAI); Particle swarm optimization algorithm; ROLLED STEEL STRIPS; CLASSIFICATION;
D O I
10.1007/s00170-023-11789-0
中图分类号
TP [自动化技术、计算机技术];
学科分类号
0812 ;
摘要
The quality of the surface and plate form of hot-rolled strip steel, a crucial raw material to produce automobiles, household appliances, and other goods greatly influences the final products that end users make. The identification of surface flaws is crucial to the manufacture of steel strips. Furthermore, typical fault identification techniques have issue of poor detecting reliability, and lower accuracy is obtained by the explainable single pre-trained networks which led to the development of the feature fusion network (FFN). The major objective of the work is to design a traditional deep network model is enhanced by the application of a transfer learning model to detect surface flaws in steel strips. The use of pre-trained models reduces negative effects by drastically reducing training time and improving the accuracy of image classification. Transfer learning models such as VGG16, InceptionV3, and ResNet50 are used to train the Northeastern University-DETection (NEU-DET) Dataset which significantly reduces the time for the training. Generative adversarial network is used for data augmentation to increase the input images. An explainable artificial intelligence (XAI) classifier is applied to the pre-trained networks to understand the classification of the surface defects. A hybrid FFN (HFFN) is proposed which combines the features of pre-trained networks (VGG16, InceptionV3, and ResNet50) to accurately classify flaws in the hot-rolled strips surface. To reduce the features in the HFFN, particle swarm optimization (PSO) algorithm (PFFN) is used. On the NEU-DET, FFN by three-pre-trained model achieves 98.65%, 98.42%, 98.51%, and 98.54% for precision, recall, f-score, and accuracy respectively.
引用
收藏
页数:18
相关论文
共 50 条
  • [1] Explainable efficient and optimized feature fusion network for surface defect detection
    Sundarrajan, Kavitha
    Rajendran, Baskaran Kuttuva
    International Journal of Advanced Manufacturing Technology, 2023,
  • [2] DFFNet: a lightweight approach for efficient feature-optimized fusion in steel strip surface defect detection
    Hu, Xianming
    Lin, Shouying
    COMPLEX & INTELLIGENT SYSTEMS, 2024, 10 (05) : 6705 - 6723
  • [3] Attention-Based Multiscale Feature Fusion for Efficient Surface Defect Detection
    Zhao, Yuhao
    Liu, Qing
    Su, Hu
    Zhang, Jiabin
    Ma, Hongxuan
    Zou, Wei
    Liu, Song
    IEEE TRANSACTIONS ON INSTRUMENTATION AND MEASUREMENT, 2024, 73 : 1 - 10
  • [4] Small object detection method with shallow feature fusion network for chip surface defect detection
    Haixin Huang
    Xueduo Tang
    Feng Wen
    Xin Jin
    Scientific Reports, 12
  • [5] Small object detection method with shallow feature fusion network for chip surface defect detection
    Huang, Haixin
    Tang, Xueduo
    Wen, Feng
    Jin, Xin
    SCIENTIFIC REPORTS, 2022, 12 (01)
  • [6] An Efficient Network for Surface Defect Detection
    Lin, Zesheng
    Ye, Hongxia
    Zhan, Bin
    Huang, Xiaofeng
    APPLIED SCIENCES-BASEL, 2020, 10 (17):
  • [7] CSANet: Contour and Semantic Feature Alignment Fusion Network for Rail Surface Defect Detection
    Yang, Jinxin
    Zhou, Wujie
    Wu, Ruiming
    Fang, Meixin
    IEEE SIGNAL PROCESSING LETTERS, 2023, 30 : 972 - 976
  • [8] MFNet: A Novel Multilevel Feature Fusion Network With Multibranch Structure for Surface Defect Detection
    Zhu, Jiangping
    He, Guohuan
    Zhou, Pei
    IEEE TRANSACTIONS ON INSTRUMENTATION AND MEASUREMENT, 2023, 72
  • [9] Fabric defect detection based on feature fusion of a convolutional neural network and optimized extreme learning machine
    Zhou, Zhiyu
    Deng, Wenxiong
    Zhu, Zefei
    Wang, Yaming
    Du, Jiayou
    Liu, Xiangqi
    TEXTILE RESEARCH JOURNAL, 2022, 92 (7-8) : 1161 - 1182
  • [10] An efficient multi-scale feature enhancement network for industrial surface defect detection
    Chen, Jiusheng
    Zha, Haoxiang
    Zhang, Xiaoyu
    Guo, Runxia
    Wu, Jun
    MEASUREMENT SCIENCE AND TECHNOLOGY, 2025, 36 (03)