边界框回归的魔法:揭秘精准高效的MPDIoU损失函数

文章目录

  • 摘要
  • 1、简介
  • 2、相关工作
    • 2.1、目标检测和实例分割
    • 2.2. 场景文本识别
    • 2.3、边界框回归的损失函数
  • 3、点距最小的并集交点
  • 4、实验结果
    • 4.1、 实验设置
    • 4.2、数据集
    • 4.3、 评估协议
    • 4.4、 目标检测的实验结果
    • 4.5、 字符级场景文本识别的实验结果
    • 4.6、 实例分割的实验结果
  • 5、 结论

在这里插入图片描述

摘要

https://arxiv.org/pdf/2307.07662.pdf
边界框回归(BBR)已广泛应用于目标检测和实例分割,这是目标定位的重要步骤。然而,现有的大多数边界框回归损失函数在预测框与地面真相框具有相同长宽比但宽度和高度值完全不同的情况下无法优化。为了解决上述问题,我们充分探索了水平矩形的几何特征,并基于最小点距离提出了一种新颖的边界框相似性比较度量MPDIoU,该度量包含了现有损失函数中考虑的所有相关因素,即重叠或非重叠区域、中心点距离以及宽度和高度偏差,同时简化了计算过程。在此基础上,我们提出了基于MPDIoU的边界框回归损失函数 L MPDIoU \text{L}_{\text{MPDIoU}} LMPDIoU。实验结果表明,将MPDIoU损失函数应用于在PASCAL VOC、MS COCO和IIIT5k上训练的现有最先进实例分割(例如YOLACT)和目标检测(例如YOLOv7)模型,其性能优于现有损失函数。

关键词:目标检测、实例分割、边界框回归、损失函数

1、简介

目标检测和实例分割是计算机视觉领域的两个重要问题,在过去的几年里吸引了大量研究者的兴趣。大多数最先进的目标检测器(例如,YOLO系列[1,2,3,4,5,6],Mask R-CNN[7],Dynamic R-CNN[8]和DETR[9])都依赖于边界框回归(BBR)模块来确定物体的位置。基于这个范式,一个精心设计的损失函数对于BBR的成功至关重要。到目前为止,大多数现有的BBR损失函数分为两类: ℓ n \ell_n n范数损失函数和交并比(IoU)损失函数。

然而,现有的大多数用于边界框回归的损失函数在相同预测结果下具有相同的值,这降低了边界框回归的收敛速度和准确性。因此,考虑到现有用于边界框回归的损失函数的优点和缺点,受水平矩形几何特征的启发,我们尝试设计一种基于最小点距离的用于边界框回归的新型损失函数LMPDIoU,并在边界框回归过程中使用MPDIoU作为一种新措施来比较预测边界框与地面真实边界框之间的相似性。我们还提供了一种易于实现的解决方案,用于计算两个轴对齐矩形之间的MPDIoU,使其可以作为评估指标将MPDIoU纳入最先进的目标检测和实例分割算法中,并且在PASCAL VOC [10],MS COCO [11],IIIT5k [12]和MTHv2 [13]等一些主流目标检测,场景文本检测和实例分割数据集上进行了测试,以验证我们提出的MPDIoU的性能。

本文的贡献可以概括如下:

  • 1.我们考虑了现有IoU损失和 ℓ n \ell_n n范数损失的优点和缺点,然后提出了基于最小点距离的IoU损失,称为 L M P D I o U \mathcal{L}_{MPD IoU} LMPDIoU,以解决现有损失的问题,并获得更快的收敛速度和更准确的回归结果。
  • 2.在目标检测、字符级场景文本检测和实例分割任务上进行了广泛的实验。优秀的实验结果验证了所提出的MPD IoU损失的优越性。详细的消融研究展示了不同损失函数和参数值的设置效果。

2、相关工作

2.1、目标检测和实例分割

在过去的几年中,来自不同国家和地区的许多研究人员提出了大量的基于深度学习的目标检测和实例分割方法。总的来说,边界框回归已被许多代表性的目标检测和实例分割框架采用作为基本组件[14]。在用于目标检测的深度模型中,R-CNN系列[15][16][17]采用两个或三个边界框回归模块来获得更高的定位精度,而YOLO系列[2][3][6]和SSD系列[18][19][20]采用一个来获得更快的推理。RepPoints[21]预测几个点来定义一个矩形框。FCOS[22]通过预测采样点到地面真实边界框的顶部、底部、左侧和右侧的欧几里得距离来定位对象。

例如,对于实例分割,PolarMask [23]在n个方向上预测从采样点到物体边缘的射线的长度,以分割实例。还有其他检测器,例如RRPN [24]和R2CNN [25],通过添加旋转角度回归以检测任意定向的目标,用于遥感检测和场景文本检测。Mask R-CNN [7]在Faster R-CNN [15]上添加了一个额外的实例掩码分支,而最近的state-of-the-art YOLACT [26]在RetinaNet [27]上也做了同样的事情。总之,边界框回归是用于目标检测和实例分割的最新深度模型的关键组件之一。

