LLM2D
通过利用标签分布改进多标签对比学习
Improving Multi-Label Contrastive Learning by Leveraging Label Distribution
作者: Ning Chen, Shen-Huan Lyu, Tian-Shuang Wu, Yanyan Wang, Bin Tang
发布日期: 2/3/2025
arXiv ID: oai:arXiv.org:2501.19145v1

摘要

arXiv:2501.19145v1 宣告类型: 跨越性 摘要: 在多标签学习中,利用对比学习学习更好的表示面临着一个关键挑战:选择正样本和负样本以及有效地利用标签信息。之前的研究基于标签之间的重叠来选择正样本和负样本,并使用它们来实现标签级别的损失平衡。然而,这些方法遭受了复杂的选择过程,并且未能考虑不同标签的重要性差异。为了解决这些问题,我们提出了一种新方法,通过标签分布改进多标签对比学习。具体来说,在选择正样本和负样本时,我们只需要考虑标签之间是否有交集。为了建模标签之间的关系,我们引入了两种方法,分别基于径向基函数(RBF)和对比损失从逻辑标签恢复标签分布。我们在包括图像和向量数据在内的九个广泛使用的多标签数据集上评估了我们的方法。结果表明,我们的方法在六个评估指标中均优于现有最佳方法。