用语义+几何联合优化,让目标框旋转更稳定准确。
Geometry Meets Semantics: Fractional Gradient Stabilization for Semantic-Driven Bounding Box Optimization in Visual Detection Tasks

- 引入语义相似度引导的损失函数,融合视觉语义信息
- 提出分数阶梯度机制,解决旋转框优化中的梯度震荡问题
- 适用于各类检测任务,尤其在复杂方向目标上表现更优
边界框是视觉检测中目标定位的基础。其中,方向性边界框能提供更精确的方向表征,被广泛使用。现有基于IoU的损失函数虽普遍用于框回归优化,但存在两大缺陷:一是仅依赖几何属性,忽略语义信息;二是方向优化时梯度不稳定,导致方向收敛振荡。本文提出分数阶语义IoU损失(FrSIoU),实现语义-几何统一学习并稳定梯度。首先设计语义相似度度量,构建自适应梯度门控的语义IoU损失(SIoU);其次重新审视方向框优化中的梯度不稳定性,将SIoU扩展为分数阶形式,通过累积历史IoU状态来正则化异常梯度。大量实验表明,该方法在不同框表示形式和多样化检测任务中均实现稳定性能提升。代码将开源于GitHub。
原文摘要 · Abstract (English)
Bounding boxes are fundamental for object localization in visual detection tasks. Among them, oriented bounding boxes are widely used in visual detection tasks, which provide a more precise directional representation. Generally, IoU-based losses are widely adopted to optimize box regression. However, we observed that IoU-driven box optimization suffers from two key issues: (1) it relies solely on geometric properties while ignoring semantic cues; (2) orientation optimization suffers from unstable gradients, causing oscillations in orientation convergence. In this paper, we propose a Fractional Semantic IoU loss to achieve unified semantic-geometric learning with gradient stabilization. First, we design a semantic similarity metric to guide IoU optimization, building a Semantic IoU loss (SIoU loss) with an adaptive gradient gating mechanism. Then, we revisit the gradient instability issue in oriented box optimization and extend the SIoU loss to a fractional-order formulation to build the \textbf{Fr}actional \textbf{S}emantic \textbf{IoU} \textbf{loss} (FrSIoU loss). The FrSIoU loss accumulates historical IoU states to regularize abnormal gradients during bounding box optimization process. Extensive experiments demonstrate that our approach achieves stable performance gains across different bounding box formulations and diverse visual detection tasks. The code will be available on GitHub.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。