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

基于长尾视觉识别的蒸馏过程自监控

Self Supervision to Distillation for Long-Tailed Visual Recognition

Nanjing University(南京大学)

Tianhao Li, Limin Wang, Gangshan Wu

摘要:

深度学习在大规模平衡数据集上的视觉识别方面取得了显着进展,但在现实世界的长尾数据上仍然表现不佳。以往的方法通常采用类再平衡训练策略来有效缓解不平衡问题,但可能存在过拟合尾类的风险。最近的解耦方法通过使用多阶段训练方案克服了过拟合问题,然而,它仍然无法在特征学习阶段捕获尾类信息。在本文中,我们表明,软标签可以作为一个强大的解决方案,将标签相关到一个多阶段的训练计划长尾识别。软标签所体现的类之间的内在联系,有助于长尾识别,将知识从头部转移到尾部类。具体来说,我们提出了一个概念上简单但特别有效的多阶段训练计划,称为自监督蒸馏(SSD)。该方案由两部分组成。首先,我们介绍了一个自蒸馏框架的长尾识别,它可以自动挖掘标签关系。其次,提出了一种新的自监督引导的蒸馏标签生成模块。提取的标签集成了标签和数据域的信息,可以有效地建模长尾分布。我们进行了广泛的实验,我们的方法在三个长尾识别基准测试中取得了最先进的结果:ImageNet-LT,CIFAR 100-LT和iNaturalist 2018。我们的SSD在各种数据集上从$2.7\%$到$4.5\%$都优于强大的LWS基线。该代码可在https://github.com/MCG-NJU/SSD-LT上获得。

英文摘要:

Deep learning has achieved remarkable progress for visual recognition on large-scale balanced datasets but still performs poorly on real-world long-tailed data. Previous methods often adopt class re-balanced training strategies to effectively alleviate the imbalance issue, but might be a risk of over-fitting tail classes. The recent decoupling method overcomes over-fitting issues by using a multi-stage training scheme, yet, it is still incapable of capturing tail class information in the feature learning stage. In this paper, we show that soft label can serve as a powerful solution to incorporate label correlation into a multi-stage training scheme for long-tailed recognition. The intrinsic relation between classes embodied by soft labels turns out to be helpful for long-tailed recognition by transferring knowledge from head to tail classes. Specifically, we propose a conceptually simple yet particularly effective multi-stage training scheme, termed as Self Supervised to Distillation (SSD). This scheme is composed of two parts. First, we introduce a self-distillation framework for long-tailed recognition, which can mine the label relation automatically. Second, we present a new distillation label generation module guided by self-supervision. The distilled labels integrate information from both label and data domains that can model long-tailed distribution effectively. We conduct extensive experiments and our method achieves the state-of-the-art results on three long-tailed recognition benchmarks: ImageNet-LT, CIFAR100-LT and iNaturalist 2018. Our SSD outperforms the strong LWS baseline by from $2.7\%$ to $4.5\%$ on various datasets. The code is available at https://github.com/MCG-NJU/SSD-LT.

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

↑