arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2602.08071cs.CV

ViT-5:2020年代中期的视觉变换器

ViT-5: Vision Transformers for The Mid-2020s

Johns Hopkins University(约翰霍普金斯大学)

Feng Wang, Sucheng Ren, Tiezheng Zhang, Predrag Neskovic, Anand Bhattad, Cihang Xie, Alan Yuille

AI总结:

ViT-5通过改进组件和架构,提升了视觉变换器的性能和泛化能力,成为2020年代中期更优的视觉骨干网络。

摘要:

本文系统研究了通过利用过去五年的架构进展来现代化视觉变换器骨干网络。在保持标准的注意力-前馈网络结构的同时,我们对各个组件进行了精细化改进,包括归一化、激活函数、位置编码、门控机制和可学习标记。这些更新形成了一代新的视觉变换器,我们称之为ViT-5。广泛实验表明,ViT-5在理解和生成基准测试中均优于最先进的纯视觉变换器。在ImageNet-1k分类任务中,ViT-5-Base在可比的计算量下达到84.2%的top-1准确率,超过DeiT-III-Base的83.8%。此外,ViT-5作为生成建模的更强骨干网络:当嵌入到SiT扩散框架中时,其FID值为1.84,而普通ViT骨干网络为2.06。除了 headline 指标外,ViT-5还表现出改进的表示学习和有利的空间推理行为,并能可靠地跨任务迁移。与当代基础模型实践相一致的ViT-5为2020年代中期的视觉骨干网络提供了一个简单的直接升级。

英文摘要:

This work presents a systematic investigation into modernizing Vision Transformer backbones by leveraging architectural advancements from the past five years. While preserving the canonical Attention-FFN structure, we conduct a component-wise refinement involving normalization, activation functions, positional encoding, gating mechanisms, and learnable tokens. These updates form a new generation of Vision Transformers, which we call ViT-5. Extensive experiments demonstrate that ViT-5 consistently outperforms state-of-the-art plain Vision Transformers across both understanding and generation benchmarks. On ImageNet-1k classification, ViT-5-Base reaches 84.2\% top-1 accuracy under comparable compute, exceeding DeiT-III-Base at 83.8\%. ViT-5 also serves as a stronger backbone for generative modeling: when plugged into an SiT diffusion framework, it achieves 1.84 FID versus 2.06 with a vanilla ViT backbone. Beyond headline metrics, ViT-5 exhibits improved representation learning and favorable spatial reasoning behavior, and transfers reliably across tasks. With a design aligned with contemporary foundation-model practices, ViT-5 offers a simple drop-in upgrade over vanilla ViT for mid-2020s vision backbones.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