arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2402.18411cs.CV

基于原型最优传输的无监督跨域图像检索

Unsupervised Cross-Domain Image Retrieval via Prototypical Optimal Transport

ShanghaiTech University(上海科技大学);Beihang University(北京航空航天大学)

Bin Li, Ye Shi, Qian Yu, Jingya Wang

AI总结:

本文提出ProtoOT框架,将K-means原型生成与最优传输结合,并引入对比学习,统一解决无监督跨域图像检索中的域内表示学习与跨域对齐问题,在多个基准上显著超越现有方法。

摘要:

无监督跨域图像检索(UCIR)旨在不依赖标注数据的情况下,检索跨不同域但属于同一类别的图像。先前的方法通常将UCIR问题分解为两个独立任务:域内表示学习和跨域特征对齐。然而,这种分离式策略忽视了这些任务之间潜在的协同作用。本文提出ProtoOT,这是一种专门为UCIR设计的新型最优传输(Optimal Transport)公式,它将域内特征表示学习和跨域对齐整合到一个统一框架中。ProtoOT利用K-means聚类方法的优势,有效管理UCIR中固有的分布不平衡问题。通过使用K-means生成初始原型并近似类别边际分布,我们相应修改了最优传输中的约束条件,显著提升了其在UCIR场景下的性能。此外,我们将对比学习引入ProtoOT框架,以进一步改进表示学习。这促使具有相似语义的特征之间保持局部语义一致性,同时明确强制特征与不匹配原型之间的分离,从而增强全局判别力。ProtoOT在基准数据集上以显著优势超越了现有最先进方法。值得注意的是,在DomainNet上,ProtoOT实现了平均P@200提升18.17%,在Office-Home上,其P@15提升了3.83%。

英文摘要:

Unsupervised cross-domain image retrieval (UCIR) aims to retrieve images sharing the same category across diverse domains without relying on labeled data. Prior approaches have typically decomposed the UCIR problem into two distinct tasks: intra-domain representation learning and cross-domain feature alignment. However, these segregated strategies overlook the potential synergies between these tasks. This paper introduces ProtoOT, a novel Optimal Transport formulation explicitly tailored for UCIR, which integrates intra-domain feature representation learning and cross-domain alignment into a unified framework. ProtoOT leverages the strengths of the K-means clustering method to effectively manage distribution imbalances inherent in UCIR. By utilizing K-means for generating initial prototypes and approximating class marginal distributions, we modify the constraints in Optimal Transport accordingly, significantly enhancing its performance in UCIR scenarios. Furthermore, we incorporate contrastive learning into the ProtoOT framework to further improve representation learning. This encourages local semantic consistency among features with similar semantics, while also explicitly enforcing separation between features and unmatched prototypes, thereby enhancing global discriminativeness. ProtoOT surpasses existing state-of-the-art methods by a notable margin across benchmark datasets. Notably, on DomainNet, ProtoOT achieves an average P@200 enhancement of 18.17%, and on Office-Home, it demonstrates a P@15 improvement of 3.83%.

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

↑