2.2. 场景文本识别

为了解决任意形状场景文本检测和识别的问题,ABCNet[28]及其改进版本ABCNet v2[29]使用BezierAlign将任意形状的文本转换为规则形状。这些方法通过使用校正模块将检测和识别统一为端到端可训练的系统,取得了很大进展。[30]提出RoI Masking以提取任意形状文本识别的特征。类似于[30,31]尝试使用更快的检测器进行场景文本检测。AE TextSpotter[32]使用识别结果通过语言模型来指导检测。受[33]的启发,[34]提出了一种基于变换器的场景文本检测方法,该方法提供了实例级文本分割结果。

2.3、边界框回归的损失函数

边界框回归的魔法:揭秘精准高效的MPDIoU损失函数_第1张图片

起初, ℓ n \ell_{n} n范数损失函数被广泛用于边界框回归,它非常简单但对各种尺度都很敏感。在YOLO v1[35]中,对w和h采用平方根以减轻这种影响,而YOLO v3[2]使用 2 × w h 2 \times wh 2×wh。为了更好地计算地面真实边界框和预测边界框之间的差异,自Unitbox[36]开始使用IoU损失。为了确保训练稳定性,Bounded-IoU损失[37]引入了IoU的上界。对于训练目标检测和实例分割的深度模型,建议使用基于IoU的度量标准,比 ℓ n \ell_{n} n范数更一致[38,37,39]。原始的IoU表示预测边界框和地面真实边界框的交集面积和并集面积的比率(如图1(a)所示),可以表示为
I o U = B g t ∩ B p r d B g t ∪ B p r d , (1) I o U=\frac{\mathcal{B}_{g t} \cap \mathcal{B}_{p r d}}{\mathcal{B}_{g t} \cup \mathcal{B}_{p r d}}, \tag{1} IoU=BgtBprdBgtBprd,(1)
其中, B g t \text{B}_{gt} Bgt表示groundtruth边界框, B p r d \text{B}_{prd} Bprd表示预测边界框。可以看到,原始IoU只计算两个边界框的并集面积,无法区分两个框不重叠的情况。如方程1所示,如果 ∣ B g t ∩ B p r d ∣ = 0 \left|\text{B}_{gt} \cap \text{B}_{prd}\right|=0 BgtBprd=0,则 I o U ( B g t , B p r d ) = 0 IoU(\text{B}_{gt}, \text{B}_{prd})=0 IoU(Bgt,Bprd)=0。在这种情况下,IoU无法反映两个框是彼此接近还是相距很远。接着,为了解决这个问题,提出了GIoU [39]。GIoU可以表示为
G I o U = I o U − ∣ C − B g t ∪ B p r d ∣ ∣ C ∣ , (2) G I o U=I o U-\frac{\left|C-\mathcal{B}_{g t} \cup \mathcal{B}_{p r d}\right|}{|C|}, \tag{2} GIoU=IoUCCBgtBprd,(2)
其中,C表示覆盖 B g t \text{B}_{gt} Bgt B p r d \text{B}_{prd} Bprd的最小框(如图1(a)中的黑色虚框所示), ∣ C ∣ |C| C表示框C的面积。由于GIoU损失中引入了惩罚项,因此即使在不重叠的情况下,预测框也会向目标框移动。GIoU损失已被应用于训练最先进的目标检测器,例如YOLO v3和Faster R-CNN,并且取得了比MSE损失和IoU损失更好的性能。然而,当预测边界框完全被groundtruth边界框覆盖时,GIoU会失去效果。为了解决这个问题,考虑到预测边界框和groundtruth边界框之间的中心点距离,提出了DIoU[40]。DIoU的公式可以表示为
D I o U = I o U − ρ 2 ( B g t , B p r d ) C 2 (3) D I o U=I o U-\frac{\rho^{2}\left(\mathcal{B}_{g t}, \mathcal{B}_{p r d}\right)}{C^{2}} \tag{3} DIoU=IoUC2ρ2(Bgt,Bprd)(3)

其中, ρ 2 ( B g t , B p r d ) \rho^{2}(\text{B}_{gt}, \text{B}_{prd}) ρ2(Bgt,Bprd)表示预测边界框和groundtruth边界框的中心点之间的欧几里得距离(如图1(b)中的红虚线所示)。 C 2 C^{2} C2表示最小外接矩形的对角线长度(如图1(b)中的黑虚线所示)。我们可以看到, L D I o U \text{L}_{DIoU} LDIoU的目标直接最小化预测边界框的中心点和groundtruth边界框的中心点之间的距离。然而,当预测边界框的中心点和groundtruth边界框的中心点重合时,它会退化为原始IoU。为了解决这个问题,考虑到中心点距离和 aspect ratio,提出了CIoU。CIoU的公式可以写为:

