Apr, 2024

基于卷积的概率梯度损失用于语义分割

TL;DR本文介绍了一种新颖的基于卷积的概率梯度(CPG)损失方法,它使用类似 Sobel 算子的卷积核计算图像中像素强度的梯度,从而能够计算真值和预测的类别概率之间的梯度。这种方法通过最大化这两个概率梯度之间的相似性来提高网络性能。此外,为了专门增强物体边界附近的准确性,我们根据真值概率梯度提取物体边界,并仅将 CPG 损失应用于属于边界的像素。CPG 损失被证明非常方便和有效,通过卷积建立像素间的关系,并且与像素级损失函数(例如交叉熵损失)相比,从独特的维度计算错误。我们对三个常用的分割数据集(Cityscapes,COCO-Stuff,ADE20K)中三个已建立网络(DeepLabv3-Resnet50,HRNetV2-OCR,LRASPP_MobileNet_V3_Large)的 CPG 损失的影响进行了定性和定量分析,广泛的实验结果一致且明显地表明 CPG 损失提高了平均交并比。