关键词:
智能巡检
深度学习
鲁棒性
对比学习
训练方法
摘要:
输电线路的智能巡检视觉任务对电力系统的安全稳定至关重要。尽管深度学习网络在分布一致的训练和测试数据集上表现良好,但实际应用中数据分布的偏差常常会降低模型性能。为解决这一问题,提出了一种基于对比学习的训练方法(Training Method based on Contrastive Learning,TMCL),旨在增强模型鲁棒性。首先,构建了专为输电线路场景设计的基准测试集(Transmission Line Dataset-Corruption,TLD-C),用于评估模型在面对图像损坏时的鲁棒性。然后,TMCL通过构建对类别特征敏感的正负样本对,提升了模型对不同类别特征的区分能力。接着,结合对比损失和交叉熵损失的联合优化策略,对特征提取过程施加额外约束,以优化特征向量的表征。最后,引入了非局部特征去噪网络,专注于提取与类别密切相关的特征。实验结果表明,模型应用改进后的训练方法,在输电线路数据集上的平均精度比原始方法高出3.40个百分点,在TLD-C数据集上的相对损坏精度比原始方法高出4.69个百分点。