C I o U = I o U − ρ 2 ( B g t , B p r d ) C 2 − α V , (4) C I o U=I o U-\frac{\rho^{2}\left(\mathcal{B}_{g t}, \mathcal{B}_{p r d}\right)}{C^{2}}-\alpha V, \tag{4} CIoU=IoUC2ρ2(Bgt,Bprd)αV,(4)

V = 4 π 2 ( arctan ⁡ w g t h g t − arctan ⁡ w p r d h p r d ) 2 , (5) V=\frac{4}{\pi^{2}}\left(\arctan \frac{w^{g t}}{h^{g t}}-\arctan \frac{w^{p r d}}{h^{p r d}}\right)^{2}, \tag{5} V=π24(arctanhgtwgtarctanhprdwprd)2,(5)

α = V 1 − I o U + V . (6) \alpha=\frac{V}{1-I o U+V} . \tag{6} α=1IoU+VV.(6)

但是,CIoU中的 aspect ratio 的定义是相对值而不是绝对值。为了解决这个问题,基于DIoU提出了EIoU [41],其定义如下:
E I o U = D I o U − ρ 2 ( w p r d , w g t ) ( w c ) 2 − ρ 2 ( h p r d , h g t ) ( h c ) 2 (7) E I o U=D I o U-\frac{\rho^{2}\left(w_{p r d}, w_{g t}\right)}{\left(w^{c}\right)^{2}}-\frac{\rho^{2}\left(h_{p r d}, h_{g t}\right)}{\left(h^{c}\right)^{2}} \tag{7} EIoU=DIoU(wc)2ρ2(wprd,wgt)(hc)2ρ2(hprd,hgt)(7)
然而,如图2所示,上述用于边界框回归的损失函数在预测边界框和groundtruth边界框具有相同的长宽比但宽度和高度值不同的情况下会失去有效性,这将限制收敛速度和准确性。因此,我们尝试设计一种名为 L M P D I o U \text{L}_{MPDIoU} LMPDIoU的新型损失函数用于边界框回归,同时考虑到 L G I o U [ 39 ] 、 L D I o U [ 40 ] 、 L C I o U [ 42 ] 、 L E I o U \text{L}_{GIoU} [39]、\text{L}_{DIoU}[40]、\text{L}_{CIoU}[42]、\text{L}_{EIoU} LGIoU[39]LDIoU[40]LCIoU[42]LEIoU [41]等损失函数的优点,但对于边界框回归具有更高的效率和准确性。然而,现有损失函数并没有充分利用边界框回归的几何特性。因此,我们通过最小化预测边界框和groundtruth边界框之间的左上角和右下角点之间的距离来提出MPDIoU损失,以更好地训练目标检测、字符级场景文本检测和实例分割的深度模型。
边界框回归的魔法:揭秘精准高效的MPDIoU损失函数_第2张图片

3、点距最小的并集交点

在分析了上述基于IoU的损失函数的优缺点后,我们开始思考如何提高边界框回归的准确性和效率。一般来说,我们使用左上角和右下角点的坐标来定义一个唯一的矩形。受边界框的几何属性的启发,我们设计了一种名为MPDIoU的新型IoU度量来最小化预测边界框和groundtruth边界框之间的左上角和右下角点之间的距离。MPDIoU的计算总结在算法1中。
边界框回归的魔法:揭秘精准高效的MPDIoU损失函数_第3张图片

边界框回归的魔法:揭秘精准高效的MPDIoU损失函数_第4张图片

总之,我们提出的MPDIoU简化了两个边界框之间的相似性比较,可以适应重叠或非重叠边界框回归。因此,MPDIoU可以成为2D/3D计算机视觉任务中使用的所有性能度量的IoU的适当替代品。本文只关注2D目标检测和实例分割,我们可以很容易地将MPDIoU作为度量和损失。轴对齐3D情况的扩展留作未来的工作。

在训练阶段,通过最小化下面的损失函数迫使模型预测的每个边界框 B p r d = [ x p r d , y p r d , w p r d , h p r d ] T \text{B}_{prd}=\left[x^{prd},y^{prd},w^{prd},h^{prd}\right]^T Bprd=[xprd,yprd,wprd,hprd]T接近其groundtruth边界框 B g t = [ x g t , y g t , w g t , h g t ] T \text{B}_{gt}=\left[x^{gt},y^{gt},w^{gt},h^{gt}\right]^T Bgt=[xgt,ygt,wgt,hgt]T
L = min ⁡ Θ ∑ B g t ∈ B g t L ( B g t , B p r d ∣ Θ ) (8) \mathcal{L}=\min _{\Theta} \sum_{\mathcal{B}_{g t} \in \mathbb{B}_{g t}} \mathcal{L}\left(\mathcal{B}_{g t}, \mathcal{B}_{p r d} \mid \Theta\right) \tag{8} L=ΘminBgtBgtL(Bgt,BprdΘ)(8)

