Skip to main navigation Skip to search Skip to main content

Learning Intrinsic Hierarchy for Generalized Category Discovery

  • Yu Duan
  • , Junzhi He
  • , Zhanxuan Hu
  • , Mengda Ji
  • , Rong Wang
  • , Quanxue Gao
  • Xidian University
  • Northwestern Polytechnical University Xian
  • Yunnan Normal University

Research output: Contribution to journalConference articlepeer-review

Abstract

Generalized Category Discovery (GCD) aims to classify unlabeled data by leveraging knowledge from labeled categories. While existing methods have achieved remarkable progress, they often treat images as flat feature sets, neglecting the intrinsic hierarchy: where key objects dominate meaning and backgrounds serve as context. For instance, in images of a dog either standing on grass or lying on a bed, the dog remains the central semantic element, whereas the background varies. Motivated by this, we propose LEArning Intrinsic Hierarchy (LEAH), a lightweight module designed to model hierarchical structure within images. LEAH consists of two components: a pruner that filters task irrelevant tokens to extract key objects, and a constructor that embeds key objects and full images into hyperbolic space using adaptive entailment cones to capture compositional semantics. LEAH can be easily integrated into existing GCD frameworks with minimal modification. When applied to SimGCD, it achieves up to 13.2% accuracy improvement on fine-grained benchmarks, demonstrating its effectiveness in discovering subtle inter-class differences through hierarchical modeling.

Original languageEnglish
Pages (from-to)20950-20958
Number of pages9
JournalProceedings of the AAAI Conference on Artificial Intelligence
Volume40
Issue number25
DOIs
StatePublished - 2026
Event40th AAAI Conference on Artificial Intelligence, AAAI 2026 - Singapore, Singapore
Duration: 20 Jan 202627 Jan 2026

Fingerprint

Dive into the research topics of 'Learning Intrinsic Hierarchy for Generalized Category Discovery'. Together they form a unique fingerprint.

Cite this