arXiv:2511.09891cs.CVcs.AI2025-11

提升航拍图中微小目标检测效果,通过注意力机制和自适应损失实现

Scale-Aware Relay and Scale-Adaptive Loss for Tiny Object Detection in Aerial Images

  • 引入跨尺度注意力层增强特征表达,加强层间信息传递
  • 设计自适应损失函数,降低大目标惩罚权重,聚焦微小目标训练
  • 在多个数据集上显著提升检测性能,尤其在噪声环境下表现稳健

尽管物体检测技术进步显著,现有检测器在航拍图像中仍难以识别微小目标。主要原因在于微小目标特征有限,在深层网络传播中易被削弱或丢失;同时训练时小目标受的回归惩罚远高于大目标。为此,本文提出尺度感知中继层(SARL)与尺度自适应损失(SAL),均兼容当前主流检测框架。SARL利用跨尺度空间-通道注意力,逐层增强特征并促进层间共享;SAL重构基于IoU的损失函数,动态降低大目标权重,使训练更聚焦于微小目标。在AI-TOD、DOTA-v2.0和VisDrone2019三个基准测试上验证,嵌入YOLOv5(有锚点)与YOLOx(无锚点)后,平均精度(AP)提升5.5%。尤其在真实噪声数据集AI-TOD-v2.0上达到29.0%的AP,显著增强泛化与鲁棒性。

原文摘要 · Abstract (English)

Recently, despite the remarkable advancements in object detection, modern detectors still struggle to detect tiny objects in aerial images. One key reason is that tiny objects carry limited features that are inevitably degraded or lost during long-distance network propagation. Another is that smaller objects receive disproportionately greater regression penalties than larger ones during training. To tackle these issues, we propose a Scale-Aware Relay Layer (SARL) and a Scale-Adaptive Loss (SAL) for tiny object detection, both of which are seamlessly compatible with the top-performing frameworks. Specifically, SARL employs a cross-scale spatial-channel attention to progressively enrich the meaningful features of each layer and strengthen the cross-layer feature sharing. SAL reshapes the vanilla IoU-based losses so as to dynamically assign lower weights to larger objects. This loss is able to focus training on tiny objects while reducing the influence on large objects. Extensive experiments are conducted on three benchmarks (\textit{i.e.,} AI-TOD, DOTA-v2.0 and VisDrone2019), and the results demonstrate that the proposed method boosts the generalization ability by 5.5\% Average Precision (AP) when embedded in YOLOv5 (anchor-based) and YOLOx (anchor-free) baselines. Moreover, it also promotes the robust performance with 29.0\% AP on the real-world noisy dataset (\textit{i.e.,} AI-TOD-v2.0).

目标检测航拍图像微小目标自适应损失

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。