用双曲几何建模标签关系,提升单正样本多标签学习的准确性与可解释性。
Hyperbolic Structured Classification for Robust Single Positive Multi-label Learning
- 将标签表示为双曲球体,通过球体交互显式建模包含、重叠、分离等多重关系。
- 在四个数据集上表现优于现有方法,且嵌入结果与真实共现模式高度相关。
- 适合关注标签结构建模与模型可解释性的研究者,尤其适用于层次化标注场景。
单正样本多标签学习(SPMLL)处理每样本仅标注一个正标签但可能属于多个类别的情况,难以捕捉复杂的标签关系与层级结构。现有方法虽通过距离相似性隐式建模标签关系,但缺乏对不同关系类型的显式几何定义。为此,我们提出首个面向SPMLL的双曲分类框架,将每个标签表示为双曲球而非点或向量,通过球体间的几何交互自然建模多种关系:包含关系用于表达层级结构,重叠关系用于共现模式,分离关系用于语义独立性。进一步引入温度自适应双曲球分类器和受物理启发的双阱正则化,引导球体趋向有意义的配置。在四个基准数据集(MS-COCO、PASCAL VOC、NUS-WIDE、CUB-200-2011)上的大量实验表明,该方法性能具有竞争力,且可解释性显著优于现有方法。统计分析显示,学习到的嵌入与真实世界共现模式强相关,验证了双曲几何在不完整监督下的结构化分类中更具鲁棒性。
原文摘要 · Abstract (English)
Single Positive Multi-Label Learning (SPMLL) addresses the challenging scenario where each training sample is annotated with only one positive label despite potentially belonging to multiple categories, making it difficult to capture complex label relationships and hierarchical structures. While existing methods implicitly model label relationships through distance-based similarity, lacking explicit geometric definitions for different relationship types. To address these limitations, we propose the first hyperbolic classification framework for SPMLL that represents each label as a hyperbolic ball rather than a point or vector, enabling rich inter-label relationship modeling through geometric ball interactions. Our ball-based approach naturally captures multiple relationship types simultaneously: inclusion for hierarchical structures, overlap for co-occurrence patterns, and separation for semantic independence. Further, we introduce two key component innovations: a temperature-adaptive hyperbolic ball classifier and a physics-inspired double-well regularization that guides balls toward meaningful configurations. To validate our approach, extensive experiments on four benchmark datasets (MS-COCO, PASCAL VOC, NUS-WIDE, CUB-200-2011) demonstrate competitive performance with superior interpretability compared to existing methods. Furthermore, statistical analysis reveals strong correlation between learned embeddings and real-world co-occurrence patterns, establishing hyperbolic geometry as a more robust paradigm for structured classification under incomplete supervision.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。