A Classification Method for Electronic Components Based on Siamese Network

Yahui Cheng, Aimin Wang*, Long Wu

*Corresponding author for this work

Research output: Contribution to journalArticlepeer-review

7 Citations (Scopus)

Abstract

In the field of electronics manufacturing, electronic component classification facilitates the management and recycling of the functional and valuable electronic components in electronic waste. Current electronic component classification methods are mainly based on deep learning, which requires a large number of samples to train the model. Owing to the wide variety of electronic components, collecting datasets is a time-consuming and laborious process. This study proposed a Siamese network-based classification method to solve the electronic component classification problem for a few samples. First, an improved visual geometry group 16 (VGG-16) model was proposed as the feature extraction part of the Siamese neural network to improve the recognition performance of the model under small samples. Then, a novel channel correlation loss function that allows the model to learn the correlation between different channels in the feature map was designed to further improve the generalization performance of the model. Finally, the nearest neighbor algorithm was used to complete the classification work. The experimental results show that the proposed method can achieve high classification accuracy under small sample conditions and is robust for electronic components with similar appearances. This improves the classification quality of electronic components and reduces the training sample collection cost.

Original languageEnglish
Article number6478
JournalSensors
Volume22
Issue number17
DOIs
Publication statusPublished - Sept 2022

Keywords

  • Siamese network
  • channel correlation loss
  • electronic components classification
  • few-shot learning
  • vgg

Fingerprint

Dive into the research topics of 'A Classification Method for Electronic Components Based on Siamese Network'. Together they form a unique fingerprint.

Cite this