$\text{H}^2$em: Learning Hierarchical Hyperbolic Embeddings for Compositional Zero-Shot Learning
Lin Li, Jiahui Li, Jiaming Lei, Jun Xiao, Feifei Shao, Long Chen
TL;DR
CZSL requires recognizing unseen state–object pairs, a task hampered by Euclidean embeddings that poorly reflect large hierarchical taxonomies. H^2em introduces hierarchical hyperbolic embeddings in the Lorentz model, paired with a taxonomic entailment loss and a discriminative alignment loss, plus a Hyperbolic Cross-Modal Attention module to enable instance-aware fusion. The framework maps Euclidean CLIP features into hyperbolic space, enforcing semantic and conceptual hierarchies while distinguishing semantically similar compositions, and achieves state-of-the-art results in both closed-world and open-world CZSL across MIT-States, UT-Zappos, and CGQA. This work demonstrates the effectiveness of hyperbolic geometry for scalable, fine-grained compositional reasoning and offers a robust basis for extending hierarchical hyperbolic learning to other multimodal reasoning tasks.
Abstract
Compositional zero-shot learning (CZSL) aims to recognize unseen state-object compositions by generalizing from a training set of their primitives (state and object). Current methods often overlook the rich hierarchical structures, such as the semantic hierarchy of primitives (e.g., apple fruit) and the conceptual hierarchy between primitives and compositions (e.g, sliced apple apple). A few recent efforts have shown effectiveness in modeling these hierarchies through loss regularization within Euclidean space. In this paper, we argue that they fail to scale to the large-scale taxonomies required for real-world CZSL: the space's polynomial volume growth in flat geometry cannot match the exponential structure, impairing generalization capacity. To this end, we propose H2em, a new framework that learns Hierarchical Hyperbolic EMbeddings for CZSL. H2em leverages the unique properties of hyperbolic geometry, a space naturally suited for embedding tree-like structures with low distortion. However, a naive hyperbolic mapping may suffer from hierarchical collapse and poor fine-grained discrimination. We further design two learning objectives to structure this space: a Dual-Hierarchical Entailment Loss that uses hyperbolic entailment cones to enforce the predefined hierarchies, and a Discriminative Alignment Loss with hard negative mining to establish a large geodesic distance between semantically similar compositions. Furthermore, we devise Hyperbolic Cross-Modal Attention to realize instance-aware cross-modal infusion within hyperbolic geometry. Extensive ablations on three benchmarks demonstrate that H2em establishes a new state-of-the-art in both closed-world and open-world scenarios. Our codes will be released.
