LLM2D
部分域适应的软遮罩半对偶最优传输
Soft-Masked Semi-Dual Optimal Transport for Partial Domain Adaptation
作者: Yi-Ming Zhai, Chuan-Xian Ren, Hong Yan
发布日期: 5/6/2025
arXiv ID: oai:arXiv.org:2505.01664v1

摘要

arXiv:2505.01664v1 宣传类型: 交叉 摘要: 视觉领域适应旨在通过利用标记源域的知识来学习未标记目标域的判别性和领域不变表示。部分领域适应(PDA)是一个通用且实际的场景,其中目标标签空间是源标签空间的子集。由于领域漂移以及不同领域的不相同标签空间,PDA 存在挑战。本文提出了一种软掩码半对偶最优运输(SSOT)方法来解决 PDA 问题。具体来说,估计了域的类别权重,然后构造了一个加权后的源域,这有利于在目标域上进行条件类别分布匹配。通过类别预测构建了一个软掩码的运输距离矩阵,这将增强在共用特征空间中最优运输的类别导向表示能力。为了处理大规模最优运输问题,采用半对偶形式的带熵正则化 Kantorovich 问题,因为其可以通过梯度优化算法进行优化。此外,由于其强大的拟合能力,利用神经网络来近似 Kantorovich 潜力函数。这种网络参数化也允许将对偶变量推广到输入分布的支持之外。SSOT 模型基于神经网络,并可通过端到端的方式交替优化。在四个基准数据集上进行了广泛的实验,以证明 SSOT 的有效性。