arXiv:2510.27649cs.CV2025-10中稿 · the GRSL in 2025被引 10

提出GCD度量方法,提升小目标检测精度与模型收敛速度。

Gaussian Combined Distance: A Generic Metric for Object Detection

  • 结合高斯分布建模框位置与尺度,设计新距离度量GCD。
  • 在AI-TOD-v2上实现小目标检测最佳性能,比Wasserstein提升1.8%。
  • 适合需要高精度定位的小目标检测场景,如遥感、医疗图像。

在目标检测中,良好的相似性度量可显著提升模型性能。当前主流使用基于IoU的度量,但在检测小目标时因对位置微偏敏感而表现不佳。已有研究尝试用Wasserstein距离替代IoU,用于高斯分布边界框的相似性度量,但其缺乏尺度不变性,影响模型泛化能力;且作为损失函数时,中心属性独立优化导致收敛慢、精度不足。为此,本文提出高斯组合距离(Gaussian Combined Distance, GCD)。通过分析GCD及其梯度,证明其兼具尺度不变性与联合优化特性,提升定位性能。在AI-TOD-v2数据集上的大量实验表明,将GCD作为边界框回归损失和标签分配度量,可在多种检测器上达到领先水平。进一步在MS-COCO-2017和Visdrone-2019数据集上验证其泛化能力,相比Wasserstein距离在不同尺度下均表现更优。代码已开源:https://github.com/MArKkwanGuan/mmdet-GCD。

原文摘要 · Abstract (English)

In object detection, a well-defined similarity metric can significantly enhance model performance. Currently, the IoU-based similarity metric is the most commonly preferred choice for detectors. However, detectors using IoU as a similarity metric often perform poorly when detecting small objects because of their sensitivity to minor positional deviations. To address this issue, recent studies have proposed the Wasserstein Distance as an alternative to IoU for measuring the similarity of Gaussian-distributed bounding boxes. However, we have observed that the Wasserstein Distance lacks scale invariance, which negatively impacts the model's generalization capability. Additionally, when used as a loss function, its independent optimization of the center attributes leads to slow model convergence and unsatisfactory detection precision. To address these challenges, we introduce the Gaussian Combined Distance (GCD). Through analytical examination of GCD and its gradient, we demonstrate that GCD not only possesses scale invariance but also facilitates joint optimization, which enhances model localization performance. Extensive experiments on the AI-TOD-v2 dataset for tiny object detection show that GCD, as a bounding box regression loss function and label assignment metric, achieves state-of-the-art performance across various detectors. We further validated the generalizability of GCD on the MS-COCO-2017 and Visdrone-2019 datasets, where it outperforms the Wasserstein Distance across diverse scales of datasets. Code is available at https://github.com/MArKkwanGuan/mmdet-GCD.

目标检测小目标度量学习高斯分布

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