1. Introduction
Along with the sustained development of the Chinese economy, the mileage of transmission lines has increased rapidly [
1]. As the core infrastructure for long-distance energy transmission, their reliable operation is essential for meeting the growing demand for electricity [
2]. However, transmission lines are exposed to harsh outdoor environments [
3], where prolonged exposure significantly raises the risk of diverse faults—ranging from component damage to foreign object intrusions—collectively referred to as transmission line defects [
4].
Unmanned aerial vehicles (UAVs) are increasingly employed to capture imagery of transmission lines for post-flight visual inspection and defect detection [
5]. In such UAV-acquired imagery, defects often appear as small objects due to long shooting distances and the inherent resolution limitations of aerial imaging systems [
6]. These small objects typically occupy regions smaller than 32 × 32 pixels, lacking clear contours and distinct textures, which severely hampers reliable recognition [
7]. The challenge is further exacerbated by complex backgrounds such as vegetation, towers, and shadows, where small defects are frequently obscured or confused with surrounding structures [
8].
From a system perspective, the present work focuses on the post-processing stageof the inspection workflow, where UAVs execute routine patrols, capture high-resolution imagery along the transmission corridor, and subsequently upload these data to a ground-based server for automated analysis. In this paradigm, defect detection is conducted offline as a post-processing task rather than on-board the UAV, thus avoiding the stringent real-time and energy constraints associated with aerial platforms. This server-side processing setting enables the adoption of more expressive Transformer-based architectures and advanced optimization strategies while still meeting operational maintenance timelines. Meanwhile, the proposed model remains sufficiently lightweight to support potential migration to edge-computing platforms, achieving a practical balance between detection accuracy, robustness, and deployment flexibility.
With the rapid advancement of smart grid development, transmission line defect detection has increasingly adopted deep learning-based object detection methods [
9]. Alongside this progress, numerous general-purpose detectors—such as YOLO [
10], DINO [
11], and RT-DETR [
12]—have been developed. While these models achieve remarkable success in generic scenarios, most of them on the COCO dataset [
13], they face notable limitations in transmission line defect detection under complex conditions involving occlusion, background clutter, and variable lighting [
14]. Common issues include missed detections, false positives, and false negatives, which substantially undermine the practical utility of UAV-acquired imagery in real-world transmission line maintenance [
8]. These shortcomings are consistently reported in previous studies and further confirmed by our preliminary experiments, highlighting the critical need for specialized detection frameworks capable of accurately identifying small-scale defects in UAV operational scenarios.
In this article, we propose TinyDef-DETR, a novel transmission line defect detection network built upon the Transformer [
15] architecture, which aims to enhance detection accuracy, recall, and robustness for UAV-acquired imagery under complex inspection scenarios. The main contributions of this work are summarized as follows:
- 1.
We construct an Edge Enhanced ResNet (EE-ResNet) by embedding an Edge-Enhanced Convolution (EEConv) module into the residual backbone. The EEConv integrates central, horizontal, and vertical difference convolutions to encode gradient priors, thereby improving sensitivity to fine-grained details and object boundaries. Through a re-parameterization strategy, the multi-branch structure of EEConv is fused into a single kernel at inference, incurring no additional computational overhead. By augmenting the ResNet backbone with such edge-aware feature extraction, EE-ResNet substantially strengthens the representation capability for small-scale defects in UAV inspection imagery.
- 2.
We introduce a stride-free downsampling module based on space-to-depth transformation, which reorganizes spatial information into the channel dimension using a fixed partition without stride convolution. This design preserves pixel-level details while reducing spatial resolution, thereby improving recall for extremely small defects that are often missed by conventional strided convolution architectures.
- 3.
We propose a lightweight attention block with a dual-domain FSCA (Frequency-Spatial-Channel Attention) mechanism, integrated in a cross-stage partial manner. This design jointly enhances global semantic reasoning and local detail preservation, enabling the network to better capture anisotropic defect patterns such as missing bolts, loose clamps, and sagging wires.
- 4.
We develop a novel Focaler-Wise-SIoU loss that combines normalized IoU scaling, Scylla-IoU geometric penalties, and non-monotonic focal modulation. This adaptively emphasizes moderately difficult samples while suppressing trivial or noisy ones, thereby stabilizing training and improving localization accuracy for small-scale defect regions.
3. Materials and Methods
3.1. Overview
In this study, we employ RT-DETR-r18 [
12] as the baseline detection framework owing to its superior accuracy and robustness in complex scenarios. Compared with traditional DETR models, RT-DETR exhibits significantly faster training convergence and improved deployment efficiency, making it more suitable for real-time and resource-constrained applications such as UAV-based inspection tasks [
29].
RT-DETR incorporates an enhanced encoder–decoder architecture together with refined query interaction mechanisms, while for the backbone network, RT-DETR-r18 adopts a CNN-based design, namely the popular ResNet-18 [
30]. From the perspective of real-time performance, employing a CNN architecture for feature extraction further enhances the real-time capability of DETR-like models, thereby improving their practical applicability. As a result, RT-DETR not only accelerates convergence and improves detection precision, but also preserves the global context modeling capability of Transformer-based frameworks [
15]. In addition, its lightweight optimization strategies—specifically, the design choice to apply the encoder only to high-level semantic features (S5) with relatively small spatial dimensions—effectively reduce resource consumption. This balances performance and speed by significantly decreasing computational overhead and accelerating inference without compromising accuracy, thereby making RT-DETR particularly suitable for post-flight UAV imagery analysis, where large volumes of data must be processed efficiently under computational and energy constraints [
16].
By contrast, YOLO-based detectors, though widely favored for their real-time efficiency and ease of deployment [
31], tend to degrade in performance when handling small, occluded, or densely distributed objects, owing to their anchor-dependent design and limited global receptive field. This aligns with the findings of Ismail et al. [
32], who reported that while YOLO excels in real-time scenarios, it struggles with small-object detection and occlusion, whereas DETR achieves higher accuracy in complex environments at the cost of greater computational demand. Our preliminary experiments further compared RT-DETR with representative YOLO series models, including YOLOv3 [
33], YOLOv7 [
34], and YOLO 11 [
35], and showed that RT-DETR consistently outperformed most of them in overall performance. In cases where certain YOLO variants (e.g., YOLO 11m) surpassed RT-DETR, their gains were typically accompanied by substantially larger parameter sizes, thereby underscoring RT-DETR’s advantageous balance between model complexity and detection accuracy.
Taken together, the consistent advantages of RT-DETR over both traditional DETR models and representative YOLO series substantiate its adoption as the baseline framework in this study.
Building upon this strong foundation, we propose TinyDef-DETR as a dedicated solution for small object defect detection in UAV-based transmission line inspection. The overall architecture is illustrated in
Figure 1.
First, the backbone adopts the Edge-Enhanced ResNet (EE-ResNet) structure to strengthen low-level detail preservation. Second, a compact Space-to-Depth Convolution (SPD) module is inserted before each downsampling layer to reorganize spatial features in a manner that preserves critical spatial information while reducing resolution, thereby improving the recall rate of extremely small objects. Third, a lightweight Cross-Stage Dual-Domain Multi-Scale Attention Module (CSDMAM) is embedded into the backbone network to enhance directional sensitivity and fine-grained structural modeling, effectively capturing anisotropic features of typical defects such as missing bolts, loose clamps, or sagging wires. Finally, a Focaler-Wise-SIoU loss function is employed to jointly optimize classification and localization. This loss function adaptively balances easy and hard samples while providing more accurate bounding box regression through shape-aware optimization, which further improves detection robustness for small and ambiguous objects. Detailed descriptions of the proposed modules can be found in
Section 3.2,
Section 3.3,
Section 3.4 and
Section 3.5.
3.2. Edge Enhanced ResNet
ResNet [
30] represents a milestone in deep learning for computer vision, introducing residual learning that enables the effective training of very deep networks. Owing to its simple yet modular design, it has become a widely adopted backbone across a broad spectrum of vision tasks. Nevertheless, when applied to UAV-based visual inspection for transmission line defect detection, native ResNet architectures, as verified in preliminary experiments, exhibit limited capability in capturing sufficiently fine-grained features, thereby constraining their effectiveness in identifying small and subtle defect objects, particularly in challenging transmission line inspection environments with interference from vegetation, man-made structures, and adjacent electrical installations.
Inspection of the heatmaps exported from ResNet further reveals that they lack clear edges and fine contours, which exacerbates the difficulty of accurately localizing subtle defects. Unlike conventional convolution, where the high performance of CNN-based edge feature extraction typically relies on large pretrained backbones—resulting in high memory consumption and energy cost, which contradicts the real-time requirements of UAV-based transmission line defect detection tasks—difference convolution introduces differential operations to enhance sensitivity to gradient variations and fine-grained structural details, thereby improving the extraction of local details and object boundaries [
36].
To address the aforementioned problem, we propose an EEConv (edge enhancement convolution). Building on this idea, EEConv explicitly incorporates three difference convolution operators—central difference convolution (CDC), horizontal difference convolution (HDC), and vertical difference convolution (VDC)—to embed gradient priors into the feature extraction process. The overall structure of EEConv is illustrated in
Figure 2a. To ensure deployment efficiency, EEConv adopts a re-parameterization strategy that fuses its multi-branch convolutions into a single equivalent 3 × 3 kernel during inference. Once fused, the auxiliary branches are discarded, leaving only a lightweight standard convolution. This allows the module to preserve the expressive power gained from multi-branch training while maintaining the same computational cost as a vanilla convolution.During deployment, the structure of EEConv becomes as shown in
Figure 2b.
Formally, let the four convolutional branches be denoted as
, corresponding to CDC, HDC, VDC, and the Vanilla convolution, respectively. During training, their weights and biases are linearly aggregated to obtain an equivalent
kernel and bias:
The intermediate feature map is then obtained as
followed by batch normalization and a non-linear activation
:
For inference, the BN parameters are absorbed into the convolution kernel, yielding per-channel scaling:
and the final equivalent kernel and bias are computed as
Thus, during deployment, EEConv degenerates into a single
convolution followed by activation:
which guarantees equivalence with the training-time formulation while ensuring zero additional inference overhead.
To integrate this into our framework, we replace the second
convolution in the residual branch of standard ResNet basic blocks with EEConv, thereby preserving the backbone’s modular design while substantially improving sensitivity to fine-grained structures. This design choice is motivated by the observation that the second
convolution in each residual block primarily functions as a feature refiner, responsible for aggregating local structural details after the first convolution has extracted preliminary features. In this way, we obtain the Edge-Enhanced Block (EEBlock), whose structure is illustrated in
Figure 2c.
By substituting the refining convolution with EEConv, we embed directional gradient priors precisely at the stage where fine-grained information is consolidated, thus maximizing the impact of EEConv on enhancing edge and detail representation. Meanwhile, the first convolution and the shortcut connection remain unchanged, ensuring full compatibility with the original ResNet architecture and preserving its residual learning mechanism.
Integrating EEBlocks throughout the backbone yields the Edge-Enhanced ResNet (EE-ResNet), which significantly improves sensitivity to subtle structural details while maintaining computational efficiency. EE-ResNet provides a robust feature extraction foundation, offering stronger boundary-aware representations and improved resilience against background interference. This enhanced backbone design directly addresses the core challenge of detecting small-scale and fine-grained defects in transmission line imagery, thereby serving as a crucial cornerstone for the subsequent TinyDef-DETR framework. The overall architecture of the proposed EE-ResNet is illustrated in
Figure 3.
3.3. Space-to-Depth Convolution
3.3.1. Principle of SPD (Space-to-Depth Convolution)
Given the nature of transmission line defect detection—where the dataset is predominantly composed of small objects—preserving fine-grained information, maintaining balanced feature weighting, and performing effective feature representation learning contribute to improving both accuracy and recall [
2].
During an extensive review of the literature, we came across the work of Sunkara and Luo [
37], who pointed out that conventional CNN architectures exhibit an inherent limitation, namely the use of strided convolution. They emphasized that such designs inevitably lead to the loss of fine-grained information and result in less effective feature representations, which is particularly detrimental for low-resolution images and small object detection. This explains why conventional CNN-based approaches often fail to meet the requirements of small object detection, where the preservation of subtle spatial details is crucial.
Inspired by their findings, we incorporate the SPD(Space-to-depth Convolution) module into our proposed TinyDef-DETR framework. SPD adopts a stride-free downsampling strategy that redistributes pixels along the channel axis instead of discarding them, thereby preserving pixel-level information while reducing spatial resolution. In this way, TinyDef-DETR achieves more effective feature representations, improving both accuracy and recall for transmission line defect detection.
The SPD operation is defined on an input feature map
where
S denotes the spatial resolution and
C is the channel dimension. The goal is to downsample the spatial resolution by a factor of scale while preserving all pixel-level information. Unlike conventional strided convolution that inevitably discards part of the input, SPD achieves lossless downsampling by spatial-to-channel rearrangement. Formally, the output feature map is given by
This transformation can be expressed as a structured partitioning of
X. Specifically, for each spatial position
in
, its corresponding feature vector is constructed by concatenating the values of
X sampled from the
local neighborhood in the original feature map:
where ⨁ denotes concatenation along the channel dimension. In this way, all pixels from
X are preserved in
, but redistributed from the spatial dimensions into the channel dimension.
After the SPD transformation, the spatial resolution of the feature map is reduced by a factor of scale, while all pixel-level information is preserved through spatial-to-channel rearrangement. Consequently, the number of channels increases by a factor of
. To further process this rearranged representation, a subsequent convolution layer is applied, which can be formally expressed as
where
and
denote the convolutional kernel weights and bias, respectively. This convolution simultaneously fuses information across the expanded channel dimension and adjusts it to the object dimension
. In this way, the operation not only regulates the channel dimensionality but also enhances local feature representation through the structured partitioning.
An illustrative example of the SPDConv operation is presented in
Figure 4.
3.3.2. Choice of the Downsampling Factor
A critical design decision in the SPD module is the choice of the downsampling factor scale.
Constraint of Small Object Size. In our dataset, small objects including polluted insulator, missing tie wire, and bird nest frequently span only pixels in the original image. If (or larger) were adopted, the smallest objects would be reduced to merely pixels at the first feature level. After further downsampling by the backbone and feature pyramid, these objects would rapidly degenerate into near-point responses, severely hindering recall. By contrast, preserves the effective resolution of small objects at the pixel level, which can still be robustly captured and aggregated by local operators.
Compatibility with Detector Stride Hierarchy. Unlike many mainstream detectors that organize feature pyramids with strides , our Edge Enhanced ResNet employs stage-wise downsampling with stride = 2. Consequently, setting SPD is equivalent to a lossless stride-2 operation, perfectly aligned with the subsequent stride-2 downsampling in the backbone/pyramid, preserving high-resolution shallow features crucial for small objects. In contrast, starting with effectively performs a stride-4 rearrangement at the outset, over-compressing shallow maps and weakening the small-object modeling capacity of early pyramid levels.
Anti-Aliasing and Information Fidelity. When , the neighborhood is completely mapped into the channel dimension, ensuring that no samples are discarded. Although also performs rearrangement, the resulting features must be processed at a much lower spatial resolution, reducing the discriminability of local topology and diminishing the effective anti-aliasing capability.
Efficiency in Computation and Memory. After SPD, the channel dimension increases by a factor of . When , channels are quadrupled while spatial dimensions are reduced by one quarter. As a result, the computational cost of the following convolution remains comparable to that of a standard stride-2 convolution, but without discarding information. In contrast, would increase the channel dimension by sixteen while reducing spatial size by one sixteenth, leading to heavier memory and bandwidth requirements and potentially unstable training.
Taken together, these factors demonstrate that setting provides the optimal trade-off between recall and efficiency in datasets dominated by small objects, and we therefore adopt it as the fixed parameter in the SPD module of TinyDef-DETR.
3.3.3. Design of the Subsequent Convolution
Following SPD, we adopt a carefully designed convolutional layer to refine the rearranged features. Specifically, we employ a non-strided convolution with a kernel size of . This configuration is chosen because the kernel provides the minimal receptive field capable of covering the local neighborhood that is redistributed during SPD. In contrast, a kernel would only perform channel mixing without explicitly modeling cross-partition spatial relationships, which are essential for capturing subtle structural variations in small defects. Larger kernels such as would increase computation without offering proportional benefits for objects of the scale encountered in transmission line inspection.
We set the stride to 1 in order to avoid any additional downsampling. Since the primary motivation of SPD is to preserve pixel-level detail while reducing resolution in a lossless manner, further spatial reduction at this stage would counteract the benefits of SPD and degrade the detection of small defects.
In terms of channel configuration, the input dimensionality after SPD expands to four times the original number of channels, consistent with . This expansion ensures that all pixel-level information is preserved in the channel space. The output channel dimension is then set to match the base width of the subsequent stage in TinyDef-DETR, ensuring both computational efficiency and stable gradient propagation across the backbone.
3.4. Cross-Stage Dual-Domain Multi-Scale Attention Module
Transmission line defects in UAV imagery—such as cracks, rust spots, and insulator damages—are typically small, weakly salient, and embedded in cluttered backgrounds of vegetation, sky, or towers. Their ambiguous boundaries and subtle textural cues challenge conventional CNNs, whose local receptive fields hinder simultaneous modeling of global context and fine detail. Simply enlarging kernels inflates parameters and computation, often causing oversmoothing that erases critical structures [
38]. Likewise, attention mechanisms restricted to a single domain—spatial or channel—offer limited gains for tiny, low-contrast targets [
39]. Building on the joint frequency–spatial perspective outlined in
Section 2.2, we introduce the Cross-Stage Dual-Domain Multi-Scale Attention Module (CSDMAM), which unifies frequency-domain selectivity with spatial-domain localization in an end-to-end attention framework.
The central idea is to embed learnable frequency gating directly within attention formation while retaining anisotropic, large-kernel spatial paths tailored to line-shaped defects. Frequency responses preserve edges, textures, and other high-frequency cues and suppress low-frequency clutter, whereas spatial aggregation localizes potential defects under complex context. Rather than treating frequency information as an auxiliary branch, CSDMAM integrates it adaptively so that frequency cues steer attention weights toward defect-relevant structures and away from distractors.
Given an input feature map
, the module first applies a
convolution to adjust channel dimensionality. The transformed tensor then flows through several parallel, specialized branches that operate across complementary scales. One branch performs a learnable frequency decomposition whose gated responses accentuate sharp structural variations indicative of defects while attenuating smooth backgrounds; these gated maps participate directly in attention computation. In parallel, an anisotropic large-kernel spatial path—implemented with directional and dilated filters—approximates a wide receptive field without excessive parameters and mitigates oversmoothing, thereby enhancing sensitivity to elongated conductor- and insulator-like patterns. Additional scale-selective paths aggregate broader semantic context and preserve fine detail. The branch outputs are merged by residual summation and projected with a final
convolution, as illustrated in
Figure 5.
Through this dual-domain, multi-scale design, CSDMAM preserves high-frequency, edge-dominated details, captures long-range semantics with modest overhead, and adaptively highlights subtle, defect-related responses. The resulting module addresses the principal failure modes of purely convolutional and single-domain attention approaches, delivering robust detection of small, low-saliency defects in complex aerial scenes.
To effectively capture complementary information at different scales and semantic levels, CSDMAM is composed of four parallel branches: a Large-Receptive-Field Branch, a Local Detail Preservation Branch, a Global Branch based on the Frequency–Spatial–Channel Attention (FSCA) module, and a Residual Aggregation Branch.
Large-Receptive-Field Branch. To enlarge the receptive field, we employ depth-wise-separable convolutions with kernels
,
and
, where
. The square
kernel aggregates global context, while the horizontal and vertical strip kernels capture elongated dependencies that are critical for the line-shaped structures typical of transmission infrastructure. The choice
was determined by systematic empirical evaluation rather than by heuristic selection (see
Section 4.3). Kernels substantially smaller than 31 fail to capture the long-range, directionally elongated dependencies of line-like defects, whereas substantially larger kernels introduce excessive smoothing and disproportionate computational overhead without commensurate accuracy gains. Therefore, a
depth-wise-separable kernel provides the most favorable trade-off between global context aggregation and detail preservation for our task. We further restrict
k to the family
(here
) for two practical reasons. First, an odd kernel size preserves a single central pixel, which simplifies symmetric, centered convolutional responses and padding. Second, when features are aggregated across hierarchical stages that expand spatial support by factors of two—via pooling, striding, or stage-wise downsampling—the geometric series
naturally arises. Selecting
aligns the kernel’s spatial support with an
n-level binary-scale expansion, facilitating efficient cross-scale context aggregation and predictable padding behavior. The ablation in
Section 4.3 confirms that
is optimal in our application.
Local Detail Preservation Branch. To prevent important fine-grained cues from being washed out by large-kernel operations, we introduce a lightweight local path using a depth-wise convolution, which preserves high-frequency details such as micro-cracks and corrosion spots while adding negligible computational burden.
Global Branch. To strengthen semantic discrimination and capture long-range contextual information, we design a global branch built upon the FSCA module that jointly exploits frequency-, spatial-, and channel-wise dependencies. Given the input
X, we first compute the frequency-domain representation
where
denotes the discrete Fourier transform (DFT). Channel-wise weights are then obtained via global average pooling followed by a
convolution, yielding
. These weights modulate the frequency response
which is transformed back to the spatial domain through the inverse DFT
Finally, a spatial–channel reweighting map
is generated via global pooling and a
convolution to adaptively emphasize defect-relevant regions, producing the branch output
The effectiveness of FSCA and the integration of frequency-domain cues is validated in
Section 4.3.
Residual Aggregation Branch. To effectively integrate complementary information from all branches while preserving the original signals, we adopt a residual fusion mechanism that combines the outputs of the large-receptive-field branch, the local branch, and the FSCA branch with the input via element-wise summation:
where
denotes depth-wise convolution with the specified kernel size. This design enhances gradient flow, stabilizes optimization, and consolidates global context, fine details, and dual-domain attention responses into a unified representation for precise defect localization.
To further improve computational efficiency and facilitate gradient propagation, we extend CSDMAM into a cross-stage structure following the Cross Stage Partial (CSP) strategy. The input is split into two parts: one processed by the multi-branch attention module and the other preserved as an identity mapping. The two parts are concatenated and fused by a
convolution to produce the final output:
By jointly leveraging large receptive fields, local detail preservation, FSCA-based dual-domain enhancement, and cross-stage fusion, CSDMAM achieves a favorable balance between global semantic understanding and precise defect localization, significantly improving the detection of small, visually ambiguous defects in UAV inspection imagery.
3.5. Focaler–Wise–SIoU Loss
In transmission-line defect detection, targets are extremely small and exhibit subtle spatial variations, so precise bounding-box regression is pivotal for reliable recognition. IoU-based losses are attractive for their geometric interpretability and optimization efficiency, yet the canonical IoU measures only overlap and provides no informative gradients when the boxes are disjoint, which stalls learning on difficult examples. Subsequent variants—GIoU [
40], DIoU [
41], and CIoU [
42]—mitigate this to some extent by adding enclosure, center-distance, and aspect-ratio terms. Nevertheless, they still struggle to capture spatial alignment and shape consistency in a unified manner. More advanced designs such as SIoU [
43] integrate angular, distance, and shape components, but typically weight all samples uniformly; gradients then become dominated by easy cases while hard, tiny, low-SNR objects receive insufficient emphasis. Focaler–IoU [
44] further reshapes the overlap signal by linearly remapping IoU within a fixed interval to favor certain ranges, but its emphasis is uniform and monotonic and it lacks angle-aware distance/shape coupling. Wise-IoU [
45] introduces a dynamic, non-monotonic focusing mechanism that suppresses low-quality examples, yet its default formulation is ill-suited to our setting: it deemphasizes geometric shape and aspect consistency that are crucial for sub-pixel localization of tiny defects, it normalizes outlier degree at the batch level—which can miscalibrate under low-SNR, highly imbalanced anchors and inadvertently down-weight informative hard-but-valid samples—and it omits explicit angle-aware distance coupling, yielding weaker guidance for axis-aligned center drift near overlap. Compounding these issues, conventional IoU-based objectives offer limited discrimination in the high-IoU regime where fine-grained localization matters most, leading to slow convergence and reduced accuracy.
To address these shortcomings, we propose Focaler–Wise–SIoU, a unified regression objective that couples normalized IoU scaling, comprehensive geometric penalties, and difficulty-aware modulation. The goal is to sustain strong, stable gradients across the entire overlap spectrum while explicitly promoting spatial alignment and shape fidelity for tiny objects. Concretely, we first introduce a Focaler-IoU overlap surrogate that clips and linearly rescales IoU to sharpen gradients near perfect alignment, while preserving the raw IoU for principled reweighting. We then design SIoU-style geometric penalties that combine an angle-aware, axis-normalized center-distance term with a smooth, saturating shape discrepancy, encouraging accurate alignment and aspect consistency under tiny-object noise. Finally, we incorporate a non-monotonic focal modulation (Wise) driven by an EMA-normalized IoU gap, which highlights informative, moderately hard samples and down-weights both trivial cases and extreme outliers. Raw IoU is used exclusively for difficulty estimation to retain geometric interpretability, and the overall objective remains closed-form, differentiable, and implementation-friendly. These components jointly accelerate convergence and substantially improve localization accuracy on small, low-SNR defects.
The following subsections introduce notation, define each component with its rationale, and present the final loss composition.
3.5.1. Preliminaries and Notation
Let the predicted and target bounding boxes be
and
, parameterized by center coordinates
and width/height
. We write the top-left/bottom-right corners as
The intersection size is
with area
. The union area
, and the standard IoU is
We also denote the center displacement by
, and the side lengths of the minimum enclosing rectangle of
and
by
We use small constants where needed:
in Equation (
7) and
in
Section 3.5.5 to avoid singularities; the EMA momentum
balances variance and adaptability across training phases.
3.5.2. Focaler-IoU Overlap Term
In transmission-line defect detection most objects are tiny; training benefits from non-saturating gradients even at high overlaps. To strengthen gradient sensitivity near perfect alignment while keeping the geometric meaning of Equation (
3) for reweighting, we use a clipped and linearly rescaled proxy
only inside the overlap penalty:
and define
Using
solely in
preserves gradient magnitude as
while the
raw IoU (Equation (
3)) is retained for difficulty modulation, maintaining geometric interpretability.
3.5.3. Angle-Weighted Distance Penalty
In transmission line defect detection tasks, objects such as insulator cracks, missing bolts, or bird nests often appear at arbitrary orientations and with substantial variation in scale and spatial distribution. These defects are typically small, elongated, and densely distributed along complex backgrounds such as towers or cables, especially in aerial inspection imagery. In such conditions, even minor center displacements between predicted and ground-truth bounding boxes can result in significant localization errors, particularly when the offset directions are not well aligned. Conventional distance-based penalties treat all directions uniformly—whether horizontal, vertical, or diagonal—thereby neglecting the geometric structure inherent in these domain-specific misalignments.
This limitation motivates us to develop an angle-weighted distance penalty that dynamically adjusts the localization penalty according to the angular alignment of the offset vector. By doing so, the model is encouraged to favor axis-aligned center corrections over diagonal ones, improving directional sensitivity and enhancing localization accuracy under the challenging settings encountered in power line defect detection.
We define the offset between the predicted and ground-truth box centers as
, and let
denote the acute angle between
and its nearest coordinate axis. This angle reflects the degree of directional misalignment:
indicates perfect alignment with one axis, while
corresponds to a fully diagonal offset. It is computed as:
where a small constant in the denominator ensures numerical stability near zero displacement.
In the original SIoU formulation, the angle cost is defined as:
which increases as the offset becomes more axis-aligned. When the displacement vector approaches either the horizontal or vertical axis, the angle
becomes small, resulting in a higher
, indicating better alignment between the predicted and target boxes.
To incorporate this angular alignment into the penalty formulation, we define a scaling factor that modulates the influence of the distance terms:
which ensures that
. This negative-valued quantity is then used in an exponential decay to penalize displacement, such that larger directional misalignment results in steeper penalties. Specifically, when the offset direction is diagonal (i.e., small
), the penalty grows more aggressively; when it is axis-aligned (i.e., large
), the penalty grows more slowly. This behavior makes the model more sensitive to the type of misalignment rather than only its magnitude.
Based on this principle, we formulate the distance penalty as:
where
and
represent the width and height of the smallest enclosing box that covers both the predicted and ground-truth boxes. Each exponential term decreases as the normalized offset increases due to the negative scaling factor, ensuring that
is monotonic in misalignment. Moreover, axis-wise normalization promotes scale invariance, which is crucial for handling object size variations in diverse datasets.
In summary, this angle-weighted distance penalty effectively captures both the magnitude and direction of misalignment. It provides stronger supervision in challenging cases such as small, rotated, or overlapping objects, as commonly observed in the DIOR and VisDrone datasets. Empirically, we find that incorporating this term improves detection accuracy by enhancing the localization sensitivity to angular deviation.
3.5.4. Shape Consistency Penalty
Tiny defects often suffer from annotation noise; a saturating yet smooth transform mitigates the dominance of extremely large aspect errors while keeping sensitivity to moderate mismatches. We adopt
For small relative deviations, is near-linear; for large deviations it saturates towards 1, and the exponent further curbs outliers while preserving gradients for mid-range shape errors. This saturating design improves robustness to annotation noise on tiny objects without letting large, noisy deviations dominate the loss.
Combining the overlap penalty (Equation (
6)) with the distance and shape terms (Equations (
9) and (
10)), we obtain the SIoU-style geometric core:
The factor balances alignment/shape against overlap so that none dominates the gradient field in typical training regimes. Note again that only uses the rescaled ; all reweighting below relies on the raw IoU.
3.5.5. Wise Difficulty Modulation via EMA Normalization
Precise regression of small defects in UAV imagery is hindered by an imbalance in sample difficulty: abundant trivial cases dominate gradients, while naïve hard mining overfits noisy outliers and mislabels. We therefore propose Wise Difficulty Modulation (WDM)—a normalization-driven reweighting scheme that scales each sample’s gradient by an IoU-based difficulty estimate. WDM stabilizes training via an EMA-normalized difficulty distribution that adapts as accuracy improves, and uses a non-monotonic modulation to emphasize moderately hard, informative samples while down-weighting both trivial and extreme cases. This yields smoother optimization and consistently better localization, especially for small or ambiguous defects common in UAV datasets.
We quantify sample difficulty using the normalized IoU gap:
where the expectation is tracked by an exponential moving average (EMA). We use
, which provides a stable yet responsive adaptation to phase shifts during training; in practice, a tiny constant
is added to the denominator for numerical stability. Crucially, Equation (
12) operates on the
raw IoU from Equation (
3) to preserve geometric interpretability. This normalization eliminates the influence of absolute IoU scale drift, allowing difficulty to be compared consistently across epochs and batch conditions.
To emphasize moderately hard examples while avoiding overfitting to extreme outliers, we employ the following modulation:
This non-monotonic weighting concentrates higher weights on samples with moderate normalized difficulty while gently down-weighting trivially easy cases and preventing domination by extreme outliers as increases. Empirically, such weighting increases the proportion of informative samples contributing to effective gradient updates, accelerating convergence and improving the recall of subtle, low-SNR targets without introducing instability.
Overall, Wise Difficulty Modulation via EMA Normalization offers a self-adaptive, statistically grounded mechanism to stabilize training under severe difficulty imbalance. In the context of transmission line defect datasets, it effectively improves localization precision and reduces missed detections of small and ambiguous defects by ensuring that learning focuses on the most informative samples throughout the optimization process.
3.5.6. Final Objective and Batch-Wise Computation
Multiplying the geometric core (Equation (
11)) by the difficulty weight (Equation (
13)) yields the
Focaler–Wise–SIoU objective. Recalling that
the final loss can be compactly written as
where
is defined from the raw IoU as in Equation (
12), and
is given in Equation (
13). The EMA-based normalization acts only through
, and no additional normalization is introduced in the final objective.
4. Results
4.1. Datasets
In this paper, we use the CSG-ADCD dataset. The CSG-ADCD (China Southern Grid Aerial Defective Component Dataset) is a large-scale, high-resolution UAV-based image dataset specifically developed for the detection of defects in power transmission line components, including insulators, tie wires, and poles. Constructed under authentic inspection scenarios by China Southern Power Grid, the dataset comprises 10,000 aerial images. Each image was manually annotated by a single domain expert to ensure both consistency and high-quality labeling.
In total, the dataset contains 73,448 annotated instances covering nine component states or defect types: Normal Insulator (zcjyz), Polluted Insulator (jyzwh), Damaged Pole (dgss), Missing Tie Wire (zxqs), Loose Tie Wire (zxst), Insulator Flashover (jyzsl), Bird Nest (nw), Broken Insulator (jyzps), and Shattered Post Insulator (zyzsl).
Table 1 provides a correspondence between the labels and category names for each defect type in the dataset, offering a clear overview of the classification scheme used.
On average, each object occupies a pixel area of 422.12, with the dataset exhibiting a highly imbalanced distribution in terms of object sizes: 94.51% small objects (69,413 instances), 5.48% medium-sized (4022 instances), and only 0.02% large (13 instances). This reflects the intrinsic characteristics of UAV-based transmission line inspection objects, which are often small, localized, and partially occluded.
For training consistency, all images were preprocessed by resizing the longer side to 640 pixels while preserving aspect ratio, followed by padding to a fixed resolution of 640 × 640 pixels using gray borders. The dataset is split into 8000 training, 1000 validation, and 1000 test images.
Statistical analysis further highlights two key features of the dataset. First, the scatter distribution of object width and height reveals a strong concentration of samples in the lower-left region, confirming the overwhelming dominance of small-scale objects. Second, the number of objects per image follows a relatively stable distribution, averaging 7.34 instances per image, with 2285 images containing between 7 and 8 objects. This regularity is particularly beneficial for developing models optimized for dense and small-object detection.
In summary, the CSG-ADCD dataset demonstrates strong representativeness by simultaneously addressing two critical aspects. On the one hand, it captures a comprehensive range of component defects and status indicators that are directly relevant to transmission safety, while also reflecting the complexity of real-world inspection conditions such as occlusion, cluttered backgrounds, and varying illumination. On the other hand, its pronounced dominance of small-scale objects—over 94% of all annotated instances—together with the presence of subtle and fine-grained anomalies, makes it a highly challenging and domain-specific benchmark. In this regard, CSG-ADCD not only advances research in defect detection for power systems but also provides a valuable testbed for generic small-object detection in aerial remote sensing, offering greater domain relevance than existing datasets such as VisDrone [
46].
Overall, CSG-ADCD constitutes a robust and high-fidelity benchmark, offering significant value for advancing research in both domain-specific defect detection and general small-object detection tasks.
As shown in
Figure 6, the vast majority of objects are smaller than 32 × 32 pixels, placing them within the small-object category. The scatter plot depicts the distribution of annotated instances by pixel width and height after preprocessing, with different categories distinguished by distinct colors in the legend. The strong concentration of points in the lower-left corner highlights the predominance of small and compact objects, consistent with the characteristics of transmission-line inspection imagery. Dashed reference lines at 32 × 32 and 96 × 96 pixels mark the boundaries between small, medium, and large regimes, clearly revealing the relative scarcity of medium- and large-sized objects.
Overall, the visualization emphasizes the dominance of small instances and underscores the dense, fine-grained nature of aerially captured transmission-line defects.
4.2. Model Training and Evaluation Metrics
The proposed model was implemented in the PyTorch framework and built on top of the Ultralytics RT-DETR backbone. Experiments were conducted on a workstation equipped with an NVIDIA GeForce RTX 3090 (24 GB) running Ubuntu 20.04, Python 3.10.14, PyTorch 2.3.1, and CUDA 12.1. Training proceeded for 100 epochs with a batch size of 4 and an input resolution of . Optimization employed AdamW with an initial learning rate of , a final learning-rate factor , momentum , and weight decay . We used a 2000-iteration warm-up during which the optimizer momentum was set to and the bias learning rate was scaled by a factor of .
Evaluation is conducted on the test split following the COCO evaluation protocol, where we report , the mean Average Precision at an IoU threshold of 0.50, and , the mean Average Precision averaged across IoU thresholds from 0.50 to 0.95 in increments of 0.05. During inference, class-agnostic non-maximum suppression (NMS) with an IoU threshold of 0.7 is applied, retaining at most 300 detections per image. Note that this NMS threshold is independent of the IoU thresholds used for AP and mAP computation.
The COCO protocol additionally provides scale-sensitive metrics that illuminate detector performance across object sizes: measures average precision for small objects (area pixels) and measures average precision for medium objects (area between and pixels). Because most defects in transmission-line inspection datasets are small, is especially informative of the model’s ability to detect fine-grained, low–pixel-count anomalies.
We evaluate inference efficiency and model complexity using standard measures: floating-point operations (FLOPs) quantify computational complexity and model size is reported as the number of parameters (Params, in millions). Detection performance is analyzed using precision (
P), recall (
R), average precision (AP), and mean average precision (mAP). These metrics are defined as
where
,
, and
denote true positives, false positives, and false negatives, respectively. The average precision for a single category is computed as the area under the precision–recall curve
and the mean average precision across
k categories is given by
To provide a more fine-grained assessment relevant to transmission-line inspection, we additionally report scale-sensitive metrics (, ) and category-specific results for the two most prevalent defect types: Polluted Insulator (jyzwh) and Damaged Pole (dgss). For these categories we present both recall and to highlight the model’s ability to both locate and precisely localize commonly occurring, safety-critical defects. Together, the described computational and detection metrics provide a balanced appraisal of accuracy, completeness, and operational efficiency for real-time UAV-based transmission-line inspection.
Finally, consistent with the above accuracy–efficiency analyses and evaluation protocol, we emphasize the intended deployment mode: post-mission processing of UAV imagery on a server (rather than on-board inference). Using the same input as in our comparisons, TinyDef-DETR sustains 152.19 FPS on a single NVIDIA RTX 3090 (24 GB)—that is, 152.19 images/s or 6.57 ms per image ( ms). This measured throughput is well aligned with the accuracy–complexity trade-offs reported above and is operationally meaningful: a typical batch of 1000 UAV images can be processed in approximately 6.57 s of pure inference time, readily meeting the requirements of server-side, after-mission defect screening for transmission-line inspection.
4.3. Effects of Kernel Size and Frequency-Domain Features on Detection Performance
We quantitatively assessed the impact of the convolutional kernel size in the Large-Receptive-Field branch of CSDMAM (
Section 3.4) by sweeping
.
Table 2 summarizes model complexity (Params, GFLOPs) and detection metrics (Precision, Recall,
,
,
,
), including the newly added configuration
.
A consistent trend emerges. Very small kernels () under-capture long-range structure: despite minimal complexity they yield markedly lower Recall and on elongated, line-like defects (Recall , ). Moving to moderate sizes () modestly improves contextual aggregation and occasionally Precision, but the gains in Recall and AP remain limited, indicating insufficient coverage of the anisotropic spatial extents typical of transmission-line defects.
Expanding to provides the best overall balance. This setting delivers the highest Recall () and the strongest () among all candidates, with clear improvements for small and medium objects (, ) at only a marginal computational increase (GFLOPs ). We attribute these gains to a receptive field that is large enough to aggregate global line-structure context while still preserving fine detail, aided by the depth-wise/separable parameterization and the frequency-guided FSCA path that stabilizes high-frequency cues without amplifying noise.
Further enlarging the kernel beyond this point does not help. At , Precision inches up (0.589) but Recall and small/medium APs drop; at , Precision peaks (0.627) yet Recall falls to and overall AP degrades (, ). We observe two coupled effects: (i) oversmoothing of fine structures caused by excessively wide spatial averaging, which suppresses weak, small targets emphasized by FSCA; and (ii) optimization and efficiency penalties—GFLOPs rise from 65.3 (at ) to 84.1 (at ), with diminishing returns and a measurable recall deficit on subtle defects.
Taken together, the ablation supports selecting for the Large-Receptive-Field branch of CSDMAM. This configuration yields the most favorable trade-off among global context capture, preservation of fine-grained defect morphology, and computational efficiency in UAV-based transmission-line inspection. Frequency-domain enhancement (FSCA) remains beneficial across settings, but its effect is maximized when paired with a moderately large, not extreme, spatial kernel—precisely the regime at .
To comprehensively assess the impact of the FSCA branch and the integration of frequency-domain information, we evaluate three variants of our framework: TinyDef-DETR (w/o), where the FSCA branch and all frequency-domain operations are removed; TinyDef-DETR (Id), where the FSCA branch is retained but replaced with a pure identity mapping that forwards features without any modulation or frequency-domain interaction; and TinyDef-DETR (Full), which incorporates the complete FSCA-based dual-domain enhancement. The TinyDef-DETR (Id) variant is deliberately constructed as a neutral-control configuration. It is analogous to a biomedical trial investigating the effect of a soft drink on human health, in which one group receives the soft drink while another receives an equivalent volume of plain water. In our setting, TinyDef-DETR (Id) maintains the architectural footprint and computational cost of the FSCA branch while eliminating its adaptive behavior, thereby enabling a clean isolation of the genuine contribution of FSCA-based modulation from the mere presence of an additional branch.
As reported in
Table 3, TinyDef-DETR (Full) achieves the most favorable trade-off between detection accuracy and computational complexity. Compared with TinyDef-DETR (w/o), it consistently improves both Precision and Recall, with
rising from 0.206 to 0.275 and
increasing from 0.0926 to 0.1187, while adding only negligible overhead in parameters and GFLOPs. Although TinyDef-DETR (Id) attains the highest Precision (0.597), its
(0.232) and
(0.1015) remain clearly below those of TinyDef-DETR (Full). This discrepancy indicates that simply introducing an identity shortcut under the same computational budget cannot explain the observed performance gains; instead, they stem from the adaptive dual-domain feature modulation enabled by the FSCA mechanism.
The per-class and scale-aware results in
Table 4 further reinforce this conclusion. For challenging categories such as
jyzwh and
dgss, TinyDef-DETR (Full) achieves higher Recall and
than both TinyDef-DETR (w/o) and TinyDef-DETR (Id), indicating enhanced robustness to subtle, low-contrast, and visually ambiguous defects. In addition, TinyDef-DETR (Full) obtains the highest
and
, confirming that the proposed dual-domain attention mechanism is effective in preserving fine-grained structural cues for small defects while simultaneously capturing richer contextual information for medium-scale targets. By contrast, TinyDef-DETR (Id) does not exhibit comparable improvements, which substantiates that the performance gains are attributed to the FSCA-based frequency-domain enhancement rather than to increased model capacity or superficial architectural modifications.
4.4. Comparison of Heatmaps
To intuitively demonstrate the effectiveness of the proposed modules, we generate heatmaps from the baseline RT-DETR model and the model with the proposed modules (EE-ResNet, SPD Convolution, and CSDMAM) on the same input image. These heatmaps illustrate the attention distribution of the models during the recognition process. As shown in
Figure 7, subfigures (a), (c), and (e) represent the attention maps from the baseline model corresponding to the modules before integration, while subfigures (b), (d), and (f) show the attention maps after incorporating the proposed modules. It can be observed that the attention distribution after the inclusion of the modules is more fine-grained, and the object contours are more distinct, indicating that the proposed modules enhance the model’s ability to preserve more detailed information. For better readability and to highlight the subtle differences, zoom-in views are provided, overlaid on the original images, allowing readers to clearly observe the enhanced attention patterns around key object regions.
4.5. Ablation Study of the Proposed Method
To analyze the impact of each component in CSG-ADCD, we conduct ablation studies by selectively enabling and disabling four key modules: EER (Edge-Enhanced ResNet), SPD (Stride-free Space-to-Depth), CSDMAM (Cross-Stage Dual-Domain Multi-Scale Attention Module), and FWS Loss (Focaler-Wise-SIoU Loss). The results in
Table 5 quantify each module’s contribution across multiple evaluation metrics. Here, ✓ denotes an enabled module, and × denotes a disabled one.
Edge-Enhanced ResNet (EER) EER strengthens boundary-aware features and reduces background leakage. Relative to the bare baseline, precision increases from 0.369 to 0.444, recall rises from 0.177 to 0.203, and improves from 0.163 to 0.189 at an unchanged 57.0 GFLOPs. These shifts indicate fewer false positives at a fixed confidence threshold and more true positives on low-contrast contours. Importantly for the perceived instability, EER conditions the feature space so that subsequent attention behaves reliably. When EER is combined with SPD and CSDMAM, recall increases further to 0.233 and for small objects increases from 0.083 with SPD alone to 0.119 with all three. This pattern shows that EER anchors CSDMAM to genuine defect boundaries, turning a potentially volatile attention stage into a consistent contributor to small-object localization.
Space-to-Depth Convolution (SPD) Applied alone, SPD preserves fine spatial evidence while avoiding aliasing, which suppresses spurious activations. Precision increases from 0.369 to 0.486, recall holds near 0.195, and stays around 0.190, with GFLOPs rising modestly from 57.0 to 59.9. When SPD is paired with EER but without attention, recall increases from 0.195 to 0.208 and nudges from 0.190 to 0.192, while precision softens from 0.486 to 0.417, which is expected because stronger edge cues surface more borderline candidates before a selective mechanism filters them. Once CSDMAM is added, those candidates are better gated: precision recovers from 0.417 to 0.485 and recall increases from 0.208 to 0.233. Thus, SPD and EER jointly deliver detail-rich, edge-faithful inputs that convert CSDMAM from a possible source of volatility into a consistent enhancer of small-object detection, as reflected by increasing from 0.083 to 0.119 when all three are enabled.
Cross-Stage Dual-Domain Multi-Scale Attention Module (CSDMAM) The single-module ablation clarifies why a reviewer might perceive instability if attention is used in isolation. Enabling CSDMAM alone increases compute from 57.0 to 72.8 GFLOPs yet does not improve end-to-end accuracy: precision moves from 0.369 to 0.458, recall decreases from 0.177 to 0.166, and remains near baseline at 0.165. This outcome indicates that, without edge priors or detail-preserving downsampling, attention can over-weight textured background regions and depress recall. Pairwise ablations corroborate this dependency: adding CSDMAM to SPD reduces precision from 0.486 to 0.385, and adding it to EER lowers precision from 0.444 to 0.425. In contrast, when CSDMAM is combined with both EER and SPD, its intended role materializes: increases from 0.190 with SPD alone to 0.237 with all three modules, and for small objects increases from 0.083 to 0.119. Hence, CSDMAM is not intrinsically unstable; it becomes reliably beneficial once the backbone provides clean edges and SPD preserves the fine-scale evidence that attention needs to gate.
Focaler-Wise-SIoU Loss (FWS Loss) FWS Loss converts representational gains into better calibration and tighter alignment. When applied to the EER–SPD–CSDMAM stack, precision increases from 0.485 to 0.534, recall increases from 0.233 to 0.263, and increases from 0.237 to 0.275. These simultaneous gains show improved confidence shaping and stricter localization. Two interactions clarify the trade-offs. First, relative to the bare baseline, for small objects increases from 0.071 to 0.106 under the full configuration, confirming stronger tiny-target localization on average. Second, relative to the three-module model without the loss, decreases from 0.119 to 0.106 even as global metrics improve. This pattern reflects FWS Loss emphasizing medium-difficulty samples and penalizing loose boxes, which can prune marginal small-object detections at the chosen thresholds. In practice, this behavior is tunable by slightly relaxing the classification prior for the smallest anchors or moderating the modulation strength in the loss. Overall, FWS Loss interacts constructively with EER, SPD, and CSDMAM—stabilizing predictions, curbing over-confident false positives, and reinforcing the complementary effects of the modules—thereby addressing the instability observation within a co-designed pipeline.
4.6. Comparisons with Previous Methods
In this section, we conduct a rigorous comparative study between the proposed TinyDef-DETR and representative state-of-the-art detectors, including one-stage models (e.g., YOLOv5/8/10/11), two-stage/cascade detectors, and Transformer-based frameworks (e.g., RT-DETR, Conditional-DETR, DINO). All models are evaluated under a unified protocol—identical data splits, input resolutions, and training hyperparameters—to ensure a fair and reliable comparison.
Before presenting the quantitative results, we perform a qualitative case study on four representative images from the CSG-ADCD dataset to visually compare TinyDef-DETR with competing detectors. Since most targets in this dataset are small objects densely distributed in local regions, each image is cropped to focus on the areas with high object concentration, thereby providing a clearer and more informative visualization. To further enhance readability under such crowded conditions, detection results are rendered as filled regions rather than simple bounding box outlines. As illustrated in
Figure 8, TinyDef-DETR yields more accurate and consistent localization of small and densely packed objects compared with DINO, RT-DETR-R18, and YOLOv5s, clearly demonstrating its superiority in challenging small-object detection scenarios.
Table 6 summarizes the overall performance of representative detectors on the CSG-ADCD dataset. The comparison covers lightweight and standard variants of YOLO (v5, v8, 10, 11), Transformer-based models such as RT-DETR and Conditional-DETR, as well as classical baselines including RetinaNet and DETR. Key indicators, including the number of parameters, computational complexity (GFLOPs), Precision, Recall,
, and
, are reported to provide a comprehensive view of accuracy–efficiency trade-offs. The best values in each column are highlighted in bold for clarity. Note that “YOLO 11s-P2” denotes a variant in which a P2 feature map was added to the model during training, and a dash (“-”) indicates that the corresponding measurement is not available.
For a fair comparison, all methods are evaluated under a unified protocol—an input resolution of , identical NMS settings, and the same maximum number of detections per image.
It is worth noting that DINO [
11]—evaluated under
identical hardware and software configurations using the official implementation with its default 12-epoch schedule—performs worse than TinyDef-DETR across all reported metrics, while requiring 79.8 h of training. In contrast, TinyDef-DETR, trained for 100 epochs under the same setup (following the canonical schedule adopted in RT-DETR [
12]), completes in under 20 h. Although wall-clock time is inherently environment-dependent, these numbers correspond to realistic, recommended settings. Importantly, the 12-epoch schedule for DINO and the 100-epoch schedule for TinyDef-DETR (inherited from RT-DETR) are the prescribed defaults; we therefore regard these schedules as intrinsic to each method and adopt them to compare truly “native” models. Under this protocol, TinyDef-DETR delivers a substantially more favorable accuracy–efficiency trade-off for real-world UAV inspection—amusingly, it seems this “longer training, better performance” effect doesn’t always make sense.
A comprehensive illustration of the trade-off between detection accuracy and computational efficiency is presented in
Figure 9, where the results reported in
Table 6 are visualized. The plot highlights the relative positions of different detectors in terms of
versus GFLOPs, enabling an intuitive comparison of the accuracy–efficiency balance across lightweight YOLO variants, Transformer-based detectors, and the proposed TinyDef-DETR. Notably, TinyDef-DETR achieves an
of 0.275 with a computational cost of only 65.3 GFLOPs, striking a favorable balance between accuracy and efficiency compared with both lightweight YOLO models and heavier Transformer-based detectors. It is worth mentioning that while YOLO 11s-P2, which adds a P2 feature map to the original YOLO 11s, shows a moderate improvement in precision and recall (e.g.,
increases from 0.155 to 0.181), this simple addition of a P2 layer alone does not lead to a substantial boost in overall detection efficiency or accuracy–efficiency trade-off. This suggests that architectural modifications need to be more carefully designed to achieve meaningful gains, particularly for small-object detection in UAV inspection scenarios.
Furthermore,
Table 7 provides a more detailed view of performance on small objects and defect-specific categories. It reports Recall and
for Polluted Insulator (jyzwh) and Damaged Pole (dgss)—the two most frequently occurring defect types in transmission line inspection—alongside scale-sensitive metrics
and
. These results offer deeper insights into model behavior and enable fine-grained performance comparisons across representative defect categories. Although the dataset contains only 13 large objects, which is insufficient for robust evaluation, they are still part of transmission line defects. Our model achieves an
of 0.454 for large objects, representing the average over IoU thresholds from 0.50 to 0.95, demonstrating its capability in detecting large-scale defects despite limited samples.
Finally, a holistic comparison of the selected detectors is presented in the form of a radar chart (
Figure 10). The models included—YOLO v5m, YOLO v8m, YOLO 11m, RT-DETR-R18, RT-DETR-x, and TinyDef-DETR—represent both one-stage and Transformer-based detection paradigms. The evaluation encompasses a comprehensive set of indicators, namely Recall,
,
,
,
, Recall of Polluted Insulator (jyzwh), mAP
50 of Polluted Insulator, and
of Damaged Pole (dgss). For enhanced visual comparability, each metric is normalized to its maximum value.
4.7. Generalization Experiments for the Proposed Methods
To further verify the generalization ability of the proposed method for small-object detection, we conduct experiments on VisDrone-DET 2019, a publicly available and generic UAV object detection benchmark rather than a domain-specific dataset. VisDrone-DET 2019 [
46] contains 8599 drone-captured images with over 540k annotated bounding boxes across ten general-purpose categories (e.g., pedestrian, person, car, van, bus, truck, motor, bicycle, awning-tricycle, tricycle), split into 6471/548/1580 for train/val/test. Crucially, VisDrone does not include transmission-line defect classes; thus, our results on VisDrone (
Table 8) explicitly measure cross-domain transferability rather than in-domain recognition. Owing to its dense object distributions, the high prevalence of small targets, and complex real-world scenes, VisDrone offers a rigorous testbed for evaluating robustness and generalization.
As shown in
Table 8, TinyDef-DETR attains leading overall performance with moderate model complexity. Concretely, it consistently outperforms strong baselines across standard detection metrics and shows clear advantages on small objects, while keeping computation and parameters at a practical level.
These results substantiate that the proposed framework generalizes effectively to small-object detection in UAV imagery beyond transmission-line inspection, indicating practical applicability across domains with different object semantics.
5. Discussion
5.1. Theoretical Mechanisms of Performance Enhancement
The performance gains observed with TinyDef-DETR arise from the mutually reinforcing effects of its architectural and loss-design innovations, each addressing a specific bottleneck in UAV-based defect detection. At the representational level, the edge-enhanced ResNet backbone strengthens the model’s sensitivity to fine structural cues by amplifying high-frequency components associated with defect boundaries. This modification counteracts the intrinsic attenuation of edge information caused by deep convolutional downsampling, enabling the model to preserve discriminative contours essential for identifying small or visually ambiguous defects. From a theoretical standpoint, such enhancement effectively increases the signal-to-noise ratio for fine-grained features and improves the separability of defect patterns within the feature manifold.
The stride-free space-to-depth (SPD) module introduces a second source of improvement by performing spatial reduction without discarding pixel-level detail. Unlike conventional stride-based downsampling, which induces aliasing and information loss, SPD reorganizes high-resolution inputs into channel-enriched tensors. This preserves local texture, minimizes feature degradation for small targets, and stabilizes gradient flow. Theoretically, this operation enhances the mutual information between shallow and deep features, enabling a more coherent propagation of small-object descriptors throughout the network hierarchy.
Furthermore, the cross-stage dual-domain multi-scale attention mechanism integrates spatial and frequency-domain representations to model global context and local variations more holistically. By enabling multi-scale interactions across network stages, this module addresses the scale variability characteristic of UAV imagery and improves the model’s ability to capture long-range structural dependencies. In essence, the mechanism expands the effective receptive field while simultaneously preserving the locality required for small-object detection, creating a balanced and robust feature fusion process.
Finally, the Focaler–Wise–SIoU loss contributes an optimization-level enhancement by modulating regression difficulty through an EMA-normalized difficulty indicator. The unimodal weighting function prioritizes moderately hard samples while attenuating extreme outliers and trivial cases, resulting in a smoother and more stable error landscape. This adaptive reweighting improves convergence behavior, enhances bounding-box localization for small defects, and mitigates gradient domination by noisy samples. From a theoretical perspective, the loss introduces a dynamic sample-importance prior that aligns training emphasis with statistically informative examples, thereby improving both robustness and generalization.
Collectively, these mechanisms form a coherent theoretical foundation explaining why TinyDef-DETR exhibits strong performance on small, ambiguous, and cluttered defect instances in UAV imagery. Each component contributes a targeted enhancement—representational fidelity, information preservation, multi-scale context integration, and difficulty-aware optimization—that together yield a substantial improvement over conventional detection pipelines.
5.2. Limitations and Future Directions
Despite its effectiveness, TinyDef-DETR still exhibits certain limitations that warrant further exploration. First, although SPD alleviates information loss during downsampling, the increased channel dimensionality demands additional memory and may introduce latency on resource-limited embedded UAV platforms. Lightweight kernel decomposition or dynamic channel pruning may help reduce overhead while preserving detail fidelity.
Second, the dual-domain attention module, while beneficial for multi-scale fusion, introduces additional complexity to the training pipeline. In scenarios involving extremely high-resolution imagery or large-scale datasets, the global context modeling may become computationally expensive. Future work may explore hierarchical or sparse attention variants to reduce computational burden while maintaining cross-stage consistency.
Third, the difficulty-modulated regression loss relies on reliable EMA statistics of IoU distributions. In datasets with extreme imbalance or rapidly shifting difficulty patterns, the moving average may adapt too slowly, reducing the responsiveness of the modulation mechanism. Developing more flexible or instance-adaptive normalization strategies could further stabilize optimization and improve performance on highly heterogeneous data.
Additionally, although the present study focuses on transmission line defect detection, the generalization of TinyDef-DETR to other aerial inspection domains—such as railway, bridge, and pipeline monitoring—remains to be systematically evaluated. Cross-domain validation, domain adaptation methods, or self-supervised pretraining may help broaden its applicability.
Finally, the current framework does not explicitly incorporate geometric priors, structural constraints, or transformer-based depth modeling that could further improve detection under severe occlusion or low-contrast imaging. Integrating 3D contextual cues, physical-structure modeling, or multi-modal inputs (e.g., infrared or LiDAR) could provide additional robustness in complex operational environments.
6. Conclusions
In this paper, we introduced TinyDef-DETR, a DETR-based framework tailored to defect detection in UAV imagery of transmission lines. The design combines an Edge-Enhanced ResNet backbone, a stride-free space-to-depth (SPD) downsampling module, a cross-stage dual-domain multi-scale attention mechanism, and a Focaler–Wise–SIoU regression loss. Together, these components target the core difficulties of this task—tiny object scale, visual ambiguity, and background clutter—by strengthening edge/detail representation, preserving pixel-level cues during downsampling, enhancing global–local feature interaction, and stabilizing box regression for small targets.
Comprehensive experiments on the CSG-ADCD dataset and real-world UAV imagery show that TinyDef-DETR consistently surpasses representative detectors in detection accuracy, recall, and robustness, while maintaining competitive computational efficiency. Ablation studies further confirm that each module contributes complementary gains to the final system, clarifying the role of edge-aware features, lossless downsampling, dual-domain attention, and shape/angle-aware localization in the overall performance.
These results indicate that TinyDef-DETR delivers strong detection capability for small and challenging defects and is well-suited for practical UAV-based inspection. Owing to its modular and lightweight design, the framework is readily adaptable to broader power-system inspection workflows and can be extended to other small-object detection problems in aerial and remote sensing applications.