Input attention, squeeze and excitation, and spatial transformer of YOLO for fault detection using UAV

Abstract

The detection of faults in insulators is important to guarantee the continuous supply of electricity. To identify faults in these components, various object detection methods based on deep learning have been explored. This paper investigates architectural enhancements to the You Only Look Once (YOLO) framework for fault detection in electrical power grid insulators. Three structural variants are proposed: the Input Attention Transformer (IAT-YOLO) for spatial feature refinement, Squeeze-and-Excitation (SAE-YOLO) modules for channel recalibration, and Spatial Transformer Networks (STN-YOLO) for geometric alignment. Experiments were conducted on a publicly available insulator dataset from Unmanned Aerial Vehicles (UAVs), comprising seven defect categories, including pollution, breakage, and flashover damage. Results demonstrate that STN-YOLO and SAE-YOLO consistently improve generalization and robustness, achieving mAP values of up to 0.995 for specific classes. The findings highlight the effectiveness of integrating attention mechanisms and spatial transformations to enhance YOLO-based detection, contributing to improved automated inspection of the power grid.