arXivDaily arXiv每日学术速递 周一至周五更新
​通知:联邦节假日调休:哥伦布日 / 原住民日(Columbus Day / Indigenous Peoples' Day),今日(13日)无更新。
arXiv 2410.04959cs.LGstat.ML

抗坍缩的非对比自监督学习

Collapse-Proof Non-Contrastive Self-Supervised Learning

KU Leuven(鲁汶大学)

Emanuele Sansone, Tim Lebailly, Tinne Tuytelaars

AI总结:

本文提出一种基于超维计算的非对比自监督学习设计,通过引入归纳偏置同时实现表示去相关与聚类,从而避免多种训练坍缩模式并提升泛化性能。

摘要:

我们基于超维计算,为非对比自监督学习提出了一种原理化且简化的投影器与损失函数设计。我们从理论上证明,该设计引入了一种归纳偏置,能够在无需显式强制这些性质的情况下,促使表示同时实现去相关和聚类。该偏置可证明地增强泛化能力,并足以避免已知的训练失败模式,例如表示坍缩、维度坍缩、簇坍缩和簇内坍缩。我们在包括 SVHN、CIFAR-10、CIFAR-100 和 ImageNet-100 在内的图像数据集上验证了理论发现。我们的方法有效结合了特征去相关和基于聚类的自监督学习方法的优势,克服了训练失败模式,同时在聚类和线性分类任务中实现了强泛化能力。

英文摘要:

We present a principled and simplified design of the projector and loss function for non-contrastive self-supervised learning based on hyperdimensional computing. We theoretically demonstrate that this design introduces an inductive bias that encourages representations to be simultaneously decorrelated and clustered, without explicitly enforcing these properties. This bias provably enhances generalization and suffices to avoid known training failure modes, such as representation, dimensional, cluster, and intracluster collapses. We validate our theoretical findings on image datasets, including SVHN, CIFAR-10, CIFAR-100, and ImageNet-100. Our approach effectively combines the strengths of feature decorrelation and cluster-based self-supervised learning methods, overcoming training failure modes while achieving strong generalization in clustering and linear classification tasks.

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

↑