Jul, 2023

端到端监督式多标签对比学习

TL;DR该研究提出了一种新的端到端训练框架 ——KMCL(基于核的多标签对比学习),以解决模型和数据中心设计的缺点。KMCL 通过将嵌入特征转化为高斯 RKHS 中指数核的混合,然后进行目标损失编码,包括重构损失、非对称分类损失和对比损失来捕捉标签相关性。该框架同时考虑特征编码器的不确定性,并保持较低的计算量。通过对图像分类任务进行广泛实验,展示了 KMCL 相对于 SOTA 方法的一致改进。PyTorch 实现详见 https://github.com/mahdihosseini/KMCL。