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

自适应绕过视觉Transformer模块以实现高效视觉跟踪

Adaptively Bypassing Vision Transformer Blocks for Efficient Visual Tracking

College of Computer Science and Engineering, Guilin University of Technology(桂林理工大学计算机科学与工程学院);Department of Computer Science and Engineering, Southern University of Science and Technology(南方科技大学计算机科学与工程系);College of Computer Science, Sichuan University(四川大学计算机学院)

Xiangyang Yang, Dan Zeng, Xucheng Wang, You Wu, Hengzhou Ye, Qijun Zhao, Shuiwang Li

AI总结:

针对视觉跟踪中ViT推理速度慢的问题,提出ABTrack框架,通过BDM自适应绕过不重要的Transformer模块并结合ViT剪枝降低token维度,在保持跟踪精度的同时实现高效推理。

摘要:

在基于Transformer的模型推动下,视觉跟踪已取得显著进展。然而,当前跟踪器速度较慢,限制了它们在计算资源受限设备上的适用性。为应对这一挑战,我们提出ABTrack,一种自适应计算框架,通过自适应绕过Transformer模块来实现高效视觉跟踪。ABTrack的设计依据源于一个观察:语义特征或关系并非在所有抽象层级上对跟踪任务产生相同影响,其影响随目标及其所处场景的特性而变化。因此,在某些抽象层级上忽略不重要的语义特征或关系可能不会显著影响跟踪精度。我们提出一个Bypass Decision Module(BDM,绕过决策模块)来判断某个Transformer模块是否应被绕过,从而自适应地简化ViT架构并加快推理过程。为抵消BDM带来的时间开销并进一步提升ViT效率,我们引入一种新的ViT剪枝方法,以降低每个Transformer模块中token潜在表示的维度。在多个跟踪基准上的大量实验验证了所提方法的有效性和通用性,并表明其达到了最先进的性能。代码发布于:https://github.com/xyyang317/ABTrack。

英文摘要:

Empowered by transformer-based models, visual tracking has advanced significantly. However, the slow speed of current trackers limits their applicability on devices with constrained computational resources. To address this challenge, we introduce ABTrack, an adaptive computation framework that adaptively bypassing transformer blocks for efficient visual tracking. The rationale behind ABTrack is rooted in the observation that semantic features or relations do not uniformly impact the tracking task across all abstraction levels. Instead, this impact varies based on the characteristics of the target and the scene it occupies. Consequently, disregarding insignificant semantic features or relations at certain abstraction levels may not significantly affect the tracking accuracy. We propose a Bypass Decision Module (BDM) to determine if a transformer block should be bypassed, which adaptively simplifies the architecture of ViTs and thus speeds up the inference process. To counteract the time cost incurred by the BDMs and further enhance the efficiency of ViTs, we introduce a novel ViT pruning method to reduce the dimension of the latent representation of tokens in each transformer block. Extensive experiments on multiple tracking benchmarks validate the effectiveness and generality of the proposed method and show that it achieves state-of-the-art performance. Code is released at: https://github.com/xyyang317/ABTrack.

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

↑