arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2003.13249cs.LGstat.ML

Towards Stable and Comprehensive Domain Alignment: Max-Margin Domain-Adversarial Training

Nanyang Technological University(南洋理工大学);University of California, Berkeley(加州大学伯克利分校)

Jianfei Yang, Han Zou, Yuxun Zhou, Lihua Xie

英文摘要:

Domain adaptation tackles the problem of transferring knowledge from a label-rich source domain to a label-scarce or even unlabeled target domain. Recently domain-adversarial training (DAT) has shown promising capacity to learn a domain-invariant feature space by reversing the gradient propagation of a domain classifier. However, DAT is still vulnerable in several aspects including (1) training instability due to the overwhelming discriminative ability of the domain classifier in adversarial training, (2) restrictive feature-level alignment, and (3) lack of interpretability or systematic explanation of the learned feature space. In this paper, we propose a novel Max-margin Domain-Adversarial Training (MDAT) by designing an Adversarial Reconstruction Network (ARN). The proposed MDAT stabilizes the gradient reversing in ARN by replacing the domain classifier with a reconstruction network, and in this manner ARN conducts both feature-level and pixel-level domain alignment without involving extra network structures. Furthermore, ARN demonstrates strong robustness to a wide range of hyper-parameters settings, greatly alleviating the task of model selection. Extensive empirical results validate that our approach outperforms other state-of-the-art domain alignment methods. Moreover, reconstructing adapted features reveals the domain-invariant feature space which conforms with our intuition.

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

↑