Journal of Computer Applications ›› 2023, Vol. 43 ›› Issue (10): 3267-3274.DOI: 10.11772/j.issn.1001-9081.2022091413

Special Issue: 多媒体计算与计算机仿真

• Multimedia computing and computer simulation • Previous Articles     Next Articles

Long-tailed image defect detection based on gradient-guide weighted-deferred negative gradient decay loss

Wei LI(), Sixin LIANG, Jianzhou ZHANG   

  1. College of Computer Science,Sichuan University,Chengdu Sichuan 610065,China
  • Received:2022-09-26 Revised:2022-12-04 Accepted:2022-12-12 Online:2023-03-07 Published:2023-10-10
  • Contact: Wei LI
  • About author:LI Wei, born in 1998, M. S. candidate. His research interests include defect detection, object detection.
    LIANG Sixin, born in 1995, Ph. D. candidate. His researchinterests include machine vision, camera calibration, medical image processing.
    ZHANG Jianzhou, born in 1962, Ph. D., professor. His research interests include machine vision, computer vision.
  • Supported by:
    China Postdoctoral Science Foundation(2022M712235);Postdoctoral Research and Development Foundation of Sichuan University(2022SCU12074)

基于梯度引导加权延迟负梯度衰减损失的长尾图像缺陷检测

李巍(), 梁斯昕, 张建州   

  1. 四川大学 计算机学院,成都 610065
  • 通讯作者: 李巍
  • 作者简介:李巍(1998—),男,江苏泰州人,硕士研究生,主要研究方向:缺陷检测、目标检测. wana@stu.scu.edu.cn
    梁斯昕(1995—),男,云南曲靖人,博士研究生,主要研究方向:机器视觉、摄像机标定、医学图像处理
    张建州(1962—),男,河北邯郸人,教授,博士,主要研究方向:机器视觉、计算机视觉。
  • 基金资助:
    中国博士后科学基金资助项目(2022M712235);四川大学专职博士后研发基金资助项目(2022SCU12074)

Abstract:

Aiming at the problem that the current image defect detection models have poor detection effect on tail categories in long-tail defect datasets, a GGW-DND Loss (Gradient-Guide Weighted-Deferred Negative Gradient decay Loss) was proposed. First, the positive and negative gradients were re-weighted according to the cumulative gradient ratio of the classification nodes in the detector in order to reduce the suppressed state of tail classifier. Then, once the model was optimized to a certain stage, the negative gradient generated by each node was sharply reduced to enhance the generalization ability of the tail classifier. Experimental results on the self-made image defect dataset and NEU-DET (NEU surface defect database for Defect dEtection Task) show that the mean Average Precision (mAP) for tail categories of the proposed loss is better than that of Binary Cross Entropy Loss (BCE Loss), the former is increased by 32.02 and 7.40 percentage points respectively, and compared with EQL v2 (EQualization Loss v2), the proposed loss has the mAP increased by 2.20 and 0.82 percentage points respectively, verifying that the proposed loss can effectively improve the detection performance of the network for tail categories.

Key words: long-tail dataset, cumulative gradient ratio, weighted loss, image defect detection, Convolutional Neural Network (CNN)

摘要:

针对目前图像缺陷检测模型对长尾缺陷数据集中尾部类检测效果较差的问题,提出一个基于梯度引导加权?延迟负梯度衰减损失(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个百分点,验证了所提损失能有效提升网络对尾部类的检测性能。

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

CLC Number: