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

LabelEnc: A New Intermediate Supervision Method for Object Detection

Beijing University of Posts and Telecommunications(北京邮电大学);Tongji University(同济大学);MEGVII Technology(旷视科技)

Miao Hao, Yitao Liu, Xiangyu Zhang, Jian Sun

英文摘要:

In this paper we propose a new intermediate supervision method, named LabelEnc, to boost the training of object detection systems. The key idea is to introduce a novel label encoding function, mapping the ground-truth labels into latent embedding, acting as an auxiliary intermediate supervision to the detection backbone during training. Our approach mainly involves a two-step training procedure. First, we optimize the label encoding function via an AutoEncoder defined in the label space, approximating the "desired" intermediate representations for the target object detector. Second, taking advantage of the learned label encoding function, we introduce a new auxiliary loss attached to the detection backbones, thus benefiting the performance of the derived detector. Experiments show our method improves a variety of detection systems by around 2% on COCO dataset, no matter one-stage or two-stage frameworks. Moreover, the auxiliary structures only exist during training, i.e. it is completely cost-free in inference time. Code is available at: https://github.com/megvii-model/LabelEnc

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

↑