arXiv:2502.18056cs.CV2025-02被引 1

用小数据训练出强视觉表征,打破大模型依赖。

Escaping The Big Data Paradigm in Self-Supervised Representation Learning

  • 设计轻量级卷积分词器SCOTT,注入先验知识提升小数据表现
  • 在三组小规模数据上超越监督学习,接近大模型效果
  • 适合医疗、机器人等资源受限场景,推动普惠视觉技术

当前视觉表征学习严重依赖大规模数据和算力,限制了小样本领域的进展。本文提出SCOTT(稀疏卷积分词器),一种适配掩码图像建模(MIM)的浅层分词架构,将卷积归纳偏置引入视觉变压器(ViT),增强其在小数据下的性能。同时,在MIM框架下采用联合嵌入预测架构(MIM-JEPA),在潜在表示空间中捕捉更丰富的语义特征。该方法使ViT可从零开始训练于远小于传统要求的数据集——无需依赖大规模外部预训练数据。我们在三组小规模、标准分辨率的细粒度数据集(Oxford Flowers-102、Oxford IIIT Pets-37、ImageNet-100)上验证,尽管数据有限且类内相似度高,冻结的SCOTT模型经MIM-JEPA预训练后,显著优于全监督方法,并达到与依赖大规模预训练、复杂数据增强和更大模型的前沿方法相当的性能。本工作证明,仅用少量数据、算力和模型规模即可学习到鲁棒的开箱即用表征,为医疗影像、机器人等资源受限场景提供新路径。研究挑战了‘海量数据是有效表征学习必需’的主流观点,开启更具包容性的视觉技术发展新范式。

原文摘要 · Abstract (English)

The reliance on large-scale datasets and extensive computational resources has become a major barrier to advancing representation learning in vision, especially in data-scarce domains. In this paper, we address the critical question: Can we escape the big data paradigm in self-supervised representation learning from images? We introduce SCOTT (Sparse Convolutional Tokenizer for Transformers), a shallow tokenization architecture that is compatible with Masked Image Modeling (MIM) tasks. SCOTT injects convolutional inductive biases into Vision Transformers (ViTs), enhancing their efficacy in small-scale data regimes. Alongside, we propose to train on a Joint-Embedding Predictive Architecture within a MIM framework (MIM-JEPA), operating in latent representation space to capture more semantic features. Our approach enables ViTs to be trained from scratch on datasets orders of magnitude smaller than traditionally required --without relying on massive external datasets for pretraining. We validate our method on three small-size, standard-resoultion, fine-grained datasets: Oxford Flowers-102, Oxford IIIT Pets-37, and ImageNet-100. Despite the challenges of limited data and high intra-class similarity, frozen SCOTT models pretrained with MIM-JEPA significantly outperform fully supervised methods and achieve competitive results with SOTA approaches that rely on large-scale pretraining, complex image augmentations and bigger model sizes. By demonstrating that robust off-the-shelf representations can be learned with limited data, compute, and model sizes, our work paves the way for computer applications in resource constrained environments such as medical imaging or robotics. Our findings challenge the prevailing notion that vast amounts of data are indispensable for effective representation learning in vision, offering a new pathway toward more accessible and inclusive advancements in the field.

自监督学习小样本视觉模型资源受限

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