Computer Science ›› 2025, Vol. 52 ›› Issue (6A): 240800044-8.doi: 10.11896/jsjkx.240800044

• Image Processing & Multimedia Technology • Previous Articles     Next Articles

Transmission Line Fault Identification Method Based on Transfer Learning and Improved YOLOv8s

HUANG Bocheng1, WANG Xiaolong2, AN Guocheng2, ZHANG Tao1   

  1. 1 Shanghai Key Laboratory of Beidou Navigation and Location Services,College of Electronic Information and Electrical Engineering,Shanghai Jiao Tong University,Shanghai 200240,China
    2 Artificial Intelligence Research Institute of Shanghai Huaxun Network System Co.,Ltd.,Chengdu 610074,China
  • Online:2025-06-16 Published:2025-06-12
  • About author:HUANG Bocheng,born in 2000,postgraduate. His main research interests include target detection and deep lear-ning.
    WANG Xiaolong,born in 1980,master,senior engineer. His main research interests include intelligent information processing and data analysis.
  • Supported by:
    National Key R&D Program of China(2023YFC3006700).

Abstract: At present,there are serious problems when identifying some fault categories in transmission lines,such as insufficient samples and difficulty in locating small targets at long distances captured by drones,resulting in low accuracy in fault identification of transmission lines.To address the above issues,a novel transmission line fault identification method based on transfer learning and improved YOLOv8s is proposed.Firstly,taking YOLOv8s as the baseline,the transfer learning method is used to improve the recognition performance in few-shot,and a bidirectional correlation sample selection module is proposed to obtain sample categories with strong correlation with the target domain,avoiding the problem of negative transfer that may occur when using transfer learning,effectively improving the fault recognition performance of the model.Secondly,aiming at the difficult problem of small target localization,after fusing the 80*80 output feature map and the shallow feature map,EMA is introduced into multi-scale attention to enhance the feature information of the small target for designing small target attention detection layer.To improve the loss function,the CIoU loss is replaced by NWD loss in the prediction frame regression loss,which solves the problem that IoU is sensitive to small target position deviation.Specifically,the Wasserstein distance is used to measure the similarity between prediction box and the truth box.Experimental results show that in the case of few shot and small targets,the proposed method has a mAP of 51.1% in the transmission line fault dataset,which is 8.2% higher than that of the YOLOv8s baseline model,effectively improving the accuracy of fault recognition and providing new solution and method for few-shot and small target transmission line fault identification.

Key words: Fault identification, YOLOv8s, Transfer learning, Sample selection, Attention mechanism, Loss function

CLC Number: 

  • TP391
