arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2404.01699cs.CV
收藏

面向目标检测器的任务集成蒸馏

Task Integration Distillation for Object Detectors

South China Normal University(华南师范大学)

Hai Su, ZhenWen Jian, Songsen Yu

AI总结:

针对目标检测知识蒸馏忽略回归任务导致性能不佳的问题,提出一种集成分类与回归任务并引入任务重要性策略的蒸馏方法,以均衡利用检测器输出并提升蒸馏效果。

摘要:

知识蒸馏是一种被广泛采用的模型轻量化技术。然而,大多数知识蒸馏方法在目标检测领域中的性能并不令人满意。通常,知识蒸馏方法仅考虑目标检测器两个子任务中的分类任务,在很大程度上忽略了回归任务。这一疏忽导致对目标检测器综合任务的理解不完整,从而产生有偏差的估计并可能带来不利影响。因此,我们提出了一种同时处理分类和回归任务的知识蒸馏方法,并引入了任务重要性策略。通过基于检测器两个子任务的输出评估特征的重要性,我们的方法确保在目标检测中均衡地考虑分类和回归任务。受现实世界教学过程和学习条件定义的启发,我们引入了一种同时关注关键区域和薄弱区域的方法。通过根据特征的重要性差异评估其特征对知识蒸馏的价值,我们准确捕捉当前模型的学习状况。该方法有效防止了由于对检测器输出的利用不完整而导致的对模型学习实际情况的预测偏差问题。

英文摘要:

Knowledge distillation is a widely adopted technique for model lightening. However, the performance of most knowledge distillation methods in the domain of object detection is not satisfactory. Typically, knowledge distillation approaches consider only the classification task among the two sub-tasks of an object detector, largely overlooking the regression task. This oversight leads to a partial understanding of the object detector's comprehensive task, resulting in skewed estimations and potentially adverse effects. Therefore, we propose a knowledge distillation method that addresses both the classification and regression tasks, incorporating a task significance strategy. By evaluating the importance of features based on the output of the detector's two sub-tasks, our approach ensures a balanced consideration of both classification and regression tasks in object detection. Drawing inspiration from real-world teaching processes and the definition of learning condition, we introduce a method that focuses on both key and weak areas. By assessing the value of features for knowledge distillation based on their importance differences, we accurately capture the current model's learning situation. This method effectively prevents the issue of biased predictions about the model's learning reality caused by an incomplete utilization of the detector's outputs.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