arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2110.06014cs.CVcs.LG
收藏

重新思考有监督的预训练,以更好地进行下游转移

Rethinking Supervised Pre-training for Better Downstream Transferring

Tsinghua University(清华大学);Alibaba Group(阿里巴巴集团)

Yutong Feng, Jianwen Jiang, Mingqian Tang, Rong Jin, Yue Gao

摘要:

pretrain-finetune范式在深度学习的许多应用中表现出出色的性能,其中模型在上游大型数据集(例如ImageNet)上进行预训练,然后针对不同的下游任务进行微调。虽然在大多数情况下,预训练阶段是基于监督方法进行的,但最近的自监督预训练工作显示出强大的可移植性,甚至在多个下游任务上优于监督预训练。因此,如何更好地将监督预训练模型推广到下游任务仍然是一个悬而未决的问题。在本文中,我们认为,现有的监督预训练方法的可移植性差,从有价值的类内语义差异的忽视。这是因为这些方法倾向于将来自同一类的图像彼此靠近,尽管它们的视觉内容具有很大的多样性,这一问题被称为“上游任务的过拟合”。为了缓解这个问题,我们提出了一种新的基于Leave-One-Out K-Nearest-Neighbor(简称LOOK)的监督预训练方法。它通过只要求每个图像与其k个最近邻居中的大多数共享其类标签来缓解过度拟合上游任务的问题,从而允许每个类呈现多模式分布,并因此保留部分类内差异以更好地传输到下游任务。我们开发了所提出的方法的有效实现,该方法可以很好地扩展到大型数据集。对多个下游任务的实验研究表明,LOOK优于其他最先进的监督和自监督预训练方法。

英文摘要:

The pretrain-finetune paradigm has shown outstanding performance on many applications of deep learning, where a model is pre-trained on a upstream large dataset (e.g. ImageNet), and is then fine-tuned to different downstream tasks. Though for most cases, the pre-training stage is conducted based on supervised methods, recent works on self-supervised pre-training have shown powerful transferability and even outperform supervised pre-training on multiple downstream tasks. It thus remains an open question how to better generalize supervised pre-training model to downstream tasks. In this paper, we argue that the worse transferability of existing supervised pre-training methods arise from the negligence of valuable intra-class semantic difference. This is because these methods tend to push images from the same class close to each other despite of the large diversity in their visual contents, a problem to which referred as "overfit of upstream tasks". To alleviate this problem, we propose a new supervised pre-training method based on Leave-One-Out K-Nearest-Neighbor, or LOOK for short. It relieves the problem of overfitting upstream tasks by only requiring each image to share its class label with most of its k nearest neighbors, thus allowing each class to exhibit a multi-mode distribution and consequentially preserving part of intra-class difference for better transferring to downstream tasks. We developed efficient implementation of the proposed method that scales well to large datasets. Experimental studies on multiple downstream tasks show that LOOK outperforms other state-of-the-art methods for supervised and self-supervised pre-training.

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

↑