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 条
  • [21] Surface Defect Detection Method for Industrial Products Based on Photometric Stereo and Dual Stream Feature Fusion Network
    Hu, Guanghua
    Tu, Qianxi
    Huanan Ligong Daxue Xuebao/Journal of South China University of Technology (Natural Science), 2024, 52 (10): : 112 - 123
  • [22] Development of a lightweight cross-scale decoupling feature fusion network for surface defect detection in permanent magnets
    Shu, Shuangbao
    Gao, Xinyu
    Zheng, Changjie
    Fu, Yufeng
    Wang, Jiyao
    MEASUREMENT SCIENCE AND TECHNOLOGY, 2025, 36 (03)
  • [23] Steel strip surface defect detection based on multiscale feature sensing and adaptive feature fusion
    Mi, Zengzhen
    Gao, Yan
    Xu, Xingyuan
    Tang, Jing
    AIP ADVANCES, 2024, 14 (04)
  • [24] A Transformer-Based Network With Feature Complementary Fusion for Crack Defect Detection
    Ma, Mingyang
    Yang, Lei
    Liu, Yanhong
    Yu, Hongnian
    IEEE TRANSACTIONS ON INTELLIGENT TRANSPORTATION SYSTEMS, 2024, 25 (11) : 16989 - 17006
  • [25] A Lightweight Spatial and Temporal Multi-Feature Fusion Network for Defect Detection
    Hu, Bozhen
    Gao, Bin
    Woo, Wai Lok
    Ruan, Lingfeng
    Jin, Jikun
    Yang, Yang
    Yu, Yongjie
    IEEE Transactions on Image Processing, 2021, 30 : 472 - 486
  • [26] A Lightweight Spatial and Temporal Multi-Feature Fusion Network for Defect Detection
    Hu, Bozhen
    Gao, Bin
    Woo, Wai Lok
    Ruan, Lingfeng
    Jin, Jikun
    Yang, Yang
    Yu, Yongjie
    IEEE TRANSACTIONS ON IMAGE PROCESSING, 2021, 30 : 472 - 486
  • [27] Micro-defect Varifocal Network: Channel attention and spatial feature fusion for turbine blade surface micro-defect detection
    Liu, Peng
    Yuan, Xiaolong
    Han, Qiang
    Xing, Baowen
    Hu, Xiaolian
    Zhang, Jianhai
    ENGINEERING APPLICATIONS OF ARTIFICIAL INTELLIGENCE, 2024, 133
  • [28] EFAFN: An Efficient Feature Adaptive Fusion Network with Facial Feature for Multimodal Sarcasm Detection
    Sun, Yukuan
    Zhang, Hangming
    Yang, Shengjiao
    Wang, Jianming
    APPLIED SCIENCES-BASEL, 2022, 12 (21):
  • [29] A Method for Surface Defect Detection Based on Multiscale Feature Fusion and Pyramid Attention
    Tang, Ying
    Wang, Hongyuan
    Zhou, Qunying
    Sun, Boyan
    IEEE ACCESS, 2024, 12 : 36457 - 36465
  • [30] Surface Defect Detection Based on Adaptive Multi-Scale Feature Fusion
    Wen, Guochen
    Cheng, Li
    Yuan, Haiwen
    Li, Xuan
    SENSORS, 2025, 25 (06)