基于多任务学习的输电线路小金具缺失推理加速算法

作者:程绳; 葛雄; 肖非*; 朱传刚; 吴军; 肖海涛; 李嗣; 楚江平; 袁雨薇
来源:计算机测量与控制, 2023, 31(07): 251-257.
DOI:10.16526/j.cnki.11-4762/tp.2023.07.036

摘要

针对输电线路小金具缺失的检测问题,对小金具缺失算法的加速推理进行研究,采用多任务头的学习方法,将小金具缺失检测任务使用一个Swin Transformer网络结构[12]和多个由多层感知机组成的任务头组合的方式进行多任务学习推理,并进行单任务和多任务学习的推理精度和推理性能对比实验,最后还通过实验得到在多任务学习中插拔式扩展任务,实验结果表明在输电线路小金具缺失推理算法中多任务学习比单任务学习的推理性能提升了2倍多,同时显存占用降低了22%以上;通过插拔式扩展任务实验,验证了扩展任务的效果,可灵活扩展配置任务。

全文