其中, B g t \text{B}_{gt} Bgt是groundtruth边界框的集合, Θ \Theta Θ是用于回归的深度模型的参数。 L \text{L} L的典型形式是\ell_{n}-范数,例如,均方误差(MSE)损失和Smooth-\text{}\ell_{1}损失[43],这些损失已广泛应用于目标检测[44],行人检测[45,46],场景文本检测[34,47],3D目标检测[48,49],姿态估计[50,51]和实例分割[52,26]。然而,最近的研究表明,基于 ℓ n \text{}\ell_{n} n-范数的损失函数与IoU评估指标不一致,因此提出基于IoU的损失函数[53,37,39]。根据前面一节中MPDIoU的定义,我们定义基于MPDIoU的损失函数如下:
L MPDIoU  = 1 −  MPDIoU  (9) \mathcal{L}_{\text {MPDIoU }}=1-\text { MPDIoU } \tag{9} LMPDIoU =1 MPDIoU (9)
因此,现有的边界框回归损失函数的所有因子都可以由四个点坐标确定。换算公式如下:
∣ C ∣ = ( max ⁡ ( x 2 g t , x 2 p r d ) − min ⁡ ( x 1 g t , x 1 p r d ) ) ∗ ( max ⁡ ( y 2 g t , y 2 p r d ) − min ⁡ ( y 1 g t , y 1 p r d ) ) , (10) |C|=\left(\max \left(x_{2}^{g t}, x_{2}^{p r d}\right)-\min \left(x_{1}^{g t}, x_{1}^{p r d}\right)\right) *\left(\max \left(y_{2}^{g t}, y_{2}^{p r d}\right)-\min \left(y_{1}^{g t}, y_{1}^{p r d}\right)\right), \tag{10} C=(max(x2gt,x2prd)min(x1gt,x1prd))(max(y2gt,y2prd)min(y1gt,y1prd)),(10)

x c g t = x 1 g t + x 2 g t 2 , y c g t = y 1 g t + y 2 g t 2 , y c p r d = y 1 p r d + y 2 p r d 2 , x c p r d = x 1 p r d + x 2 p r d 2 , (11) x_{c}^{g t}=\frac{x_{1}^{g t}+x_{2}^{g t}}{2}, y_{c}^{g t}=\frac{y_{1}^{g t}+y_{2}^{g t}}{2}, y_{c}^{p r d}=\frac{y_{1}^{p r d}+y_{2}^{p r d}}{2}, x_{c}^{p r d}=\frac{x_{1}^{p r d}+x_{2}^{p r d}}{2}, \tag{11} xcgt=2x1gt+x2gt,ycgt=2y1gt+y2gt,ycprd=2y1prd+y2prd,xcprd=2x1prd+x2prd,(11)

w g t = x 2 g t − x 1 g t , h g t = y 2 g t − y 1 g t , w p r d = x 2 p r d − x 1 p r d , h p r d = y 2 p r d − y 1 p r d . (12) w_{g t}=x_{2}^{g t}-x_{1}^{g t}, h_{g t}=y_{2}^{g t}-y_{1}^{g t}, w_{p r d}=x_{2}^{p r d}-x_{1}^{p r d}, h_{p r d}=y_{2}^{p r d}-y_{1}^{p r d} . \tag{12} wgt=x2gtx1gt,hgt=y2gty1gt,wprd=x2prdx1prd,hprd=y2prdy1prd.(12)
其中,|C|代表包围 B g t \mathcal{B}_{g t} Bgt B p r d \mathcal{B}_{p r d} Bprd的最小矩形的面积, ( x c g t , y c g t ) \left(x_{c}^{g t},y_{c}^{g t}\right) (xcgt,ycgt) ( x c p r d , y c p r d ) \left(x_{c}^{p r d},y_{c}^{p r d}\right) (xcprd,ycprd)分别代表groundtruth边界框和预测边界框的中心点的坐标, w g t w_{g t} wgt h g t h_{g t} hgt代表groundtruth边界框的宽度和高度, w p r d w_{p r d} wprd h p r d h_{p r d} hprd代表预测边界框的宽度和高度。

根据式(10)-(12),我们可以发现现有损失函数中考虑的所有因素都可以通过左上角点和右下角点的坐标确定,例如非重叠区域、中心点距离、宽度和高度偏差,这意味着我们提出的 L M P D I o U \mathcal{L}_{M P D I o U} LMPDIoU不仅考虑周全,而且简化了计算过程。

根据定理3.1,如果预测边界框和groundtruth边界框的宽高比相同,则groundtruth边界框内的预测边界框比groundtruth边界框外的预测边界框具有更低的 L M P D I o U \mathcal{L}_{M P D I o U} LMPDIoU值。这一特性确保了边界框回归的准确性,倾向于为预测边界框提供更少的冗余。

