通过双域特征融合与知识蒸馏,提升小目标检测精度与效率
SO-DETR: Leveraging Dual-Domain Features and Knowledge Distillation for Small Object Detection
- 采用空间-频率双域编码器,融合多尺度特征增强细节表达
- 动态选择高分锚框初始化查询,优化小目标检测资源分配
- 轻量骨干搭配知识蒸馏,在低算力下实现领先性能
基于检测的Transformer方法在通用目标检测中取得显著进展,但在小目标检测方面仍面临挑战。主要难点在于现有编码器难以高效融合低层特征,且查询选择策略未针对小目标优化。为此,本文提出一种高效模型SO-DETR,包含三个核心组件:双域混合编码器、增强查询选择机制和知识蒸馏策略。双域混合编码器融合空间与频域信息,有效整合多尺度特征,增强高分辨率特征表征,同时保持较低计算开销。增强查询选择机制通过扩展交并比(expanded IoU)动态选取高分锚框初始化查询,优化查询资源分配。结合轻量级骨干网络与知识蒸馏策略,构建出面向小目标的高效检测器。在VisDrone-2019-DET与UAVVaste数据集上的实验结果表明,SO-DETR在相似计算成本下优于现有方法。
原文摘要 · Abstract (English)
Detection Transformer-based methods have achieved significant advancements in general object detection. However, challenges remain in effectively detecting small objects. One key difficulty is that existing encoders struggle to efficiently fuse low-level features. Additionally, the query selection strategies are not effectively tailored for small objects. To address these challenges, this paper proposes an efficient model, Small Object Detection Transformer (SO-DETR). The model comprises three key components: a dual-domain hybrid encoder, an enhanced query selection mechanism, and a knowledge distillation strategy. The dual-domain hybrid encoder integrates spatial and frequency domains to fuse multi-scale features effectively. This approach enhances the representation of high-resolution features while maintaining relatively low computational overhead. The enhanced query selection mechanism optimizes query initialization by dynamically selecting high-scoring anchor boxes using expanded IoU, thereby improving the allocation of query resources. Furthermore, by incorporating a lightweight backbone network and implementing a knowledge distillation strategy, we develop an efficient detector for small objects. Experimental results on the VisDrone-2019-DET and UAVVaste datasets demonstrate that SO-DETR outperforms existing methods with similar computational demands. The project page is available at https://github.com/ValiantDiligent/SO_DETR.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。