关键词: 图像分类, 电力巡检, Fine-tuning训练策略, 余弦相似度, 小样本学习, 绝缘子缺陷

Abstract: To address the issue of low defect recognition accuracy due to the scarcity of insulator defect samples in power inspection, we propose a recognition method that combines a Fine-tuning training strategy with a Cosine Similarity Softmax (CSM) classifier. This method applies few-shot learning based on metric learning to the task of insulator defect classification. The approach consists of two main steps: first, pre-training the neural network on a large dataset; and second, using the Fine-tuning training strategy and CSM classifier to optimize the model, integrating meta-learning and metric learning to transfer the relevant knowledge acquired during the pre-training and meta-learning phases to the domain of insulator defect image classification. Ablation experiments demonstrate that incorporating the Fine-tuning training strategy increases recall by more than 0.5% and precision by more than 0.53%. Adding the CSM further improves recall by more than 0.38% and precision by more than 0.4%. Compared to other mainstream few-shot learning methods, our method shows improvements in recall and precision by over 0.51% and 0.47%, respectively. Experimental results indicate that the proposed method exhibits high performance in terms of accuracy, robustness, and generalization capability, confirming its effectiveness in the task of insulator defect image classification for power inspection.

Key words: image classification, power inspection, Fine-tuning training strategy, cosine similarity, few-shot learning, insulator defect