定理3.1。我们定义一个groundtruth边界框为 B gt \mathcal{B}_{\text {gt}} Bgt,两个预测边界框为 B prd 1 \mathcal{B}_{\text {prd}1} Bprd1 B prd 2 \mathcal{B}_{\text {prd}2} Bprd2。输入图像的宽度和高度分别为w和h。假设 B g t \mathcal{B}_{g t} Bgt B prd 1 \mathcal{B}_{\text {prd}1} Bprd1 B prd 2 \mathcal{B}_{\text {prd}2} Bprd2的左上角和右下角坐标为 ( x 1 g t , y 1 g t , x 2 g t , y 2 g t ) \left(x_{1}^{g t},y_{1}^{g t},x_{2}^{g t},y_{2}^{g t}\right) (x1gt,y1gt,x2gt,y2gt) ( x 1 p r d 1 , y 1 p r d 1 , x 2 p r d 1 , y 2 p r d 1 ) \left(x_{1}^{p r d 1},y_{1}^{p r d 1},x_{2}^{p r d 1},y_{2}^{p r d 1}\right) (x1prd1,y1prd1,x2prd1,y2prd1) ( x 1 p r d 2 , y 1 p r d 2 , x 2 p r d 2 , y 2 p r d 2 ) \left(x_{1}^{p r d 2},y_{1}^{p r d 2},x_{2}^{p r d 2},y_{2}^{p r d 2}\right) (x1prd2,y1prd2,x2prd2,y2prd2),则 B g t \mathcal{B}_{g t} Bgt B prd 1 \mathcal{B}_{\text {prd}1} Bprd1 B prd 2 \mathcal{B}_{\text {prd}2} Bprd2的宽度和高度可表示为 ( w g t = y 2 g t − y 1 g t , h g t = x 2 g t − x 1 g t ) \left(w_{g t}=y_{2}^{g t}-y_{1}^{g t},h_{g t}=x_{2}^{g t}-x_{1}^{g t}\right) (wgt=y2gty1gt,hgt=x2gtx1gt) ( w p r d 1 = y 2 p r d 1 − y 1 p r d 1 , h p r d 1 = x 2 p r d 1 − x 1 p r d 1 ) \left(w_{p r d 1}=y_{2}^{p r d 1}-y_{1}^{p r d 1},h_{p r d 1}=x_{2}^{p r d 1}-x_{1}^{p r d 1}\right) (wprd1=y2prd1y1prd1,hprd1=x2prd1x1prd1) ( w p r d 2 = y 2 p r d 2 − y 1 p r d 2 , h p r d 2 = x 2 p r d 2 − x 1 p r d 2 ) \left(w_{p r d 2}=y_{2}^{p r d 2}-y_{1}^{p r d 2},h_{p r d 2}=x_{2}^{p r d 2}-x_{1}^{p r d 2}\right) (wprd2=y2prd2y1prd2,hprd2=x2prd2x1prd2).若 w p r d 1 = k ∗ w g t w_{p r d 1}=k*w_{g t} wprd1=kwgt h p r d 1 = k ∗ h g t h_{p r d 1}=k*h_{g t} hprd1=khgt w p r d 2 = 1 k ∗ w g t w_{p r d 2}=\frac{1}{k}*w_{g t} wprd2=k1wgt h p r d 2 = 1 k ∗ h g t h_{p r d 2}=\frac{1}{k}*h_{g t} hprd2=k1hgt,其中 k > 1 k>1 k>1 k ∈ N ∗ k\in N* kN

