This study proposes a transfer learning (TL) pipeline for detecting surface defects in ball screw drives, which are vital in industries like robotics and aerospace. The pipeline uses pre-trained CNN models—InceptionV3, VGG16 and VGG19—to extract features from defect images, followed by classification with SVM (Support Vector Machine) and kNN (k-Nearest Neighbors). The InceptionV3 + SVM combination excels in accuracy, recall, precision, and F1 score, highlighting its effectiveness in defect detection. The study underscores the importance of selecting appropriate CNN architectures and classifiers for specific defect detection tasks. The dataset from the Karlsruhe Institute of Technology, consisting of 2000 images, is used to evaluate the TL pipeline's performance. The findings suggest that the InceptionV3 + SVM model offers a reliable method for identifying ball screw drive defects, with potential for further optimization through expanded datasets and hyperparameter tuning.

错误:搜索内容不能为空,请输入英文关键词
错误:关键词超出字数限制,请精简
高级检索

Ball Screw Drive Surface Defect Model Based on Transfer Learning Approach

  • Yifeng Xu,
  • Yang Luo,
  • Anwar P. P. Abdul Majeed,
  • Xiaoyan Liu,
  • Yuyi Zhu,
  • Wei Chen

摘要

This study proposes a transfer learning (TL) pipeline for detecting surface defects in ball screw drives, which are vital in industries like robotics and aerospace. The pipeline uses pre-trained CNN models—InceptionV3, VGG16 and VGG19—to extract features from defect images, followed by classification with SVM (Support Vector Machine) and kNN (k-Nearest Neighbors). The InceptionV3 + SVM combination excels in accuracy, recall, precision, and F1 score, highlighting its effectiveness in defect detection. The study underscores the importance of selecting appropriate CNN architectures and classifiers for specific defect detection tasks. The dataset from the Karlsruhe Institute of Technology, consisting of 2000 images, is used to evaluate the TL pipeline's performance. The findings suggest that the InceptionV3 + SVM model offers a reliable method for identifying ball screw drive defects, with potential for further optimization through expanded datasets and hyperparameter tuning.