Automatically Discovering Novel Visual Categories With Adaptive Prototype Learning

This article targets the task of novel category discovery (NCD), which aims to discover unknown categories when a certain number of classes are already known. The NCD task is challenging due to its closeness to real-world scenarios, where we have only encountered some partial classes and corresponding images. Unlike previous approaches to NCD, we propose a novel adaptive prototype learning method that leverages prototypes to emphasize category discrimination and alleviate the issue of missing annotations for novel classes. Concretely, the proposed method consists of two main stages: prototypical representation learning and prototypical self-training. In the first stage, we develop a robust feature extractor that could effectively handle images from both base and novel categories. This ability of instance and category discrimination of the feature extractor is boosted by self-supervised learning and adaptive prototypes. In the second stage, we utilize the prototypes again to rectify offline pseudo labels and train a final parametric classifier for category clustering. We conduct extensive experiments on four benchmark datasets, demonstrating our method's effectiveness and robustness with state-of-the-art performance.

Medienart:

E-Artikel

Erscheinungsjahr:

2024

Erschienen:

2024

Enthalten in:

Zur Gesamtaufnahme - volume:46

Enthalten in:

IEEE transactions on pattern analysis and machine intelligence - 46(2024), 4 vom: 23. März, Seite 2533-2544

Sprache:

Englisch

Beteiligte Personen:

Zhang, Lu [VerfasserIn]
Qi, Lu [VerfasserIn]
Yang, Xu [VerfasserIn]
Qiao, Hong [VerfasserIn]
Yang, Ming-Hsuan [VerfasserIn]
Liu, Zhiyong [VerfasserIn]

Links:

Volltext

Themen:

Journal Article

Anmerkungen:

Date Revised 07.03.2024

published: Print-Electronic

Citation Status PubMed-not-MEDLINE

doi:

10.1109/TPAMI.2023.3335962

funding:

Förderinstitution / Projekttitel:

PPN (Katalog-ID):

NLM364864168