B g t \mathcal{B}_{g t} Bgt B prd  1 \mathcal{B}_{\text {prd } 1} Bprd 1 B prd  2 \mathcal{B}_{\text {prd } 2} Bprd 2的中央点都重叠。因此,GIoU( B gt  \mathcal{B}_{\text {gt }} Bgt  B prd  1 \mathcal{B}_{\text {prd } 1} Bprd 1)=GIoU( B gt  \mathcal{B}_{\text {gt }} Bgt  B prd  2 \mathcal{B}_{\text {prd } 2} Bprd 2),DIoU( B g t \mathcal{B}_{g t} Bgt B prd  1 \mathcal{B}_{\text {prd } 1} Bprd 1)=DIoU( B g t \mathcal{B}_{g t} Bgt B prd  2 \mathcal{B}_{\text {prd } 2} Bprd 2),CIoU( B g t \mathcal{B}_{g t} Bgt B prd  1 \mathcal{B}_{\text {prd } 1} Bprd 1)=CIoU( B gt  \mathcal{B}_{\text {gt }} Bgt  B prd  2 \mathcal{B}_{\text {prd } 2} Bprd 2),EIoU( B gt  \mathcal{B}_{\text {gt }} Bgt  B prd  1 \mathcal{B}_{\text {prd } 1} Bprd 1)=EIoU( B gt  \mathcal{B}_{\text {gt }} Bgt  B prd  2 \mathcal{B}_{\text {prd } 2} Bprd 2),但是MPDIoU( B gt  \mathcal{B}_{\text {gt }} Bgt  B prd  1 \mathcal{B}_{\text {prd } 1} Bprd 1)>MPDIoU( B g t \mathcal{B}_{g t} Bgt B p r d 2 \mathcal{B}_{p r d 2} Bprd2)。
边界框回归的魔法:揭秘精准高效的MPDIoU损失函数_第5张图片
边界框回归的魔法:揭秘精准高效的MPDIoU损失函数_第6张图片
考虑到groundtruth边界框 B g t \mathcal{B}_{g t} Bgt是一个面积大于零的矩形,即 A g t > 0 A^{g t}>0 Agt>0。算法2(1)和算法2(6)中的条件分别确保预测面积 A p r d A^{p r d} Aprd和交集面积 I \mathcal{I} I是非负值,即 A prd  ≥ 0 A^{\text {prd }} \geq 0 Aprd 0 I ≥ 0 \mathcal{I} \geq 0 I0,对于任何预测边界框 B prd  = ( x 1 prd  , y 1 prd  , x 2 p r d , y 2 prd  ) ∈ R 4 \mathcal{B}_{\text {prd }}=\left(x_{1}^{\text {prd }}, y_{1}^{\text {prd }}, x_{2}^{p r d}, y_{2}^{\text {prd }}\right) \in \mathbb{R}^{4} Bprd =(x1prd ,y1prd ,x2prd,y2prd )R4。因此,对于任何预测边界框 B p r d = ( x 1 p r d , y 1 p r d , x 2 p r d , y 2 p r d ) ∈ R 4 \mathcal{B}_{p r d}=\left(x_{1}^{p r d}, y_{1}^{p r d}, x_{2}^{p r d}, y_{2}^{p r d}\right) \in \mathbb{R}^{4} Bprd=(x1prd,y1prd,x2prd,y2prd)R4,并集面积 U \mathcal{U} U总是大于交集面积 I \mathcal{I} I,即 U ≥ I \mathcal{U} \geq \mathcal{I} UI。因此, L M P D I o U \mathcal{L}_{M P D I o U} LMPDIoU总是有界的,即 0 ≤ L M P D I o U < 3 0 \leq \mathcal{L}_{M P D I o U}<3 0LMPDIoU<3,对于任何预测边界框 B prd  ∈ R 4 \mathcal{B}_{\text {prd }} \in \mathbb{R}^{4} Bprd R4

I o U = 0 IoU=\mathbf{0} IoU=0时,MPDIoU的行为:对于MPDIoU损失,我们有 L M P D I o U = 1 − M P D I o U = 1 + d 1 2 d 2 + d 2 2 d 2 − I o U \mathcal{L}_{M P D I o U}=1-M P D I o U=1+\frac{d_{1}^{2}}{d^{2}}+\frac{d_{2}^{2}}{d^{2}}-I o U LMPDIoU=1MPDIoU=1+d2d12+d2d22IoU。在 B g t \mathcal{B}_{g t} Bgt B prd \mathcal{B}_{\text {prd}} Bprd不重叠的情况下,这意味着IoU=0,MPDIoU损失可以简化为 L M P D I o U = 1 − M P D I o U = 1 + d 1 2 d 2 + d 2 2 d 2 \mathcal{L}_{M P D I o U}=1-M P D I o U=1+\frac{d_{1}^{2}}{d^{2}}+\frac{d_{2}^{2}}{d^{2}} LMPDIoU=1MPDIoU=1+d2d12+d2d22。在这种情况下,通过最小化 L M P D I o U \mathcal{L}_{M P D I o U} LMPDIoU,我们实际上最小化了 d 1 2 d 2 + d 2 2 d 2 \frac{d_{1}^{2}}{d^{2}}+\frac{d_{2}^{2}}{d^{2}} d2d12+d2d22。该项是一个介于0和1之间的归一化度量,即0≤ d 1 2 d 2 + d 2 2 d 2 < 2 \frac{d_{1}^{2}}{d^{2}}+\frac{d_{2}^{2}}{d^{2}}<2 d2d12+d2d22<2

4、实验结果

我们通过将新的边界框回归损失 L M P D I o U \mathcal{L}_{M P D I o U} LMPDIoU纳入最流行的2D对象检测器和实例分割模型(如YOLO v7[6]和YOLACT[26])来评估它。为此,我们将它们的默认回归损失替换为 L M P D I o U \mathcal{L}_{M P D I o U} LMPDIoU,即替换YOLACT[26]中的 ℓ 1 − \ell_{1} - 1smooth和YOLO v7[6]中的 L CIoU  \mathcal{L}_{\text {CIoU }} LCIoU 。我们还将基线损失与 L G I O U \mathcal{L}_{G I O U} LGIOU进行比较。

4.1、 实验设置

实验环境可以总结如下:内存为32 GB,操作系统为Windows 11,CPU为Intel i9-12900k,显卡为NVIDIA Geforce RTX 3090,内存为24GB。为了进行公平比较,所有实验都使用PyTorch[54]实现。

4.2、数据集

我们训练所有对象检测和实例分割基线,并在两个标准基准上报告所有结果,即PASCAL VOC [10]和Microsoft Common Objects in Context (MS COCO 2017) [11]挑战。它们的训练协议和评估细节将在各自的章节中解释。

