SelEx: Self-Expertise in Fine-Grained Generalized Category Discovery

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Rastegar, Sarah, Salehi, Mohammadreza, Asano, Yuki M., Doughty, Hazel, Snoek, Cees G. M.
Format: Preprint
Published: 2024
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866909381863407616
author Rastegar, Sarah
Salehi, Mohammadreza
Asano, Yuki M.
Doughty, Hazel
Snoek, Cees G. M.
author_facet Rastegar, Sarah
Salehi, Mohammadreza
Asano, Yuki M.
Doughty, Hazel
Snoek, Cees G. M.
contents In this paper, we address Generalized Category Discovery, aiming to simultaneously uncover novel categories and accurately classify known ones. Traditional methods, which lean heavily on self-supervision and contrastive learning, often fall short when distinguishing between fine-grained categories. To address this, we introduce a novel concept called `self-expertise', which enhances the model's ability to recognize subtle differences and uncover unknown categories. Our approach combines unsupervised and supervised self-expertise strategies to refine the model's discernment and generalization. Initially, hierarchical pseudo-labeling is used to provide `soft supervision', improving the effectiveness of self-expertise. Our supervised technique differs from traditional methods by utilizing more abstract positive and negative samples, aiding in the formation of clusters that can generalize to novel categories. Meanwhile, our unsupervised strategy encourages the model to sharpen its category distinctions by considering within-category examples as `hard' negatives. Supported by theoretical insights, our empirical results showcase that our method outperforms existing state-of-the-art techniques in Generalized Category Discovery across several fine-grained datasets. Our code is available at: https://github.com/SarahRastegar/SelEx.
format Preprint
id arxiv_https___arxiv_org_abs_2408_14371
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle SelEx: Self-Expertise in Fine-Grained Generalized Category Discovery
Rastegar, Sarah
Salehi, Mohammadreza
Asano, Yuki M.
Doughty, Hazel
Snoek, Cees G. M.
Computer Vision and Pattern Recognition
Artificial Intelligence
Machine Learning
In this paper, we address Generalized Category Discovery, aiming to simultaneously uncover novel categories and accurately classify known ones. Traditional methods, which lean heavily on self-supervision and contrastive learning, often fall short when distinguishing between fine-grained categories. To address this, we introduce a novel concept called `self-expertise', which enhances the model's ability to recognize subtle differences and uncover unknown categories. Our approach combines unsupervised and supervised self-expertise strategies to refine the model's discernment and generalization. Initially, hierarchical pseudo-labeling is used to provide `soft supervision', improving the effectiveness of self-expertise. Our supervised technique differs from traditional methods by utilizing more abstract positive and negative samples, aiding in the formation of clusters that can generalize to novel categories. Meanwhile, our unsupervised strategy encourages the model to sharpen its category distinctions by considering within-category examples as `hard' negatives. Supported by theoretical insights, our empirical results showcase that our method outperforms existing state-of-the-art techniques in Generalized Category Discovery across several fine-grained datasets. Our code is available at: https://github.com/SarahRastegar/SelEx.
title SelEx: Self-Expertise in Fine-Grained Generalized Category Discovery
topic Computer Vision and Pattern Recognition
Artificial Intelligence
Machine Learning
url https://arxiv.org/abs/2408.14371