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

M-ADDA:无监督领域自适应与深度度量学习

M-ADDA: Unsupervised Domain Adaptation with Deep Metric Learning

University of British Columbia(不列颠哥伦比亚大学)

Issam Laradji, Reza Babanezhad

摘要:

无监督领域自适应技术已经成功地用于监督标签有限的广泛问题。该任务是通过利用来自稍微相似的分布的标记的“源”数据集来对未标记的“目标”数据集进行分类。我们提出了基于度量的对抗性判别领域自适应(M-ADDA),它执行两个主要步骤。首先,它使用度量学习方法通过优化三元组损失函数来在源数据集上训练源模型。这会导致集群中相同标签的嵌入彼此接近,而具有不同标签的嵌入彼此远离。接下来,它使用对抗方法(如ADDA \cite{2017arXiv170205464T}中使用的方法)使从源数据集和目标数据集提取的特征无法区分。同时,我们优化了一个新的损失函数,鼓励目标数据集的嵌入形成集群。虽然ADDA和M-ADDA使用类似的架构,但我们表明M-ADDA在MNIST和USPS的数字自适应数据集上表现得更好。这表明,使用度量学习的领域自适应可以导致大的改进,在分类精度的领域自适应任务。代码可在\url{https://github.com/IssamLaradji/M-ADDA}上获得。

英文摘要:

Unsupervised domain adaptation techniques have been successful for a wide range of problems where supervised labels are limited. The task is to classify an unlabeled `target' dataset by leveraging a labeled `source' dataset that comes from a slightly similar distribution. We propose metric-based adversarial discriminative domain adaptation (M-ADDA) which performs two main steps. First, it uses a metric learning approach to train the source model on the source dataset by optimizing the triplet loss function. This results in clusters where embeddings of the same label are close to each other and those with different labels are far from one another. Next, it uses the adversarial approach (as that used in ADDA \cite{2017arXiv170205464T}) to make the extracted features from the source and target datasets indistinguishable. Simultaneously, we optimize a novel loss function that encourages the target dataset's embeddings to form clusters. While ADDA and M-ADDA use similar architectures, we show that M-ADDA performs significantly better on the digits adaptation datasets of MNIST and USPS. This suggests that using metric-learning for domain adaptation can lead to large improvements in classification accuracy for the domain adaptation task. The code is available at \url{https://github.com/IssamLaradji/M-ADDA}.

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

↑