arXiv:2603.03530cs.LGcs.AI2026-03被引 2

解释自监督学习为何少样本迁移能力强

Directional Neural Collapse Explains Few-Shot Transfer in Self-Supervised Learning

  • 用方向性决策轴方差衡量特征稳定性
  • 该量越小,少样本分类误差越低且多任务干扰越小
  • 适用于研究自监督表示泛化与多任务学习的学者

冻结的自监督表征在多个语义任务中仅用少量标签即可实现良好迁移。我们提出,一个几何量——方向性决策轴方差(directional CDNV)——是两种理想行为的核心:任务内强少样本迁移能力、跨任务低干扰。当类别分离方向上的变化较小时,这两种特性同时出现。首先,我们推导出多类分类的紧致非渐近泛化界,其主导项为方向性 CDNV;该界包含有限样本修正项,可清晰分离决策轴内在变异性与中心估计误差。其次,我们揭示决策轴坍缩与多任务几何的关系:对于独立平衡的标签分配,各任务间方向性 CDNV 小,则对应决策轴几乎正交,使单一表征能支持多个任务且干扰极小。实验表明,在多种自监督学习目标下,预训练过程中方向性 CDNV 会坍缩,而经典 CDNV 仍较大,且我们的泛化界能准确预测实际样本量下的少样本误差。此外,在合成多任务数据上,验证了自监督学习所学表征的决策轴近乎正交。代码与项目页面见 https://dlfundamentals.github.io/directional-neural-collapse/

原文摘要 · Abstract (English)

Frozen self-supervised representations often transfer well with only a few labels across many semantic tasks. We argue that a single geometric quantity, \emph{directional} CDNV (decision-axis variance), sits at the core of two favorable behaviors: strong few-shot transfer within a task, and low interference across many tasks. We show that both emerge when variability \emph{along} class-separating directions is small. First, we prove sharp non-asymptotic multiclass generalization bounds for downstream classification whose leading term is the directional CDNV. The bounds include finite-shot corrections that cleanly separate intrinsic decision-axis variability from centroid-estimation error. Second, we link decision-axis collapse to multitask geometry: for independent balanced labelings, small directional CDNV across tasks forces the corresponding decision axes to be nearly orthogonal, helping a single representation support many tasks with minimal interference. Empirically, across SSL objectives, directional CDNV collapses during pretraining even when classical CDNV remains large, and our bounds closely track few-shot error at practical shot sizes. Additionally, on synthetic multitask data, we verify that SSL learns representations whose induced decision axes are nearly orthogonal. The code and project page of the paper are available at [\href{https://dlfundamentals.github.io/directional-neural-collapse/}{project page}].

自监督学习少样本迁移泛化理论

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