基于动态监督知识蒸馏的输电线路螺栓缺陷图像分类 |
| |
作者单位: | 华北电力大学电气与电子工程学院,保定071003 |
| |
基金项目: | 国家自然科学基金;国家自然科学基金;北京市自然科学基金;河北省自然科学基金;中央高校基本科研业务费专项;中央高校基本科研业务费专项;模式识别国家重点实验室开放基金 |
| |
摘 要: | 螺栓是输电线路中广泛存在的紧固件,其缺陷图像具有类内差异性小、类间差异性大的特性。针对复杂度高且性能优秀的大模型在分析螺栓缺陷图像消耗大量计算资源的问题,将知识蒸馏技术引入到输电线路螺栓缺陷图像分类中,提出了一种基于动态监督知识蒸馏的输电线路螺栓缺陷图像分类方法:在网络输出层采用自适应加权方法,提高小模型学习螺栓缺陷标签的准确性;在网络隐藏层进行注意力转移,提高小模型螺栓特征的表达能力;将网络输出层的自适应加权方法与网络隐藏层的注意力转移机制相结合,以充分提高小模型的螺栓缺陷分类能力。最后通过自建螺栓缺陷图像分类数据集验证了大模型利用所提蒸馏方法指导小模型训练的有效性,实验结果表明:小模型的分类准确率提高了2.17%,小模型与大模型的分类准确率只差0.63%,且小模型的参数量仅为大模型参数量的7.8%。研究实现了螺栓缺陷的高效分类,达到了精度与资源消耗的平衡。
|
关 键 词: | 螺栓缺陷分类 大模型 小模型 知识蒸馏 自适应加权 注意力转移 |
本文献已被 CNKI 万方数据 等数据库收录! |
|