Self-Paced Unified Representation Learning for Hierarchical Multi-Label Classification
Zixuan Yuan, Hao Liu, Haoyi Zhou, Denghui Zhang, Xiao Zhang, Hao Wang, Hui Xiong
Abstract
Hierarchical Multi-Label Classification (HMLC) is a wellestablished problem that aims at assigning data instances to multiple classes stored in a hierarchical structure. Despite its importance, existing approaches often face two key limitations: (i) They employ dense networks to solely explore the class hierarchy as hard criterion for maintaining taxonomic consistency among predicted classes, yet without leveraging rich semantic relationships between instances and classes; (ii) They struggle to generalize in settings with deep class levels, since the mini-batches uniformly sampled from different levels ignore the varying complexities of data and result in a non-smooth model adaptation to sparse data. To mitigate these issues, we present a Self-Paced Unified Representation (SPUR) learning framework, which focuses on the interplay between instance and classes to flexibly organize the training process of HMLC algorithms. Our framework consists of two lightweight encoders designed to capture the semantics of input features and the topological information of the class hierarchy. These encoders generate unified embeddings of instances and class hierarchy, which enable SPUR to exploit semantic dependencies between them and produce predictions in line with taxonomic constraints. Furthermore, we introduce a dynamic hardness measurement strategy that considers both class hierarchy and instance features to estimate the learning difficulty of each instance. This strategy is achieved by incorporating the propagation loss obtained at each hierarchical level, allowing for a more comprehensive assessment of learning complexity. Extensive experiments on several empirical benchmarks demonstrate the effectiveness and efficiency of SPUR compared to state-of-the-art methods, especially in scenarios with missing features.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Your agent calls
Luneget_paper_fulltext
Free to start. No credit card required.
Terminal
Install the CLIlune papers fulltext 5e857c96-782a-4113-b33f-e4e6e3fe4b14Cited by top-tier papers2
- CoPINN: Cognitive Physics-Informed Neural NetworksSiyuan Duan, Wenyuan Wu, Peng Hu, Zhenwen Ren et al.ICML 2025
- HierUQ: Hierarchical Uncertainty Quantification with Adaptive Granularity Reconciliation for Degraded Image ClassificationYANG CHU, Xiaomeng Yang, Keli Deng, Yuntao QianCVPR 2026
Builds on4
- Fast Multi-Resolution Transformer Fine-tuning for Extreme Multi-label Text ClassificationJiong Zhang, Wei-Cheng Chang, Hsiang-Fu Yu, Inderjit S. DhillonNeurIPS 2021 · 147 citations
- Coherent Hierarchical Multi-Label Classification NetworksEleonora Giunchiglia, Thomas LukasiewiczNeurIPS 2020 · 142 citations
- ECLARE: Extreme Classification with Label Graph CorrelationsAnshul Mittal, Noveen Sachdeva, Sheshansh Agrawal, Sumeet Agarwal et al.WWW 2021 · 71 citations
- Modeling Label Space Interactions in Multi-label Classification using Box EmbeddingsDhruvesh Patel, Pavitra Dangati, Jay-Yoon Lee, Michael Boratko et al.ICLR 2022 · 28 citations
Related papers
- Hierarchical Multi-label Text Classification with Horizontal and Vertical Category CorrelationsLinli Xu, Sijie Teng, Ruoyu Zhao, Junliang Guo et al.EMNLP 2021 · 22 citations
- Use All The Labels: A Hierarchical Multi-Label Contrastive Learning FrameworkShu Zhang, Ran Xu, Caiming Xiong, Chetan RamaiahCVPR 2022 · 71 citations
- Multi-Dimensional Classification via Sparse Label EncodingBin-Bin Jia, Min-Ling ZhangICML 2021 · 24 citations
- Incorporating Hierarchy into Text Encoder: a Contrastive Learning Approach for Hierarchical Text ClassificationZihan Wang, Peiyi Wang, Lianzhe Huang, Xin Sun et al.ACL 2022 · 157 citations
- Batch Selection for Multi-Label Classification Guided by Uncertainty and Dynamic Label CorrelationsAo Zhou, Bin Liu, Jin Wang, Grigorios TsoumakasAAAI 2025 · 1 citation
