3 个月前

领域对抗训练中平滑性的深入分析

领域对抗训练中平滑性的深入分析

摘要

域对抗训练(Domain Adversarial Training)已被广泛应用于构建不变特征表示,并在各类域适应任务中得到普遍应用。近年来,收敛至平滑极小值的优化方法在监督学习任务(如分类)中展现出更优的泛化性能。本文研究了平滑性增强型优化策略对域对抗训练的影响。域对抗训练的目标函数由任务损失(如分类损失、回归损失等)与对抗项共同构成。我们发现,若在任务损失意义上收敛至平滑极小值,可显著稳定对抗训练过程,从而提升模型在目标域上的性能。相反,若在对抗损失意义上追求平滑极小值,则会导致目标域上泛化性能下降,即表现次优。基于上述分析,我们提出一种新的平滑域对抗训练方法(Smooth Domain Adversarial Training, SDAT),该方法能有效提升现有域对抗训练方法在分类与目标检测任务中的性能。此外,本研究还为领域自适应社区中为何普遍偏好使用随机梯度下降(SGD)而非Adam优化器提供了理论解释。

代码仓库

val-iisc/sdat
官方
pytorch
GitHub 中提及

基准测试

基准方法指标
domain-adaptation-on-office-homeSDAT (ViT-B/16)
Accuracy: 84.3
domain-adaptation-on-visda2017SDAT(ViT)
Accuracy: 89.8

用 AI 构建 AI

从想法到上线——通过免费 AI 协同编程、开箱即用的环境和市场最优价格的 GPU 加速您的 AI 开发

AI 协同编程
即用型 GPU
最优价格
立即开始

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
领域对抗训练中平滑性的深入分析 | 论文 | HyperAI超神经