PASCAL VOC 2007&2012:Pascal Visual Object Classes (VOC) [10]基准是最广泛使用的用于分类、对象检测和语义分割的数据集之一,包含约9963张图像。训练数据集和测试数据集各占50%,其中来自20个预定义类别的对象用水平边界框标注。由于实例分割图像的尺寸较小,导致性能较弱,我们只提供使用MS COCO 2017训练的实例分割结果。

MS COCO:MS COCO [11]是一个广泛用于图像captioning、对象检测和实例分割的基准,包含超过20万张图像,包括训练、验证和测试数据集,以及超过50万个标注的对象实例,来自80个类别。

IIIT5k:IIIT5k [12]是流行的场景文本检测基准之一,具有字符级别的注释,包含从互联网上收集的5000张裁剪的单词图像。字符类别包括英文字母和数字。有2000张图像用于训练,3000张图像用于测试。

MTHv2:MTHv2 [13]是流行的OCR基准之一,具有字符级别的注释。字符类别包括简化和传统的汉字。它包含超过3000张中文历史文献的图像和超过1百万个汉字。

4.3、 评估协议

在本文中,我们使用与MS COCO 2018挑战[11]相同的性能指标来报告所有结果,包括针对不同类别标签的mAP,以确定真正例和假正例的特定值IoU阈值。我们的实验中使用的对象检测的主要性能指标以精度(precision)和[email protected]:0.95的形式显示。我们在表中报告IoU阈值为0.75的AP75值。对于实例分割,我们的实验中使用的主要性能指标以AP和AR的形式显示,即在不同IoU阈值上平均mAP和mAR,即IoU={0.5,0.55,…0.95}。

所有对象检测和实例分割基线也使用MS COCO 2017和PASCAL VOC 2007&2012的测试集进行了评估。结果将在下一节中展示。

4.4、 目标检测的实验结果

训练协议。我们使用[6]发布的原始Darknet实现的YOLO v7。对于基线结果(使用GIoU损失进行训练),我们在所有实验中选择了DarkNet-608作为主干,并使用报告的默认参数和每个基准的迭代次数完全按照其训练协议进行训练。要使用GIoU、DIoU、CIoU、EIoU和MPDIoU损失训练YOLO v7,我们只需将边界框回归IoU损失替换为 L GIoU \mathcal{L}_{\text {GIoU}} LGIoU L DIoU \mathcal{L}_{\text {DIoU}} LDIoU L C I o U \mathcal{L}_{C I o U} LCIoU L EIoU \mathcal{L}_{\text {EIoU}} LEIoU L MPDIoU \mathcal{L}_{\text {MPDIoU}} LMPDIoU损失,这些损失在2中进行了说明。

根据原始代码的训练协议,我们将YOLOv7[6]使用每个损失在数据集的训练集和验证集上训练了最多150个epoch。我们将早期停止机制的容忍设置为5,以减少训练时间并保存性能最佳的模型。使用每种损失的最佳检查点,在PASCAL VOC 2007&2012的测试集上评估了它们的性能。结果已在表1中报告。
边界框回归的魔法:揭秘精准高效的MPDIoU损失函数_第7张图片
边界框回归的魔法:揭秘精准高效的MPDIoU损失函数_第8张图片
边界框回归的魔法:揭秘精准高效的MPDIoU损失函数_第9张图片

4.5、 字符级场景文本识别的实验结果

训练协议。我们使用与对象检测实验相似的训练协议。按照原始代码的训练协议,我们在该数据集的训练集和验证集上使用每个损失对YOLOv7[6]进行了最多30个epoch的训练。使用每个损失的最佳检查点评估了它们在IIIT5K[12]和MTHv2[55]测试集上的性能。结果已在表2和表3中报告。
边界框回归的魔法:揭秘精准高效的MPDIoU损失函数_第10张图片

正如我们所看到的,Tab.2和Tab.3中的结果显示,使用 L M P D I o U \mathcal{L}_{M P D I o U} LMPDIoU作为回归损失训练YOLO v7可以显着提高其性能,与现有的包括 L GIoU , L DIoU , L CIoU , L EIoU \mathcal{L}_{\text {GIoU}},\mathcal{L}_{\text {DIoU}},\mathcal{L}_{\text {CIoU}},\mathcal{L}_{\text {EIoU}} LGIoULDIoULCIoULEIoU的回归损失相比。我们提出的 L M P D I O U \mathcal{L}_{M P D I O U} LMPDIOU在字符级场景文本检测方面表现出优异的性能。
边界框回归的魔法:揭秘精准高效的MPDIoU损失函数_第11张图片

4.6、 实例分割的实验结果

