arXiv:2410.23200cs.CV2024-10被引 7

通过挖掘自监督学习中的层级结构,提升模型表现。

HEX: Hierarchical Emergence Exploitation in Self-Supervised Algorithms

  • 在InfoNCE损失中引入自适应分解,分别处理局部与全局维度坍塌。
  • 在ImageNet上实现最高5.6%的分类准确率提升,100轮训练下有效。
  • 适用于多种自监督算法,尤其适合关注表示学习层次性的研究者。

本文提出一种可应用于多种自监督学习(SSL)方法的算法,利用训练过程中自然涌现的层级结构。现有SSL方法通常依赖不变性项保持相似样本的一致性,并通过正则化项防止全局维度坍塌(即数据表示退化到低维子空间)。近期研究表明,随着训练进行,表示空间逐渐显现出语义层级结构:同一层级内的样本因共享特征而表现出更强的局部维度坍塌。理想情况下,应引入额外正则化项来建模这种局部坍塌效应。然而现有方法未考虑此特性。为此,我们提出一种自适应算法,将InfoNCE损失分母分解为局部层级与全局坍塌正则化两项,并基于批次内样本余弦相似度分布动态调整阈值,逐步反映表示空间的层级演化。实验表明,该层级涌现利用(HEX)方法可广泛集成于各类SSL框架,在ImageNet上以100轮训练实现最高5.6%的相对准确率提升。

原文摘要 · Abstract (English)

In this paper, we propose an algorithm that can be used on top of a wide variety of self-supervised (SSL) approaches to take advantage of hierarchical structures that emerge during training. SSL approaches typically work through some invariance term to ensure consistency between similar samples and a regularization term to prevent global dimensional collapse. Dimensional collapse refers to data representations spanning a lower-dimensional subspace. Recent work has demonstrated that the representation space of these algorithms gradually reflects a semantic hierarchical structure as training progresses. Data samples of the same hierarchical grouping tend to exhibit greater dimensional collapse locally compared to the dataset as a whole due to sharing features in common with each other. Ideally, SSL algorithms would take advantage of this hierarchical emergence to have an additional regularization term to account for this local dimensional collapse effect. However, the construction of existing SSL algorithms does not account for this property. To address this, we propose an adaptive algorithm that performs a weighted decomposition of the denominator of the InfoNCE loss into two terms: local hierarchical and global collapse regularization respectively. This decomposition is based on an adaptive threshold that gradually lowers to reflect the emerging hierarchical structure of the representation space throughout training. It is based on an analysis of the cosine similarity distribution of samples in a batch. We demonstrate that this hierarchical emergence exploitation (HEX) approach can be integrated across a wide variety of SSL algorithms. Empirically, we show performance improvements of up to 5.6% relative improvement over baseline SSL approaches on classification accuracy on Imagenet with 100 epochs of training.

自监督学习层级结构表示学习InfoNCE

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。