arXiv:2512.03862cs.CV2025-12

小模型自监督学习中,中间分类任务反而拖累性能。

Diminishing Returns in Self-Supervised Learning

  • 用500万参数ViT研究不同训练阶段的交互影响
  • 预训练和下游微调提升效果,但随监督增加收益递减
  • 中间分类任务破坏空间结构,适合小模型避免使用

基于Transformer的架构在视觉和语言领域占据主导地位,但其成功常归因于大模型容量和海量训练数据。本文在低容量场景下,使用500万参数的视觉Transformer研究自监督预训练、中间微调和下游微调的相互作用。在多个数据规模下,发现掩码图像建模预训练和下游微调能稳定提升语义分割性能,但随着监督增强出现明显收益递减。相反,加入中间分类微调阶段会持续降低下游性能,尤其在预训练效果最佳时下降最显著。通过分析像素级表示几何结构,发现基于分类的中间监督会主动干扰预训练中学习到的表示,导致密集预测所需的空间结构被压缩。结果表明,在小模型中,监督的几何对齐比训练阶段数量更重要:不匹配的中间目标可能抵消预训练收益,而非放大它。

原文摘要 · Abstract (English)

Transformer-based architectures have become a dominant paradigm in vision and language, but their success is often attributed to large model capacity and massive training data. In this work, we examine how self-supervised pre-training, intermediate fine-tuning, and downstream fine-tuning interact in a low-capacity regime, using a 5M-parameter Vision Transformer for semantic segmentation. Across multiple data scales, we find that masked image modeling pre-training and downstream fine-tuning reliably improve performance, but with clear diminishing returns as supervision increases. In contrast, inserting an intermediate classification fine-tuning stage consistently degrades downstream performance, with the largest drops occurring precisely where pre-training is most effective. Through an analysis of patch-level representation geometry, we show that classification-based intermediate supervision actively interferes with representations learned during pre-training by collapsing spatial structure critical for dense prediction. These results indicate that, in small models, the geometry of supervision matters more than the number of training stages: misaligned intermediate objectives can negate the benefits of pre-training rather than amplify them.

自监督学习小模型视觉任务表示学习

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