arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2401.03312cs.CVcs.AI

利用数据层次结构作为对比学习的新模态

Exploiting Data Hierarchy as a New Modality for Contrastive Learning

Arjun Bhalla, Daniel Levenson, Jan Bernhard, Anton Abilov

AI总结:

提出一种利用三元组间隔损失表示数据空间层次的层次对比训练方法,验证了数据集结构作为弱监督学习新模态的有效性。

摘要:

本研究探讨了层次结构化数据如何帮助神经网络学习大教堂的概念表示。底层的WikiScenes数据集提供了大教堂组件的空间组织层次结构。我们提出了一种新颖的层次对比训练方法,利用三元组间隔损失(triplet margin loss)在编码器的潜在空间中表示数据的空间层次。因此,该方法探究了数据集结构是否为自监督学习提供了有价值的信息。我们应用t-SNE可视化生成的潜在空间,并通过在通用下游分类任务上与其他特定数据集的对比学习方法进行比较来评估该方法。该方法优于可比的弱监督和基线方法。研究结果表明,数据集结构是弱监督学习的重要模态。

英文摘要:

This work investigates how hierarchically structured data can help neural networks learn conceptual representations of cathedrals. The underlying WikiScenes dataset provides a spatially organized hierarchical structure of cathedral components. We propose a novel hierarchical contrastive training approach that leverages a triplet margin loss to represent the data's spatial hierarchy in the encoder's latent space. As such, the proposed approach investigates if the dataset structure provides valuable information for self-supervised learning. We apply t-SNE to visualize the resultant latent space and evaluate the proposed approach by comparing it with other dataset-specific contrastive learning methods using a common downstream classification task. The proposed method outperforms the comparable weakly-supervised and baseline methods. Our findings suggest that dataset structure is a valuable modality for weakly-supervised learning.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