arXiv:2507.12420cs.CV2025-07被引 4

用插值箱替代人工几何惩罚,提升目标检测定位精度。

InterpIoU: Rethinking Bounding Box Regression with Interpolation-Based IoU Optimization

  • 用预测框与真实框的插值中间框计算IoU,替代传统手工设计的惩罚项。
  • 在无重叠情况下仍能提供有效梯度,避免小目标误放大问题。
  • 适用于各类检测框架,尤其改善小目标检测性能,适合做检测器优化者参考。

边界框回归(BBR)是目标检测的核心,回归损失对定位精度至关重要。现有基于IoU的损失函数常引入手工设计的几何惩罚项来解决非重叠情况下的不可导性并提升性能,但这些惩罚项对框的形状、大小和分布敏感,导致小目标优化不佳,甚至引发框异常放大等问题。为此,我们提出InterpIoU,一种新损失函数,以插值框与真实框之间的IoU作为替代项,取代手工惩罚。通过插值框弥合预测与真实框的差距,InterpIoU在非重叠情况下仍能提供有意义的梯度,并天然规避由惩罚项不匹配导致的框放大问题。仿真结果表明,IoU本身即为理想回归目标,而现有几何惩罚既不必要也非最优。在此基础上,我们进一步提出动态InterpIoU,根据IoU值自适应调整插值系数,增强对多样对象分布的适应性。在COCO、VisDrone和PASCAL VOC上的实验显示,所提方法在多种检测框架中持续优于当前最优的基于IoU的损失,尤其在小目标检测上表现显著提升,验证了其有效性。

原文摘要 · Abstract (English)

Bounding box regression (BBR) is fundamental to object detection, where the regression loss is crucial for accurate localization. Existing IoU-based losses often incorporate handcrafted geometric penalties to address IoU's non-differentiability in non-overlapping cases and enhance BBR performance. However, these penalties are sensitive to box shape, size, and distribution, often leading to suboptimal optimization for small objects and undesired behaviors such as bounding box enlargement due to misalignment with the IoU objective. To address these limitations, we propose InterpIoU, a novel loss function that replaces handcrafted geometric penalties with a term based on the IoU between interpolated boxes and the target. By using interpolated boxes to bridge the gap between predictions and ground truth, InterpIoU provides meaningful gradients in non-overlapping cases and inherently avoids the box enlargement issue caused by misaligned penalties. Simulation results further show that IoU itself serves as an ideal regression target, while existing geometric penalties are both unnecessary and suboptimal. Building on InterpIoU, we introduce Dynamic InterpIoU, which dynamically adjusts interpolation coefficients based on IoU values, enhancing adaptability to scenarios with diverse object distributions. Experiments on COCO, VisDrone, and PASCAL VOC show that our methods consistently outperform state-of-the-art IoU-based losses across various detection frameworks, with particularly notable improvements in small object detection, confirming their effectiveness.

目标检测边界框回归IoU优化小目标检测

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