arXiv:2602.24159cs.CVcs.LG2026-02

用多分辨率分支+早退出,让ViT更省算力还不降精度。

RAViT: Resolution-Adaptive Vision Transformer

  • 同一图像多分辨率并行处理,动态选择计算量
  • 在ImageNet上达原ViT精度,仅需70%计算量
  • 适合部署在算力受限的实时视觉场景

视觉变压器(Vision Transformers)在计算机视觉中表现出色,但其计算成本远高于卷积神经网络。为此,我们提出一种名为RAViT的新框架,采用多分支结构对同一图像的不同分辨率版本进行处理,从而降低计算开销并保持整体精度。该框架还引入早退出机制,使模型可自适应调整运行时的精度与计算成本权衡。例如,在双分支架构中,图像先降采样后由首个Transformer预测,再结合原始尺寸图像,以较少计算量完成第二个Transformer的最终预测。早退出机制允许模型在中间分支即做出最终预测,进一步节省计算。我们在CIFAR-10、Tiny ImageNet和ImageNet上评估该方法,结果表明,相较于经典Vision Transformer,RAViT在保持相当精度的同时,仅需约70%的浮点运算次数(FLOPs)。

原文摘要 · Abstract (English)

Vision transformers have recently made a breakthrough in computer vision showing excellent performance in terms of precision for numerous applications. However, their computational cost is very high compared to alternative approaches such as Convolutional Neural Networks. To address this problem, we propose a novel framework for image classification called RAViT based on a multi-branch network that operates on several copies of the same image with different resolutions to reduce the computational cost while preserving the overall accuracy. Furthermore, our framework includes an early exit mechanism that makes our model adaptive and allows to choose the appropriate trade-off between accuracy and computational cost at run-time. For example in a two-branch architecture, the original image is first resized to reduce its resolution, then a prediction is performed on it using a first transformer and the resulting prediction is reused together with the original-size image to perform a final prediction on a second transformer with less computation than a classical Vision transformer architecture. The early-exit process allows the model to make a final prediction at intermediate branches, saving even more computation. We evaluated our approach on CIFAR-10, Tiny ImageNet, and ImageNet. We obtained an equivalent accuracy to the classical Vision transformer model with only around 70% of FLOPs.

视觉Transformer高效计算多尺度早退出

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