计算机应用 ›› 2021, Vol. 41 ›› Issue (7): 1928-1932.DOI: 10.11772/j.issn.1001-9081.2020101615

所属专题: 人工智能

• 人工智能 • 上一篇    下一篇

基于部分加权损失函数的RefineDet

肖振远, 王逸涵, 罗建桥, 熊鹰, 李柏林   

  1. 西南交通大学 机械工程学院, 成都 610031
  • 收稿日期:2020-10-19 修回日期:2021-01-17 出版日期:2021-07-10 发布日期:2021-01-27
  • 通讯作者: 李柏林
  • 作者简介:肖振远(1994-),男,安徽界首人,硕士研究生,主要研究方向:机器视觉;王逸涵(1994-),男,湖北黄冈人,硕士研究生,主要研究方向:机器视觉;罗建桥(1991-),男,湖南湘潭人,博士研究生,主要研究方向:机器视觉;熊鹰(1974-),男,湖北武汉人,讲师,硕士,主要研究方向:机器视觉;李柏林(1962-),男,广西桂林人,教授,博士,主要研究方向:机器视觉、机器人。
  • 基金资助:
    国家自然科学基金青年科学基金资助项目(51705436)。

RefineDet based on subsection weighted loss function

XIAO Zhenyuan, WANG Yihan, LUO Jianqiao, XIONG Ying, LI Bailin   

  1. School of Mechanical Engineering, Southwest Jiaotong University, Chengdu Sichuan 610031, China
  • Received:2020-10-19 Revised:2021-01-17 Online:2021-07-10 Published:2021-01-27
  • Supported by:
    This work is partially supported by Youth Program of National Natural Science Foundation of China (51705436).

摘要: 针对目标检测网络单阶改进目标检测器(RefineDet)对类间不平衡数据集中小样本类别检测性能差的问题,提出一种部分加权损失函数SWLoss。首先,以每个训练批量中不同类别样本数量的倒数作为启发式的类间样本平衡因子,对分类损失中的不同类别进行加权,从而提高对小样本类别学习的关注程度;然后引入多任务平衡因子对分类损失和回归损失进行加权,缩小两个任务学习速率的差异;最后,在目标类别样本数量存在大幅差异的Pascal VOC 2007数据集和点阵字符数据集上进行实验。结果表明,与原始RefineDet相比,基于SWLoss的RefineDet明显提高了小样本类别的检测精度,它在两个数据集上的平均精度均值(mAP)分别提高了1.01、9.86个百分点;与基于损失平衡函数和加权成对损失的RefineDet相比,基于SWLoss的RefineDet在两个数据集上的mAP分别提高了0.68、4.73和0.49、1.48个百分点。

关键词: 目标检测, 不平衡数据集, 加权损失, 分类损失, 回归损失

Abstract: Concerning the poor performance of the Single-Shot Refinement Neural Network for Object Detection (RefineDet) of the object detection network when detecting small sample classes in inter-class imbalanced datasets, a Subsection Weighted Loss (SWLoss) function was proposed. Firstly, the inverse of the number of samples from different classes in each training batch was used as the heuristic inter-class sample balance factor to weight the different classes in the classification loss, thus strengthening the concern on the small sample class learning. After that, a multi-task balancing factor was introduced to weight classification loss and regression loss to reduce the difference between the learning rates of two tasks. At last, experiments were conducted on Pascal VOC2007 dataset and dot-matrix character dataset with large differences in the number of target class samples. The results demonstrate that compared to the original RefineDet, the SWLoss-based RefineDet clearly improves the detection precision of small sample classes, and has the mean Average Precision (mAP) on the two datasets increased by 1.01 and 9.86 percentage points, respectively; and compared to the RefineDet based on loss balance function and weighted pairwise loss, the SWLoss-based RefineDet has the mAP on the two datasets increased by 0.68, 4.73 and 0.49, 1.48 percentage points, respectively.

Key words: object detection, imbalanced dataset, weighted loss, classification loss, regression loss

中图分类号: