arXiv:2512.16922cs.CV2025-12被引 17

用嵌入预测训练视觉模型,无需对比损失也能达到顶尖性能

Next-Embedding Prediction Makes Strong Vision Learners

  • 通过预测后续图像块嵌入实现自监督学习,摒弃传统复杂设计
  • ViT-B和ViT-L在ImageNet-1K上微调后准确率分别达83.8%和85.3%
  • 方法简洁可扩展,适合追求高效自监督学习的研究者

受自然语言生成预训练成功的启发,我们探讨其是否适用于视觉自监督学习。不同于传统学习特征用于下游任务,本文提出直接训练模型生成嵌入以完成预测任务。具体地,模型通过因果掩码与停止梯度机制,基于历史嵌入预测未来嵌入,称为下一嵌入预测自回归(NEPA)。实验表明,仅以该目标预训练的简单Transformer在ImageNet-1k上即表现优异——无需像素重建、离散标记、对比损失或任务特定头。该方法保持架构简洁与可扩展性,不增加额外设计复杂度。在微调后,使用ViT-B和ViT-L骨干网络的模型在ImageNet-1K上分别取得83.8%和85.3%的top-1准确率,并在ADE20K语义分割任务上实现有效迁移。我们认为,从嵌入出发的生成预训练为视觉自监督学习提供了一种简单、可扩展且可能跨模态通用的替代方案。

原文摘要 · Abstract (English)

Inspired by the success of generative pretraining in natural language, we ask whether the same principles can yield strong self-supervised visual learners. Instead of training models to output features for downstream use, we train them to generate embeddings to perform predictive tasks directly. This work explores such a shift from learning representations to learning models. Specifically, models learn to predict future patch embeddings conditioned on past ones, using causal masking and stop gradient, which we refer to as Next-Embedding Predictive Autoregression (NEPA). We demonstrate that a simple Transformer pretrained on ImageNet-1k with next embedding prediction as its sole learning objective is effective - no pixel reconstruction, discrete tokens, contrastive loss, or task-specific heads. This formulation retains architectural simplicity and scalability, without requiring additional design complexity. NEPA achieves strong results across tasks, attaining 83.8% and 85.3% top-1 accuracy on ImageNet-1K with ViT-B and ViT-L backbones after fine-tuning, and transferring effectively to semantic segmentation on ADE20K. We believe generative pretraining from embeddings provides a simple, scalable, and potentially modality-agnostic alternative to visual self-supervised learning.

自监督学习视觉模型生成预训练Transformer

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