[1]SHI Y D,WANG H M,JING C,et al.A Few-Shot Defect Detection Method for Transmission Lines Based on Meta-Attention and Feature Reconstruction[J].Applied Sciences,2023,13:5896.
[2]WANG Z Y,GAO Q,LI D,et al.Insulator Anomaly DetectionMethod Based on Few-Shot Learning[J].IEEE Access,2021,9:94970-94980.
[3]LIU C Y,WU Y Q.Research Progress of Vision DetectionMethods Based on Deep Learning for Transmission Lines[C]//Proceedings of the CSEE.2023:7423-7445.
[4]TAO X,ZHANG D P,WANG Z H,et al.Detection of Power Line Insulator Defects Using Aerial Images Analyzed With Convolutional Neural Networks[J].IEEE Transactions on Systems,Man,and Cybernetics:Systems,2020,50(4):1486-1498.
[5]ZHOU M,LI B,WANG J,et al.Fault Detection Method ofGlass Insulator Aerial Image Based on the Improved YOLOv5[J].IEEE Transactions on Instrumentation and Measurement,2023,72:1-10.
[6]HAO S,AN B Y,MA X,et al.PKAMNet:A Transmission Line Insulator Parallel- Gap Fault Detection Network Based on Prior Knowledge Transfer and Attention Mechanism[J].IEEE Transactions on Power Delivery,2023,38(5):3387-3397.
[7]ZHAI Y J,YANG K,et al.Disc Insulator Defect DetectionBased on Mixed Sample Transfer Learning[C]//Proceedings of the CSEE.2023:2867-2876.
[8]LIU C Y,WU Y Q,LIU J J,et al.Insulator Faults Detection in Aerial Images from High-Voltage Transmission Lines Based on Deep Learning Model[J].Applied Sciences,2021,11(10):4647.
[9]YI W,MA S,LI R.Insulator and Defect Detection Model Based on Improved Yolo-S[J].IEEE Access,2023,11:93215-93226.
[10]WANG T,ZHAI Y D,LI Y H,et al.Insulator Defect Detection Based on ML-YOLOv5 Algorithm[J].Sensors,2024,24(1):204.
[11]陈华杰,吕丹妮.一种基于逆迁移学习的小样本目标识别辅助数据挑选方法[P].CN112541539A,2021-03-23.
[12]SAMET N,HICSONMEZ S,AKBAS E.HoughNet:Integrating Near and Long-Range Evidence for Visual Detection[J].IEEE Transactions on Pattern Analysis and Machine Intelligence,2023,45(4):4667-4681.
[13]JOCHER G.Ultralytics YOLOv8[EB/OL].[2023-07-05].https://github.com/ultralytics/ultralytics.
[14]WANG C Y,BOCHKOVSKIY A,LIAO H M.YOLOv7:Trainable bag-of-freebies sets new state-of-the-art for real-time object detectors[J].arXiv:2207.02696,2023.
[15]LIU S,QI L,QIN H F,et al.Path aggregation network for instance segmentation[C]//2018 IEEE Conference on Computer Vision and Pattern Recognition.2018:8759-8768.
[16]OUYANG D L,HE S,ZHANG G Z,et al.Efficient Multi-Scale Attention Module with Cross-Spatial Learning[J].arXiv:2305.13563,2023.
[17]HOU Q B,ZHOU D Q,FENG J S.Coordinate attention for efficient mobile network design[C]//IEEE Conference on Computer Vision and Pattern Recognition.2021:13708-13717.
[18]ZHENG Z H,WANG P,LIU W,et al.Distance-IoU Loss:Faster and Better Learning for Bounding Box Regression[J].AAAI Conference on Artificial Intelligence,2020,34:12993-13000.
[19]WANG J W,XU C,YANG W,et al.A Normalized GaussianWasserstein Distance for Tiny Object Detection[J].arXiv:2110.13389,2021.
[20]CHENG G,YUAN X,YAO X W,et al.Towards Large-Scale Small Object Detection:Survey and Benchmarks[J].IEEE Transactions on Pattern Analysis and Machine Intelligence,2023,45(11):13467-13488.
[21]ZHANG Y F,REN W Q,ZHANG Z,et al.Focal and efficient IoU loss for accurate bounding box regression[J].Neurocomputing,2022,506:146-157.
[22]GEVORGYAN Z.SIoU Loss:More Powerful Learning forBounding Box Regression[J].arXiv:2205.12740,2022.
[23]YANG L X,ZHANG R Y,XIE X H.SimAM:A Simple,Parameter-Free Attention Module for Convolutional Neural Networks[C]//International Conference on Machine Learning.2021.
[24]MISRA D,NALAMADA T,ARASANIPALAI A U,et al.Rotate to Attend:Convolutional Triplet Attention Module[C]//2021 IEEE Winter Conference on Applications of Computer Vision.2021:3138-3147.
[25]REN S Q,HE K M,et al.Faster R-CNN:Towards real-time object detection with region proposal networks.IEEE Transactions on Pattern Analysis and Machine Intelligence,2017,39(6):1137-1149.
[26]TERVEN J AND CORDOVA-ESPARZA D.A comprehensive review of YOLO:From YOLOv1 to YOLOv8 and beyond[J].arXiv:2304.00501,2023.
[1] GUAN Xin, YANG Xueyong, YANG Xiaolin, MENG Xiangfu. Tumor Mutation Prediction Model of Lung Adenocarcinoma Based on Pathological [J]. Computer Science, 2025, 52(6A): 240700010-8.
[2] TAN Jiahui, WEN Chenyan, HUANG Wei, HU Kai. CT Image Segmentation of Intracranial Hemorrhage Based on ESC-TransUNet Network [J]. Computer Science, 2025, 52(6A): 240700030-9.
[3] CHEN Xianglong, LI Haijun. LST-ARBunet:An Improved Deep Learning Algorithm for Nodule Segmentation in Lung CT Images [J]. Computer Science, 2025, 52(6A): 240600020-10.
[4] CHEN Qirui, WANG Baohui, DAI Chencheng. Research on Electrocardiogram Classification and Recognition Algorithm Based on Transfer Learning [J]. Computer Science, 2025, 52(6A): 240900073-8.
[5] WANG Rong , ZOU Shuping, HAO Pengfei, GUO Jiawei, SHU Peng. Sand Dust Image Enhancement Method Based on Multi-cascaded Attention Interaction [J]. Computer Science, 2025, 52(6A): 240800048-7.
[6] WANG Baohui, GAO Zhan, XU Lin, TAN Yingjie. Research and Implementation of Mine Gas Concentration Prediction Algorithm Based on Deep Learning [J]. Computer Science, 2025, 52(6A): 240400188-7.
[7] ZHENG Chuangrui, DENG Xiuqin, CHEN Lei. Traffic Prediction Model Based on Decoupled Adaptive Dynamic Graph Convolution [J]. Computer Science, 2025, 52(6A): 240400149-8.
[8] HONG Yi, SHEN Shikai, SHE Yumei, YANG Bin, DAI Fei, WANG Jianxiao, ZHANG Liyi. Multivariate Time Series Prediction Based on Dynamic Graph Learning and Attention Mechanism [J]. Computer Science, 2025, 52(6A): 240700047-8.
[9] TENG Minjun, SUN Tengzhong, LI Yanchen, CHEN Yuan, SONG Mofei. Internet Application User Profiling Analysis Based on Selection State Space Graph Neural Network [J]. Computer Science, 2025, 52(6A): 240900060-8.
[10] LI Daicheng, LI Han, LIU Zheyu, GONG Shiheng. MacBERT Based Chinese Named Entity Recognition Fusion with Dependent Syntactic Information and Multi-view Lexical Information [J]. Computer Science, 2025, 52(6A): 240600121-8.
[11] WU Zhihua, CHENG Jianghua, LIU Tong, CAI Yahui, CHENG Bang, PAN Lehao. Human Target Detection Algorithm for Low-quality Laser Through-window Imaging [J]. Computer Science, 2025, 52(6A): 240600069-6.
[12] ZENG Fanyun, LIAN Hechun, FENG Shanshan, WANG Qingmei. Material SEM Image Retrieval Method Based on Multi-scale Features and Enhanced HybridAttention Mechanism [J]. Computer Science, 2025, 52(6A): 240800014-7.
[13] HOU Zhexiao, LI Bicheng, CAI Bingyan, XU Yifei. High Quality Image Generation Method Based on Improved Diffusion Model [J]. Computer Science, 2025, 52(6A): 240500094-9.
[14] DING Xuxing, ZHOU Xueding, QIAN Qiang, REN Yueyue, FENG Youhong. High-precision and Real-time Detection Algorithm for Photovoltaic Glass Edge Defects Based onFeature Reuse and Cheap Operation [J]. Computer Science, 2025, 52(6A): 240400146-10.
[15] LI Mingjie, HU Yi, YI Zhengming. Flame Image Enhancement with Few Samples Based on Style Weight Modulation Technique [J]. Computer Science, 2025, 52(6A): 240500129-7.
Viewed
Full text


Abstract

Cited

  Shared   
  Discussed   
No Suggested Reading articles found!