计算机应用

作者：李巍, 梁斯昕, 张建州

单位：四川大学 计算机学院，成都 610065

关键词：长尾数据集,累计梯度比值,加权损失,图像缺陷检测,卷积神经网络

基金：中国博士后科学基金资助项目(2022M712235);四川大学专职博士后研发基金资助项目(2022SCU12074)

针对目前图像缺陷检测模型对长尾缺陷数据集中尾部类检测效果较差的问题，提出一个基于梯度引导加权?延迟负梯度衰减损失（GGW-DND Loss）。首先，根据检测器分类节点的累积梯度比值分别对正负梯度重新加权，减轻尾部类分类器的受抑制状态；其次，当模型优化到一定阶段时，直接降低每个节点产生的负梯度，以增强尾部类分类器的泛化能力。实验结果表明，在自制图像缺陷数据集和NEU-DET（NEU surface defect database for Defect Detection Task）上，所提损失的尾部类平均精度均值（mAP）优于二分类交叉熵损失（BCE Loss），分别提高了32.02和7.40个百分点；与EQL v2（EQualization Loss v2）相比，分别提高了2.20和0.82个百分点，验证了所提损失能有效提升网络对尾部类的检测性能。

来源：2023年第10期

《计算机应用》期刊编辑部

查看计算机应用杂志2023年第10期