arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2404.05366cs.CV

CDAD-Net:弥合广义类别发现中的域差距

CDAD-Net: Bridging Domain Gaps in Generalized Category Discovery

INRIA(法国国家信息与自动化研究所);Indian Institute of Technology Bombay(印度理工学院孟买分校);Technical University of Munich(慕尼黑工业大学)

Sai Bhargav Rongali, Sarthak Mehrotra, Ankit Jha, Mohamad Hassan N C, Shirsha Bose, Tanisha Gupta, Mainak Singha, Biplab Banerjee

AI总结:

提出跨域广义类别发现 AD-GCD 设定及 CDAD-Net,通过熵驱动对抗学习、度量融合、邻域对比与条件图像修复缓解域偏移,性能提升8–15%。

摘要:

在广义类别发现(GCD)中,我们利用已知类别的训练数据集,对已知类别和新类别的无标签样本进行聚类。由于这些数据集之间存在域偏移,一个突出挑战随之产生。为解决该问题,我们提出了一种新设定:跨域广义类别发现(AD-GCD),并推出 CDAD-NET(Class Discoverer Across Domains,跨域类别发现器)作为解决方案。CDAD-NET 的架构旨在同步有标签(源)数据集和无标签(目标)数据集中潜在的已知类别样本,同时强调目标数据的独特分类。为实现这一点,我们提出了一种熵驱动的对抗学习策略,该策略考虑目标样本相对于源域类别原型的距离分布。同时,通过融合三个度量学习目标来维持共享空间的判别性。在源域中,我们专注于优化样本与其所属类别原型之间的邻近性;而在目标域中,我们集成了一种以邻域为中心的对比学习机制,并辅以精巧的邻居挖掘方法。为进一步突出语义对齐图像之间细微的特征关联,我们倡导条件图像修复的概念,强调如下前提:语义相似的图像比彼此脱节的图像对该任务更有效。实验方面,在我们提出的三个 AD-GCD 基准上,CDAD-NET 超越现有文献,性能提升 8–15%。

英文摘要:

In Generalized Category Discovery (GCD), we cluster unlabeled samples of known and novel classes, leveraging a training dataset of known classes. A salient challenge arises due to domain shifts between these datasets. To address this, we present a novel setting: Across Domain Generalized Category Discovery (AD-GCD) and bring forth CDAD-NET (Class Discoverer Across Domains) as a remedy. CDAD-NET is architected to synchronize potential known class samples across both the labeled (source) and unlabeled (target) datasets, while emphasizing the distinct categorization of the target data. To facilitate this, we propose an entropy-driven adversarial learning strategy that accounts for the distance distributions of target samples relative to source-domain class prototypes. Parallelly, the discriminative nature of the shared space is upheld through a fusion of three metric learning objectives. In the source domain, our focus is on refining the proximity between samples and their affiliated class prototypes, while in the target domain, we integrate a neighborhood-centric contrastive learning mechanism, enriched with an adept neighborsmining approach. To further accentuate the nuanced feature interrelation among semantically aligned images, we champion the concept of conditional image inpainting, underscoring the premise that semantically analogous images prove more efficacious to the task than their disjointed counterparts. Experimentally, CDAD-NET eclipses existing literature with a performance increment of 8-15% on three AD-GCD benchmarks we present.

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

↑