arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2408.04591cs.CVcs.AI

HiLo:一种对领域偏移具有鲁棒性的广义类别发现学习框架

HiLo: A Learning Framework for Generalized Category Discovery Robust to Domain Shifts

The University of Hong Kong(香港大学);University of Oxford(牛津大学)

Hongjun Wang, Sagar Vaze, Kai Han

AI总结:

针对广义类别发现中未标注数据存在领域偏移的问题,提出HiLo网络,通过分离高层语义与低层领域特征并最小化其互信息,结合领域增强与课程学习,在构建的新基准上大幅超越现有方法。

摘要:

广义类别发现(GCD)是一项具有挑战性的任务:给定一个部分标注的数据集,模型必须对所有未标注实例进行分类,无论它们属于已标注类别还是新类别。在本文中,我们对该任务中一个尚存的假设提出了挑战,即所有图像共享相同的领域。具体而言,我们引入了一项新任务和方法,以在未标注数据包含与已标注集不同领域的图像时处理GCD。我们提出的HiLo网络提取高层语义和低层领域特征,随后最小化这些表示之间的互信息。我们的直觉是,基于领域信息和语义信息的聚类应当是相互独立的。我们进一步通过专为GCD任务设计的特定领域增强以及课程学习方法扩展了该方法。最后,我们基于损坏的细粒度数据集构建了一个基准,并在具有真实世界领域偏移的DomainNet上进行了大规模评估,同时在该设置下重新实现了多个GCD基线方法。我们证明,HiLo在所有评估中均大幅优于当前最先进(SoTA)的类别发现模型。

英文摘要:

Generalized Category Discovery (GCD) is a challenging task in which, given a partially labelled dataset, models must categorize all unlabelled instances, regardless of whether they come from labelled categories or from new ones. In this paper, we challenge a remaining assumption in this task: that all images share the same domain. Specifically, we introduce a new task and method to handle GCD when the unlabelled data also contains images from different domains to the labelled set. Our proposed `HiLo' networks extract High-level semantic and Low-level domain features, before minimizing the mutual information between the representations. Our intuition is that the clusterings based on domain information and semantic information should be independent. We further extend our method with a specialized domain augmentation tailored for the GCD task, as well as a curriculum learning approach. Finally, we construct a benchmark from corrupted fine-grained datasets as well as a large-scale evaluation on DomainNet with real-world domain shifts, reimplementing a number of GCD baselines in this setting. We demonstrate that HiLo outperforms SoTA category discovery models by a large margin on all evaluations.

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

↑