训练协议。我们使用了加州大学发布的最新PyTorch实现的YOLACT[26]。对于基线结果(使用 L G I o U \mathcal{L}_{G I o U} LGIoU训练),我们在所有实验中选择了ResNet-50作为YOLACT的主干网络架构,并使用报告的默认参数和每个基准的迭代次数按照其训练协议进行训练。要使用GIoU、DIoU、CIoU、EIoU和MPDIoU损失训练YOLACT,我们用 L GIoU \mathcal{L}_{\text {GIoU}} LGIoU L DIoU \mathcal{L}_{\text {DIoU}} LDIoU L C I O U \mathcal{L}_{C I O U} LCIOU L EIoU \mathcal{L}_{\text {EIoU}} LEIoU L M P D I O U \mathcal{L}_{M P D I O U} LMPDIOU损失替换了它们在最终边界框细化阶段的 ℓ 1 − s m o o t h \ell_{1} -smooth 1smooth损失,这些损失在2中进行了说明。与YOLO v7实验类似,我们将原始的边界框回归损失替换为我们提出的 L MPDIoU \mathcal{L}_{\text {MPDIoU}} LMPDIoU

如图8©所示,将 L G I o U \mathcal{L}_{G I o U} LGIoU L D I o U \mathcal{L}_{D I o U} LDIoU L C I o U \mathcal{L}_{C I o U} LCIoU L E I o U \mathcal{L}_{E I o U} LEIoU作为回归损失可以稍微提高YOLACT在MS COCO 2017上的性能。然而,与使用 L M P D I o U \mathcal{L}_{M P D I o U} LMPDIoU训练的情况相比,其改进是显着的,我们可视化了不同值的mask AP与不同值的IoU阈值之间的关系,即0.5≤IoU≤0.95。

与上述实验类似,使用 L M P D I o U \mathcal{L}_{M P D I o U} LMPDIoU作为回归损失可以提高检测准确率,超过现有的损失函数。如表4所示,我们提出的 L MPDIoU \mathcal{L}_{\text {MPDIoU}} LMPDIoU在大多数指标上比现有损失函数表现更好。但是,不同损失之间的改进幅度小于之前的实验。这可能是由于以下几个因素。首先,YOLACT [26]上的检测锚框比YOLO v7 [6]更密集,导致 L MPDIoU \mathcal{L}_{\text {MPDIoU}} LMPDIoU相对于 L I o U \mathcal{L}_{I o U} LIoU具有优势的情况(例如不重叠的边界框)更少。其次,过去的几年中,用于边界框回归的现有损失函数已经得到了改进,这意味着准确率的提高非常有限,但是效率仍有很大的提升空间。
边界框回归的魔法:揭秘精准高效的MPDIoU损失函数_第12张图片

我们还比较了YOLACT使用不同回归损失函数时训练期间的bbox loss和AP值的变化趋势。如图8(a)、(b)所示,使用 L M P D I O U \mathcal{L}_{M P D I O U} LMPDIOU进行训练比大多数现有损失函数(即 L G I o U \mathcal{L}_{G I o U} LGIoU L D I O U \mathcal{L}_{D I O U} LDIOU)表现更好,达到了更高的准确率和更快的收敛速度。尽管bbox loss和AP值显示出很大的波动,但我们的提出的 L M P D I o U \mathcal{L}_{M P D I o U} LMPDIoU在训练结束时表现更好。
边界框回归的魔法:揭秘精准高效的MPDIoU损失函数_第13张图片

为了更好地展示不同损失函数对于实例分割边界框回归的性能,我们提供了如图5和9所示的一些可视化结果。我们可以看到,除了 L GIoU \mathcal{L}_{\text {GIoU}} LGIoU L DIoU \mathcal{L}_{\text {DIoU}} LDIoU L CIoU \mathcal{L}_{\text {CIoU}} LCIoU L E I o U \mathcal{L}_{E I o U} LEIoU外,我们使用 L M P D I o U \mathcal{L}_{M P D I o U} LMPDIoU提供了更少冗余和更高精度的实例分割结果。
边界框回归的魔法:揭秘精准高效的MPDIoU损失函数_第14张图片

5、 结论

在本文中,我们介绍了一种新的度量方法,即基于最小点距离的MPDIoU,用于比较任意两个任意边界框。我们证明了这种新度量方法具有现有IoU基度量所具有的所有吸引力,同时简化了其计算。在所有依赖于IoU度量的2D/3D视觉任务中,它都将是更好的选择。

我们还提出了一个名为 L M P D I O U \mathcal{L}_{M P D I O U} LMPDIOU的损失函数,用于边界框回归。我们使用通用性能度量方法和我们提出的MPDIoU,将其应用于state-of-the-art对象检测和实例分割算法,从而改进了在PASCAL VOC、MS COCO、MTHv2和IIIT5K等流行基准测试上的性能。由于对于一个度量的最优损失是该度量本身,因此我们的MPDIoU损失可以作为所有需要2D边界框回归的应用中的最优边界框回归损失。

对于未来工作,我们希望在基于对象检测和实例分割的一些下游任务上进一步实验,包括场景文本检测、人员重识别等。通过上述实验,我们可以进一步验证我们提出的损失函数的泛化能力。

你可能感兴趣的:(回归,数据挖掘,人工智能)