根据图像重要性动态分配计算,高效识别视觉信息。
big.LITTLE Vision Transformer for Efficient Visual Recognition
- 双模型架构:大模型处理关键区域,小模型处理次要区域。
- 分类与分割任务中保持高精度,计算量显著降低。
- 适合资源受限场景的高效视觉模型部署。
本文提出 big.LITTLE Vision Transformer,一种用于高效视觉识别的双变换器架构。该系统由高容量高性能的 big 块和低容量高效率的 LITTLE 块组成。其核心创新在于动态推理机制:处理图像时,系统评估每个视觉令牌的重要性,将关键令牌交由 big 模型处理,非关键令牌由 little 模型处理。这种选择性计算在不牺牲整体性能的前提下大幅降低计算负载。我们在图像分类和 Segment Anything 任务上进行了全面实验,结果表明,big.LITTLE 架构在保持高准确率的同时实现了显著的计算节省。该方法成功验证了混合模型在视觉识别中平衡性能与效率的潜力,为先进神经网络在真实场景中的可扩展部署提供了新路径。
原文摘要 · Abstract (English)
In this paper, we introduce the big.LITTLE Vision Transformer, an innovative architecture aimed at achieving efficient visual recognition. This dual-transformer system is composed of two distinct blocks: the big performance block, characterized by its high capacity and substantial computational demands, and the LITTLE efficiency block, designed for speed with lower capacity. The key innovation of our approach lies in its dynamic inference mechanism. When processing an image, our system determines the importance of each token and allocates them accordingly: essential tokens are processed by the high-performance big model, while less critical tokens are handled by the more efficient little model. This selective processing significantly reduces computational load without sacrificing the overall performance of the model, as it ensures that detailed analysis is reserved for the most important information. To validate the effectiveness of our big.LITTLE Vision Transformer, we conducted comprehensive experiments on image classification and segment anything task. Our results demonstrate that the big.LITTLE architecture not only maintains high accuracy but also achieves substantial computational savings. Specifically, our approach enables the efficient handling of large-scale visual recognition tasks by dynamically balancing the trade-offs between performance and efficiency. The success of our method underscores the potential of hybrid models in optimizing both computation and performance in visual recognition tasks, paving the way for more practical and scalable deployment of advanced neural networks in real-world applications.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。