Back to Search
Start Over
Research on a lightweight electronic component detection method based on knowledge distillation
- Source :
- Mathematical Biosciences and Engineering, Vol 20, Iss 12, Pp 20971-20994 (2023)
- Publication Year :
- 2023
- Publisher :
- AIMS Press, 2023.
-
Abstract
- As an essential part of electronic component assembly, it is crucial to rapidly and accurately detect electronic components. Therefore, a lightweight electronic component detection method based on knowledge distillation is proposed in this study. First, a lightweight student model was constructed. Then, we consider issues like the teacher and student's differing expressions. A knowledge distillation method based on the combination of feature and channel is proposed to learn the teacher's rich class-related and inter-class difference features. Finally, comparative experiments were analyzed for the dataset. The results show that the student model Params (13.32 M) are reduced by 55%, and FLOPs (28.7 GMac) are reduced by 35% compared to the teacher model. The knowledge distillation method based on the combination of feature and channel improves the student model's mAP by 3.91% and 1.13% on the Pascal VOC and electronic components detection datasets, respectively. As a result of the knowledge distillation, the constructed student model strikes a superior balance between model precision and complexity, allowing for fast and accurate detection of electronic components with a detection precision (mAP) of 97.81% and a speed of 79 FPS.
Details
- Language :
- English
- ISSN :
- 15510018
- Volume :
- 20
- Issue :
- 12
- Database :
- Directory of Open Access Journals
- Journal :
- Mathematical Biosciences and Engineering
- Publication Type :
- Academic Journal
- Accession number :
- edsdoj.64fbea5db1144f499062736a3bd25673
- Document Type :
- article
- Full Text :
- https://doi.org/10.3934/mbe.2023928?viewType=HTML