轻量模型域适应新方法,通过师生协同提升性能
Collaborative Learning for Enhanced Unsupervised Domain Adaptation
- 师生模型互相优化,用学生更新教师的冗余参数
- 在GTA到Cityscapes上学生模型提升2.0% mIoU
- 适合资源受限场景下的高效域自适应应用
无监督域适应(UDA)旨在将标注源域训练的模型部署到未标注目标域。然而,现有高性能模型计算开销大,难以落地,亟需轻量但有效的方案。对于轻量模型的UDA,知识蒸馏(KD)常用师生框架,但我们发现域偏移导致教师模型产生大量非关键参数,降低泛化能力并传递错误信息给学生模型。有趣的是,这一现象在学生模型中较弱。基于此,我们提出协同学习方法CLDA:用学生模型更新教师的非关键参数,同时利用更新后的教师模型提升学生性能。实验显示,语义分割任务中,GTA-to-Cityscapes数据集下教师模型提升+0.7% mIoU,学生模型提升+1.4%;Synthia-to-Cityscapes下教师+0.8%、学生+2.0% mIoU。
原文摘要 · Abstract (English)
Unsupervised Domain Adaptation (UDA) endeavors to bridge the gap between a model trained on a labeled source domain and its deployment in an unlabeled target domain. However, current high-performance models demand significant resources, making deployment costs prohibitive and highlighting the need for compact, yet effective models. For UDA of lightweight models, Knowledge Distillation (KD) leveraging a Teacher-Student framework could be a common approach, but we found that domain shift in UDA leads to a significant increase in non-salient parameters in the teacher model, degrading model's generalization ability and transferring misleading information to the student model. Interestingly, we observed that this phenomenon occurs considerably less in the student model. Driven by this insight, we introduce Collaborative Learning for UDA (CLDA), a method that updates the teacher's non-salient parameters using the student model and at the same time utilizes the updated teacher model to improve UDA performance of the student model. Experiments show consistent performance improvements for both student and teacher models. For example, in semantic segmentation, CLDA achieves an improvement of +0.7% mIoU for the teacher model and +1.4% mIoU for the student model compared to the baseline model in the GTA-to-Cityscapes datasets. In the Synthia-to-Cityscapes dataset, it achieves an improvement of +0.8% mIoU and +2.0% mIoU for the teacher and student models, respectively.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。