arXiv:2411.15648cs.CV2024-11CVPR被引 6

XTRA通过块因果掩码提升图像生成效率,用更少数据和参数达到更好效果。

Sample- and Parameter-Efficient Auto-Regressive Image Models

  • 采用块因果掩码,逐块预测像素,捕捉更大区域的结构特征。
  • 仅用1310万样本(原模型2000万的6.5%)即超越前代模型在15个任务上的平均表现。
  • 参数减少7-16倍,在线性与注意力探测任务中仍表现更优,适合资源受限场景。

我们提出XTRA,一种基于新型自回归目标训练的视觉模型,相比以往自回归图像模型,在样本效率和参数效率上均有显著提升。与对比学习或掩码图像建模方法不同,自回归视觉模型在互联网大规模非均衡数据下展现出可扩展且稳定的性能。不同于传统自回归模型依赖标准因果掩码,XTRA采用块因果掩码(Block Causal Mask),每个块由k×k个标记组成。通过逐块重建像素值,模型能捕获更大图像区域的高层结构模式,学习更广域像素间关系,从而生成更具语义意义的表示。该改进带来两大关键结果:第一,样本高效——尽管仅使用1310万样本(相比20亿样本减少93.5%),XTRA ViT-H/14在15个多样化图像识别基准上平均准确率超越此前最优自回归模型;第二,参数高效——与ImageNet-1k训练的自回归模型相比,XTRA ViT-B/16仅需8500万参数(分别为13.6亿和6.3亿参数的6.2%~13.5%),在线性探测和注意力探测任务中表现更优。

原文摘要 · Abstract (English)

We introduce XTRA, a vision model pre-trained with a novel auto-regressive objective that significantly enhances both sample and parameter efficiency compared to previous auto-regressive image models. Unlike contrastive or masked image modeling methods, which have not been demonstrated as having consistent scaling behavior on unbalanced internet data, auto-regressive vision models exhibit scalable and promising performance as model and dataset size increase. In contrast to standard auto-regressive models, XTRA employs a Block Causal Mask, where each Block represents k $\times$ k tokens rather than relying on a standard causal mask. By reconstructing pixel values block by block, XTRA captures higher-level structural patterns over larger image regions. Predicting on blocks allows the model to learn relationships across broader areas of pixels, enabling more abstract and semantically meaningful representations than traditional next-token prediction. This simple modification yields two key results. First, XTRA is sample-efficient. Despite being trained on 152$\times$ fewer samples (13.1M vs. 2B), XTRA ViT-H/14 surpasses the top-1 average accuracy of the previous state-of-the-art auto-regressive model across 15 diverse image recognition benchmarks. Second, XTRA is parameter-efficient. Compared to auto-regressive models trained on ImageNet-1k, XTRA ViT-B/16 outperforms in linear and attentive probing tasks, using 7-16$\times$ fewer parameters (85M vs. 1.36B/0.63B).

自回归模型图像生成参数效率样本效率

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