链接:https://pan.baidu.com/s/1q9_lKpW1pu6fl6OQAbwdSw
提取码:yyds
论文地址:2311.02877.pdf (arxiv.org)
https://download.csdn.net/download/weixin_53660567/88588052
Inner-IoU: More Effective Intersection over Union Loss with Auxiliary Bounding Box
随着检测器的快速发展,边界框回归(BBR)损失函数不断进行更新和优化。然而,现有的IoU基于BBR仍然集中在通过添加新损失项来加速收敛,忽略了IoU损失项本身的局限性。尽管从理论上讲,IoU损失可以有效地描述边界框回归的状态,但在实际应用中,它无法根据不同的检测器和检测任务进行自适应调整,且不具备较强的泛化能力。基于上述情况,我们首先分析了BBR模型,并得出结论:区分不同的回归样本以及使用不同尺度的辅助边界框计算损失可以有效加速边界框回归过程。对于高IoU样本,使用较小的辅助边界框计算损失可以加速收敛,而较大的辅助边界框适合于低IoU样本。然后,我们提出了Inner-IoU损失,通过辅助边界框计算IoU损失。对于不同的数据集和检测器,我们引入一个缩放因子比来控制计算损失的辅助边界框的尺度大小。最后,将Inner-IoU集成到现有的IoU基于损失函数中进行仿真和对比实验。实验结果表明,利用本文提出的方法进一步提高了检测性能,验证了Inner-IoU损失的有效性和泛化能力。
1. 简介
目标检测是计算机视觉中的基本任务,包括目标分类和定位。边界框回归损失函数是检测器定位分支的重要组成部分,检测器的定位精度在很大程度上取决于边界框回归,在当前的检测器中发挥着不可替代的作用。在BBR中,IoU损失可以准确描述预测边界框与GT框之间匹配的程度,确保模型在训练过程中可以学习到目标的位置信息。作为现有主流边界框回归损失函数的基本部分,IoU定义如下:
B和分别代表预测框和GT框。定义IoU之后,相应的损失可以定义如下:
至今,基于IoU的损失函数逐渐成为主流并占据主导地位。大多数现有方法基于IoU并进一步添加新的损失项。例如,为了在Anchor框与GT框重叠区域为0时解决梯度消失问题,提出了GIoU。GIoU定义如下,其中C是覆盖B和的最小框:
与GIoU相比,DIoU函数在IoU的基础上添加了一个新的距离损失项,主要通过最小化两个边界框中心点之间的归一化距离来达到更快收敛和更好的性能。它表示如下:
其中b和分别是B和的中心点,ρ(·)指的是欧氏距离,是最小边界框的斜边长。
CIoU进一步考虑了形状损失,并在DIoU损失的基础上添加了形状损失项。它表示如下:
其中α是正交平衡参数:
其中v衡量纵横比的一致性:
wgt和hgt分别表示目标框的宽度和高度,w和h分别表示预测框的宽度和高度。当目标框和预测框的纵横比相同时,CIoU将退化为DIoU。
与DIoU相比,EIoU直接最小化目标框和Anchor框的宽度(w,wgt)和高度(h,hgt)以及中心位置的归一化差值。EIoU定义如下:
wc和hc分别是覆盖目标框和预测框的最小边界框的宽度和高度。
最近的SIoU在考虑了Anchor框与GT框之间角度对边界框回归影响的基础上,将角度损失引入边界框回归损失函数。它定义如下:
角度损失表示GT框和Anchor框中心点连接之间的最小角度:
这一项旨在将Anchor框移动到最近的坐标轴,并根据角度变化优先考虑接近X轴或Y轴。当角度值为45°时,Λ=1。当中心点沿X轴或Y轴对齐时,Λ=0。
在考虑角度成本后,重新定义距离损失如下:
形状损失主要描述GT框和Anchor框之间的大小差异,定义如下:
θ的值决定了形状成本的重要性。这个参数的范围从2到6。
尽管上述边界框回归损失函数可以通过向IoU损失函数添加新的几何约束来加速收敛和改进检测性能,但它们并没有考虑IoU损失本身的合理性,这决定了检测结果的质量。为了弥补这一不足,我们提出了Inner-IoU损失,通过使用辅助边界框来加速回归,而无需添加任何新的损失项。
本文的主要贡献如下:
- 我们分析了边界框回归的过程和模式,并根据边界框回归问题的固有特性,提出在模型训练过程中使用较小的辅助边界框计算损失,以对高IoU样本的回归产生正向效果,而低IoU样本产生相反效果。
- 我们提出了Inner-IoU损失,通过引入比例因子控制来生成不同尺度的辅助边界框以计算损失。将其应用于现有的IoU基于损失函数可以实现更快和更有效的回归结果。
- 我们进行了一系列仿真和比较实验,实验结果表明,我们的方法在检测性能和泛化方面优于现有方法,对于不同像素大小的数据集,它实现了SOTA。
2. 相关工作
A. 目标检测
对于目标检测,根据是否产生锚点可分为基于锚点的检测算法和无锚点的检测算法。基于锚点的算法包括Faster R-CNN[14]、YOLO (You Only Look Once)系列[10]-[13]、SSD (Single Shot MultiBox Detector)[15]和RetinaNet[16]无锚点检测算法包括CornerNet[17]、CenterNet[18]和FCOS (Fully Convolutional One Stage Object detection)[19]。在这些目标检测算法中,边界盒回归损失函数起着至关重要的作用,它使检测器能够准确定位目标,提高了检测算法的检测精度。
B. BBR损失函数
首先提出对边界框尺度变化非常敏感的n范数损失[14]作为边界框回归损失。随后,为了弥补这一不足,提出IoU损失[1]来代替ln-norm loss,与ln-norm loss相比,IoU损失[1]预测框的回归结果更加准确。然而,IoU损失不能解决非重叠样本的梯度消失问题,GIoU损失[2]通过引入最小边界框来弥补这一缺陷。DIoU损失[3]增加了距离约束,将预测框中心点与GT框中心点之间的归一化距离作为新的损失项加入到IoU损失中,提高了收敛速度和定位精度。CIoU损失[3]进一步考虑形状相似度对边界框回归的影响,并在DIoU损失中加入形状损失项。EIoU损耗[4]利用focal loss解决了训练过程中样本不平衡的问题,并重新定义了形状损失,进一步提高了检测效果。最新的SIoU[5]损失在边界框回归损失的基础上增加了预测框与GT框之间的夹角作为新的约束,得到了最快的收敛结果。通过与上述算法的比较,我们提出了Inner-IoU损失可以进一步提高收敛速度。
3. 方法
A. 边界框回归模式分析
IoU损失函数在计算机视觉任务中有广泛的应用。在边界框回归过程中,不仅可以评估回归状态,还可以通过计算回归损失来加速收敛。在这里,我们讨论IoU变化与边界框大小之间的关系,分析边界框回归问题的本质特征,并解释本文提出方法的可行性。
如图3所示,图3a显示了IoU偏差曲线,水平轴和垂直轴分别表示偏差和IoU值。三种不同颜色的曲线对应不同尺度边界框的IoU变化曲线。A、B、C、D和E分别表示Anchor框和GT框的5种不同位置关系,其中红色边界框表示长度和宽度为10的Anchor框,对应的GT框用黑色边界框表示。
图3:不同尺度边界框的回归变化曲线(a) IoU-Deviation Curve(b) ABS(Grad)-Deviation Curve
图3b显示了ABS(Grad)偏差曲线。与图3a不同,图3b的垂直轴表示IoU偏差的绝对值。我们假设实际边界框大小为10,使用大小为8和12的边界框作为辅助边界框。在图3中,A和E对应低IoU样本的回归状态,而B和D对应高IoU样本的回归状态。从图3可以得出以下结论:
- 由于辅助边界框与实际边界框之间的尺度差异,回归过程中IoU值的变化趋势与实际边界框IoU值的变化趋势一致,可以反映实际边界框回归结果的质量。
- 对于高IoU样本,较小尺度辅助边界框的IoU梯度的绝对值大于实际边界框IoU梯度的绝对值。
- 对于低IoU样本,较大尺度辅助边界框的IoU梯度的绝对值大于实际边界框IoU梯度的绝对值。
基于以上分析,使用较小尺度辅助边界框计算IoU损失有助于提升高IoU样本的回归速度并加速收敛。相反,使用较大尺度辅助边界框计算IoU损失可以加速低IoU样本的回归过程。
B. Inner-IoU损失
为弥补现有IoU损失函数在不同的检测任务中的泛化能力较弱且收敛速度较慢的不足,我们提出使用辅助边界框计算损失以加速边界框回归过程。在Inner-IoU中,我们引入了尺度因子比,可以控制辅助边界框的尺度大小。通过为不同数据集和检测器使用不同尺度的辅助边界框,可以克服现有方法在泛化能力方面的局限。
GT框和Anchor框分别表示为Bgt和B,如图1所示。GT框的中心点和内GT框的中心点用(xcgt,ycgt)
表示,而(xc,yc)
代表Anchor框和内Anchor框的中心点。GT框的宽度和高度分别用wgt和hgt表示,而Anchor框的宽度和高度用w和h表示。变量“ratio”对应缩放因子,通常在[0.5,1.5]的范围内。
Inner-IoU损失继承了IoU损失的一些特征,同时具有自己的特点。Inner-IoU损失的范围与IoU损失相同,为[0,1]。由于辅助边界框与实际边界框之间仅存在尺度差异,损失函数的计算方法相同,Inner-IoU偏差曲线与IoU偏差曲线相似。
与IoU损失相比,当比例小于1且辅助边界框尺寸小于实际边界框时,回归的有效范围小于IoU损失,但梯度的绝对值大于从IoU损失获得的梯度,可以加速高IoU样本的收敛。相反,当比例大于1时,较大尺度的辅助边界框扩展了回归的有效范围,对低IoU样本的回归具有增强效果。将Inner-IoU损失应用于现有的基于IoU的边界框回归损失函数,如LInner-IoU、LInner-GIoU、LInner-DIoU、LInner-CIoU、LInner-EIoU和LInner-SIoU,如下所示:
4. 实验
A. 仿真实验
如图5所示,本文通过模拟实验分析了两种不同场景下的边界框回归过程。在图5a和图5b中,设置了7个不同的绿色边界框作为目标框,目标框的中心点设置为(100,100),比例分别为1:4、1:3、1:2、1:1、2:1、3:1和4:1。在图5a中,Anchor框随机分配2000个点,其位置分布以(100,100)为中心,半径为3。对于每个点的尺度,Anchor框的面积设置为0.5、0.67、0.75、1、1.33、1.5和2。
图5:蓝色点表示锚点,绿色边界框表示目标框的不同大小(a)高IoU回归样本(b)低IoU回归样本
对于给定的点和尺度,适应7个aspect ratios,即遵循与目标框相同的目标设置(即1:4、1:3、1:2、1:1、2:1、3:1和4:1)。图5b的Anchor框分布不同于图5a,其位置分布以(100,100)为中心,半径为6至9。尺寸和比例与图5a相同。总之,在每个实验中,应将2000×7×7个Anchor框拟合到每个目标框。
因此,总共有686,000=7×7×7×2,000种压缩情况。模拟实验的结果如图7所示,其中图7a表示高IoU回归样本场景下的收敛结果。为了加速高IoU样本的回归,尺度因子比例设置为0.8。图7b表示低IoU回归样本场景下的收敛结果,比例设置为1.2。可以看出,图中表示作者方法的虚线收敛速度优于现有方法。
B.对比实验
PASCALVOC上的YOLOv7
本实验对比了CIoU方法和SIoU方法,使用YOLOV7-tiny作为检测器,VOC2007 trainval和VOC2012 trainval作为训练集,VOC2007test作为测试集。训练集包含16551张图像,而测试集包含4952张具有20个类别的图像。作者在训练集上训练了150个Epoch,以展示本文方法的优势。
本文可视化了所提出方法与原始方法的训练过程,如图8所示。图8a、图8b和图8c分别显示了CIoU和Inner-CIoU的训练过程曲线,相应比例分别为0.7、0.75和0.8。图8d、图8e和图8f分别为SIoU和Inner-SIoU在比例为0.7、0.75和0.8时的训练过程曲线。
在图8中,橙色曲线代表本篇论文提出的方法,而现有方法用绿色曲线表示。可以看出,在50到150个Epoch的训练过程中,本文提出的方法优于现有方法。
图8:不同比例下SIoU和CIoU损失表现
对比实验在测试集上的结果如表1所示。可以看出,应用本文方法后,检测效果得到了提高,AP50和mAP50:95均提高了0.5%以上.图2和图6展示了检测样本的比较。从图中可以看出,与现有方法相比,所提出的方法定位更准确,误检和漏检较少。
表1:CIoU和SIoU损失的表现(比率在0.7和0.8之间)
图6:使用LSIoU和LInner-SIoU的YOLO7-tiny在PASCALVOC 2007测试集上的检测示例。从左到右分别代表SIoU方法,Inner-SIoU(ratio=0.7),Inner-SIoU(ratio=0.75)和Inner-SIoU(ratio=0.8)。
YOLOv5在AI-TOD上
为了证明所提出方法的可泛化性,本文在AI-TOD数据集上进行了比较实验,使用SIoU作为比较方法。
AI-TOD包括28036张空中图像,8种目标类型,以及700621个目标实例,其中14018张图像作为训练集,其余14018张图像作为测试集。与现有的目标检测任务数据集相比,AITOD的平均大小为12.8像素,远小于其他数据集。实验结果如表II所示。
在对比实验1中,通过将比例值在0.7和0.8之间设置为小于1,生成了一个比实际边界框小的辅助边界。实验结果表明,它可以提升高IoU样本的收益。在实验2中,当比例值大于1时,通过生成更大的辅助边界框来加速低IoU样本的收敛。
图4:LSIoU(第一行)和LInner-SIoU(第二行)使用YOLOV5s在AI-TOD测试集上的检测样例
5. 结论
在本文中,我们分析了边界框回归过程,指出了IoU损失的局限性,它对不同的检测任务没有很强的泛化能力。基于边界框回归问题的固有特点,提出了一种基于辅助边界框的边界框回归损失Inner-IoU。通过比例因子比控制辅助边界框的生成,计算损失,加速收敛。它可以集成到现有的基于IoU的损失函数中。通过一系列的仿真和消融实验,验证了该方法优于现有方法。值得注意的是,本文提出的方法不仅适用于一般的检测任务,而且对于非常小目标的检测任务也表现良好,证实了该方法的泛化性。