Contrastive Representation Learning for Self-Supervised Taxonomy Completion

Abstract

Traditional Incremental Learning (IL) targets to handle sequential fully-supervised learning problems where novel classes emerge from time to time. However, due to inherent annotation uncertainty and ambiguity, collecting high-quality annotated data in a dynamic learning system can be extremely expensive. To mitigate this problem, we propose a novel weakly-supervised learning paradigm called Incremental Partial Label Learning (IPLL), where the sequentially arrived data relate to a set of candidate labels rather than the ground truth. Technically, we develop the Prototype-Guided Disambiguation and Replay Algorithm (PGDR) which leverages the class prototypes as a proxy to mitigate two intertwined challenges in IPLL, i.e., label ambiguity and catastrophic forgetting. To handle the former, PGDR encapsulates a momentum-based pseudo-labeling algorithm along with prototype-guided initialization, resulting in a balanced perception of classes. To alleviate forgetting, we develop a memory replay technique that collects well-disambiguated samples while maintaining representativeness and diversity. By jointly distilling knowledge from curated memory data, our framework exhibits a great disambiguation ability for samples of new tasks and achieves less forgetting of knowledge. Extensive experiments demonstrate that PGDR achieves superior performance over the baselines in the IPLL task.

Cite

Text

Niu et al. "Contrastive Representation Learning for Self-Supervised Taxonomy Completion." International Joint Conference on Artificial Intelligence, 2024. doi:10.24963/ijcai.2024/712

Markdown

[Niu et al. "Contrastive Representation Learning for Self-Supervised Taxonomy Completion." International Joint Conference on Artificial Intelligence, 2024.](https://mlanthology.org/ijcai/2024/niu2024ijcai-contrastive/) doi:10.24963/ijcai.2024/712

BibTeX

@inproceedings{niu2024ijcai-contrastive,
  title     = {{Contrastive Representation Learning for Self-Supervised Taxonomy Completion}},
  author    = {Niu, Yuhang and Xu, Hongyuan and Liu, Ciyi and Wen, Yanlong and Yuan, Xiaojie},
  booktitle = {International Joint Conference on Artificial Intelligence},
  year      = {2024},
  pages     = {6442-6450},
  doi       = {10.24963/ijcai.2024/712},
  url       = {https://mlanthology.org/ijcai/2024/niu2024ijcai-contrastive/}
}