Use All The Labels: A Hierarchical Multi-Label Contrastive Learning Framework
Not on the map. Why?
Abstract (arXiv)
Current contrastive learning frameworks focus on leveraging a single supervisory signal to learn representations, which limits the efficacy on unseen data and downstream tasks. In this paper, we present a hierarchical multi-label representation learning framework that can leverage all available labels and preserve the hierarchical relationship between classes. We introduce novel hierarchy preserving losses, which jointly apply a hierarchical penalty to the contrastive loss, and enforce the hierarchy constraint. The loss function is data driven and automatically adapts to arbitrary multi-label structures. Experiments on several datasets show that our relationship-preserving embedding performs well on a variety of tasks and outperform the baseline supervised and self-supervised approaches. Code is available at https://github.com/salesforce/hierarchicalContrastiveLearning.
Key facts. Cited by 59 papers in the map (data through 14 Oct 2026). Builds on 32 of its 52 references that are in the map. First posted on arXiv 27 Apr 2022.
Builds on
32 of 52 references.
References · 32
- 2015Deep Residual Learning for Image Recognitioncited by 52,096
- 2019PyTorch: An Imperative Style, High-Performance Deep Learning Librarycited by 20,305
- 2020A Simple Framework for Contrastive Learning of Visual Representationscited by 12,049
- 2014ImageNet Large Scale Visual Recognition Challengecited by 8,498
- 2018Representation Learning with Contrastive Predictive Codingcited by 7,777
- 2019Momentum Contrast for Unsupervised Visual Representation Learningcited by 7,513
- 2015FaceNet: A Unified Embedding for Face Recognition and Clusteringcited by 2,849
- 2020Supervised Contrastive Learningcited by 2,829
- 2018Unsupervised Feature Learning via Non-Parametric Instance-level Discriminationcited by 1,845
- 2019Contrastive Multiview Codingcited by 1,301
- 2018Learning deep representations by mutual information estimation and maximizationcited by 1,273
- 2015Unsupervised Visual Representation Learning by Context Predictioncited by 1,066
20 more
- 2016An overview of gradient descent optimization algorithmscited by 1,033
- 2019Data-Efficient Image Recognition with Contrastive Predictive Codingcited by 673
- 2020Prototypical Contrastive Learning of Unsupervised Representationscited by 640
- 2020Dense Contrastive Learning for Self-Supervised Visual Pre-Trainingcited by 512
- 2020A Metric Learning Reality Checkcited by 321
- 2019On Mutual Information Maximization for Representation Learningcited by 316
- 2017Sampling Matters in Deep Embedding Learningcited by 285
- 2016Split-Brain Autoencoders: Unsupervised Learning by Cross-Channel Predictioncited by 217
- 2021VideoMoCo: Contrastive Video Representation Learning with Temporally Adversarial Examplescited by 154
- 2019Making Better Mistakes: Leveraging Class Hierarchies with Deep Networkscited by 113
- 2018Deep Metric Learning with Hierarchical Triplet Losscited by 112
- 2020Image Representations Learned With Unsupervised Pre-Training Contain Human-like Biasescited by 107
- 2019MultiGrain: a unified image embedding for classes and instancescited by 85
- 2019Ranked List Loss for Deep Metric Learningcited by 77
- 2020Coherent Hierarchical Multi-Label Classification Networkscited by 65
- 2017Do Convolutional Neural Networks Learn Class Hierarchy?cited by 61
- 2020Multi-Label Contrastive Learning for Abstract Visual Reasoningcited by 24
- 2020Leveraging Class Hierarchies with Metric-Guided Prototype Learningcited by 20
- 2020Exploit Clues from Views: Self-Supervised and Regularized Learning for Multiview Object Recognitioncited by 6
- 2021Multi-level Distance Regularization for Deep Metric Learningcited by 6
Built on by
Citing papers per year: 2022: 5; 2023: 13; 2024: 14; 2025: 17; 2026: 10.
59 papers.
Sort:
By area:
- 2023BioCLIP: A Vision Foundation Model for the Tree of Lifecited by 161
- 2023OmniSeg3D: Omniversal 3D Segmentation via Hierarchical Contrastive Learningcited by 75
- 2024SelEx: Self-Expertise in Fine-Grained Generalized Category Discoverycited by 34
- 2022OpenCon: Open-world Contrastive Learningcited by 33
- 2024Multi-Label Guided Soft Contrastive Learning for Efficient Earth Observation Pretrainingcited by 28
- 2023MaskCon: Masked Contrastive Learning for Coarse-Labelled Datasetcited by 21
- 2023TransHP: Image Classification with Hierarchical Promptingcited by 16
- 2024Visually Consistent Hierarchical Image Classificationcited by 16
- 2023Hierarchical discriminative learning improves visual representations of biomedical microscopycited by 12
- 2022An Effective Deployment of Contrastive Learning in Multi-label Text Classificationcited by 9
- 2024QUEEN: Query Unlearning against Model Extractioncited by 9
- 2022Exploiting Contrastive Learning and Numerical Evidence for Confusing Legal Judgment Predictioncited by 8
38 more
- 2023End-to-End Supervised Multilabel Contrastive Learningcited by 8
- 2023Joint Multiple Intent Detection and Slot Filling with Supervised Contrastive Learning and Self-Distillationcited by 7
- 2023Introspective Deep Metric Learningcited by 7
- 2024HEX: Hierarchical Emergence Exploitation in Self-Supervised Algorithmscited by 7
- 2025SEAL: Semantic-Aware Hierarchical Learning for Generalized Category Discoverycited by 7
- 2023Revisiting Logistic-softmax Likelihood in Bayesian Meta-Learning for Few-Shot Classificationcited by 6
- 2023Instances and Labels: Hierarchy-aware Joint Supervised Contrastive Learning for Hierarchical Multi-Label Text Classificationcited by 4
- 2024Exploring Contrastive Learning for Long-Tailed Multi-Label Text Classificationcited by 4
- 2025Class Prototypes based Contrastive Learning for Classifying Multi-Label and Fine-Grained Educational Videoscited by 4
- 2023Incorporating Domain Knowledge Graph into Multimodal Movie Genre Classification with Self-Supervised Attention and Contrastive Learningcited by 3
- 2024ProbMCL: Simple Probabilistic Contrastive Learning for Multi-label Visual Classificationcited by 3
- 2024Learning Label Hierarchy with Supervised Contrastive Learningcited by 3
- 2024Language Model as Visual Explainercited by 3
- 2025Can we make NeRF-based visual localization privacy-preserving?cited by 3
- 2025Robult: Leveraging Redundancy and Modality Specific Features for Robust Multimodal Learningcited by 3
- 2025Free-Grained Hierarchical Visual Recognitioncited by 3
- 2023Multi-Similarity Contrastive Learningcited by 2
- 2024Acoustic identification of individual animals with hierarchical contrastive learningcited by 2
- 2024Similarity-Dissimilarity Loss for Multi-label Supervised Contrastive Learningcited by 2
- 2025Skin Lesion Phenotyping via Nested Multi-modal Contrastive Learningcited by 2
- 2025Generalized Fine-Grained Category Discovery with Multi-Granularity Conceptual Expertscited by 2
- 2025Boosting Medical Visual Understanding From Multi-Granular Language Learningcited by 2
- 2022TagRec++: Hierarchical Label Aware Attention Network for Question Categorizationcited by 1
- 2024Veri-Car: Towards Open-world Vehicle Information Retrievalcited by 1
- 2025Improving Multi-Label Contrastive Learning by Leveraging Label Distributioncited by 1
- 2025Multi-level Supervised Contrastive Learningcited by 1
- 2025Correcting Noisy Multilabel Predictions: Modeling Label Noise through Latent Space Shiftscited by 1
- 2025Multi-Grained Compositional Visual Clue Learning for Image Intent Recognitioncited by 1
- 2026MATANet: A Multi-context Attention and Taxonomy-Aware Network for Fine-Grained Underwater Recognition of Marine Speciescited by 1
- 2026Learning Hierarchical Knowledge in Text-Rich Networks with Taxonomy-Informed Representation Learningcited by 1
- 2026Hierarchical, Interpretable, Label-Free Concept Bottleneck Modelcited by 1
- 2026Cross-Layer Misalignment Detection in Agent Skills: A Progressive Loading-Aware Contrastive Learning Approachcited by 1
- 2022Large-Scale Product Retrieval with Weakly Supervised Representation Learningcited by 0
- 2023Contrastive Label Enhancementcited by 0
- 2024Computer User Interface Understanding. A New Dataset and a Learning Frameworkcited by 0
- 2024Automatic Alignment of Discourse Relations of Different Discourse Annotation Frameworkscited by 0
- 2025Hierarchical Multi-Label Contrastive Learning for Protein-Protein Interaction Prediction Across Organismscited by 0
- 2025ModAn-MulSupCon: Modality-and Anatomy-Aware Multi-Label Supervised Contrastive Pretraining for Medical Imagingcited by 0
50 most cited of 59 All 59 (JSON)
- 2026Adapting Dense Vision-Language Relationships for Multi-label Classification with Partial Label23 Aug 2026
- 2026Cross-Layer Misalignment Detection in Agent Skills: A Progressive Loading-Aware Contrastive Learning Approach12 Jul 2026
- 2026Online Continual Learning with Dynamic Label Hierarchies12 May 2026
- 2026Hierarchical, Interpretable, Label-Free Concept Bottleneck Model2 Apr 2026
- 2026MAPLE: Multi-Path Adaptive Propagation with Level-Aware Embeddings for Hierarchical Multi-Label Image Classification31 Mar 2026
- 2026Hierarchy-Guided Multimodal Representation Learning for Taxonomic Inference26 Mar 2026
- 2026Learning Human-Object Interaction for 3D Human Pose Estimation from LiDAR Point Clouds17 Mar 2026
- 2026HELM: Hierarchical and Explicit Label Modeling with Graph Learning for Multi-Label Image Classification12 Mar 2026
- 2026Learning Hierarchical Knowledge in Text-Rich Networks with Taxonomy-Informed Representation Learning9 Mar 2026
- 2026MATANet: A Multi-context Attention and Taxonomy-Aware Network for Fine-Grained Underwater Recognition of Marine Species7 Jan 2026
- 2025Balanced Hierarchical Contrastive Learning with Decoupled Queries for Fine-grained Object Detection in Remote Sensing Images30 Dec 2025
- 2025Boosting Medical Visual Understanding From Multi-Granular Language Learning20 Nov 2025
38 more
- 2025Climbing the label tree: Hierarchy-preserving contrastive learning for medical imaging5 Nov 2025
- 2025SEAL: Semantic-Aware Hierarchical Learning for Generalized Category Discovery21 Oct 2025
- 2025Free-Grained Hierarchical Visual Recognition16 Oct 2025
- 2025Class Prototypes based Contrastive Learning for Classifying Multi-Label and Fine-Grained Educational Videos13 Oct 2025
- 2025Feature Identification for Hierarchical Contrastive Learning1 Oct 2025
- 2025Generalized Fine-Grained Category Discovery with Multi-Granularity Conceptual Experts30 Sep 2025
- 2025Robult: Leveraging Redundancy and Modality Specific Features for Robust Multimodal Learning3 Sep 2025
- 2025Can we make NeRF-based visual localization privacy-preserving?26 Aug 2025
- 2025ModAn-MulSupCon: Modality-and Anatomy-Aware Multi-Label Supervised Contrastive Pretraining for Medical Imaging26 Aug 2025
- 2025Hierarchical Multi-Label Contrastive Learning for Protein-Protein Interaction Prediction Across Organisms3 Jul 2025
- 2025Skin Lesion Phenotyping via Nested Multi-modal Contrastive Learning29 May 2025
- 2025Multi-Grained Compositional Visual Clue Learning for Image Intent Recognition25 Apr 2025
- 2025Correcting Noisy Multilabel Predictions: Modeling Label Noise through Latent Space Shifts20 Feb 2025
- 2025Multi-level Supervised Contrastive Learning4 Feb 2025
- 2025Improving Multi-Label Contrastive Learning by Leveraging Label Distribution31 Jan 2025
- 2024Language Model as Visual Explainer8 Dec 2024
- 2024Veri-Car: Towards Open-world Vehicle Information Retrieval11 Nov 2024
- 2024HEX: Hierarchical Emergence Exploitation in Self-Supervised Algorithms30 Oct 2024
- 2024Similarity-Dissimilarity Loss for Multi-label Supervised Contrastive Learning17 Oct 2024
- 2024Acoustic identification of individual animals with hierarchical contrastive learning13 Sep 2024
- 2024SelEx: Self-Expertise in Fine-Grained Generalized Category Discovery26 Aug 2024
- 2024QUEEN: Query Unlearning against Model Extraction1 Jul 2024
- 2024Visually Consistent Hierarchical Image Classification17 Jun 2024
- 2024Multi-Label Guided Soft Contrastive Learning for Efficient Earth Observation Pretraining30 May 2024
- 2024Exploring Contrastive Learning for Long-Tailed Multi-Label Text Classification12 Apr 2024
- 2024Automatic Alignment of Discourse Relations of Different Discourse Annotation Frameworks29 Mar 2024
- 2024Computer User Interface Understanding. A New Dataset and a Learning Framework15 Mar 2024
- 2024Learning Label Hierarchy with Supervised Contrastive Learning31 Jan 2024
- 2024ProbMCL: Simple Probabilistic Contrastive Learning for Multi-label Visual Classification2 Jan 2024
- 2023BioCLIP: A Vision Foundation Model for the Tree of Life30 Nov 2023
- 2023OmniSeg3D: Omniversal 3D Segmentation via Hierarchical Contrastive Learning20 Nov 2023
- 2023Revisiting Logistic-softmax Likelihood in Bayesian Meta-Learning for Few-Shot Classification16 Oct 2023
- 2023Incorporating Domain Knowledge Graph into Multimodal Movie Genre Classification with Self-Supervised Attention and Contrastive Learning12 Oct 2023
- 2023Instances and Labels: Hierarchy-aware Joint Supervised Contrastive Learning for Hierarchical Multi-Label Text Classification8 Oct 2023
- 2023Introspective Deep Metric Learning11 Sep 2023
- 2023Joint Multiple Intent Detection and Slot Filling with Supervised Contrastive Learning and Self-Distillation28 Aug 2023
- 2023End-to-End Supervised Multilabel Contrastive Learning8 Jul 2023
- 2023Multi-Similarity Contrastive Learning6 Jul 2023
50 newest of 59 All 59 (JSON)
- 2024QUEEN: Query Unlearning against Model Extractioncited by 9
- 2024QUEEN: Query Unlearning against Model Extraction1 Jul 2024
- 2023BioCLIP: A Vision Foundation Model for the Tree of Lifecited by 161
- 2023OmniSeg3D: Omniversal 3D Segmentation via Hierarchical Contrastive Learningcited by 75
- 2024SelEx: Self-Expertise in Fine-Grained Generalized Category Discoverycited by 34
- 2022OpenCon: Open-world Contrastive Learningcited by 33
- 2024Multi-Label Guided Soft Contrastive Learning for Efficient Earth Observation Pretrainingcited by 28
- 2023MaskCon: Masked Contrastive Learning for Coarse-Labelled Datasetcited by 21
- 2023TransHP: Image Classification with Hierarchical Promptingcited by 16
- 2024Visually Consistent Hierarchical Image Classificationcited by 16
- 2023Hierarchical discriminative learning improves visual representations of biomedical microscopycited by 12
- 2022An Effective Deployment of Contrastive Learning in Multi-label Text Classificationcited by 9
- 2022Exploiting Contrastive Learning and Numerical Evidence for Confusing Legal Judgment Predictioncited by 8
- 2023End-to-End Supervised Multilabel Contrastive Learningcited by 8
12 most cited of 56 All 56 (JSON)
- 2026Adapting Dense Vision-Language Relationships for Multi-label Classification with Partial Label23 Aug 2026
- 2026Online Continual Learning with Dynamic Label Hierarchies12 May 2026
- 2026Hierarchical, Interpretable, Label-Free Concept Bottleneck Model2 Apr 2026
- 2026MAPLE: Multi-Path Adaptive Propagation with Level-Aware Embeddings for Hierarchical Multi-Label Image Classification31 Mar 2026
- 2026Hierarchy-Guided Multimodal Representation Learning for Taxonomic Inference26 Mar 2026
- 2026Learning Human-Object Interaction for 3D Human Pose Estimation from LiDAR Point Clouds17 Mar 2026
- 2026HELM: Hierarchical and Explicit Label Modeling with Graph Learning for Multi-Label Image Classification12 Mar 2026
- 2026Learning Hierarchical Knowledge in Text-Rich Networks with Taxonomy-Informed Representation Learning9 Mar 2026
- 2026MATANet: A Multi-context Attention and Taxonomy-Aware Network for Fine-Grained Underwater Recognition of Marine Species7 Jan 2026
- 2025Balanced Hierarchical Contrastive Learning with Decoupled Queries for Fine-grained Object Detection in Remote Sensing Images30 Dec 2025
- 2025Boosting Medical Visual Understanding From Multi-Granular Language Learning20 Nov 2025
- 2025Climbing the label tree: Hierarchy-preserving contrastive learning for medical imaging5 Nov 2025
12 newest of 56 All 56 (JSON)