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

通过协作一致性训练实现通用半监督模型适配

Universal Semi-supervised Model Adaptation via Collaborative Consistency Training

FNii, CUHKSZ(香港中文大学(深圳)福田诺亚方舟研究院);SSE, CUHKSZ(香港中文大学(深圳)理工学院);Cardiff University(卡迪夫大学);Sun Yat-sen University(中山大学);Research Institute, Sun Yat-sen University, Shenzhen(中山大学深圳研究院)

Zizheng Yan, Yushuang Wu, Yipeng Qin, Xiaoguang Han, Shuguang Cui, Guanbin Li

AI总结:

针对标签集部分重叠且目标域每类仅有少量标签的USMA问题,提出源模型与仅目标模型协作的一致性训练框架及样本级、类别级正则化,在多个基准上验证有效。

摘要:

本文提出了一个现实且具有挑战性的领域适配问题,称为通用半监督模型适配(USMA):其一,仅需要一个预训练的源模型;其二,允许源域和目标域具有不同的标签集合,即它们共享一个公共标签集,同时各自拥有私有标签集;其三,目标域每个类别仅需要少量有标签样本。为解决USMA,我们提出一种协作一致性训练框架,对两个模型(即预训练源模型和仅用目标数据预训练的其变体)之间的预测一致性进行正则化,并结合二者的互补优势以学习更强大的模型。该框架的依据来自如下观察:源模型在公共类别上的表现优于仅目标模型,而在目标私有类别上,仅目标模型表现更好。我们还提出了样本级和类别级双视角的一致性正则化以改进训练。实验结果证明了本方法在多个基准数据集上的有效性。

英文摘要:

In this paper, we introduce a realistic and challenging domain adaptation problem called Universal Semi-supervised Model Adaptation (USMA), which i) requires only a pre-trained source model, ii) allows the source and target domain to have different label sets, i.e., they share a common label set and hold their own private label set, and iii) requires only a few labeled samples in each class of the target domain. To address USMA, we propose a collaborative consistency training framework that regularizes the prediction consistency between two models, i.e., a pre-trained source model and its variant pre-trained with target data only, and combines their complementary strengths to learn a more powerful model. The rationale of our framework stems from the observation that the source model performs better on common categories than the target-only model, while on target-private categories, the target-only model performs better. We also propose a two-perspective, i.e., sample-wise and class-wise, consistency regularization to improve the training. Experimental results demonstrate the effectiveness of our method on several benchmark datasets.

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

↑