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

学习用于实时无人机跟踪的自适应且视角不变的Vision Transformer

Learning an Adaptive and View-Invariant Vision Transformer for Real-Time UAV Tracking

College of Computer Science and Engineering, Guilin University of Technology(桂林理工大学计算机科学与工程学院);Guangxi Key Laboratory of Embedded Technology and Intelligent Information Processing(广西嵌入式技术与智能信息处理重点实验室);School of Computer Science, Fudan University(复旦大学计算机科学学院);School of Artificial Intelligence, Sun Yat-sen University(中山大学人工智能学院);College of Computer Science, Sichuan University(四川大学计算机科学学院)

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

AI总结:

本文提出自适应跟踪框架AVTrack,通过动态激活Transformer块平衡性能与效率,并利用互信息最大化学习视角不变表示;进一步提出AVTrack-MD,采用多教师知识蒸馏提升泛化与性能,在保持精度的同时降低复杂度并提速17%以上。

摘要:

基于Transformer的模型改进了视觉跟踪,但大多数仍无法在资源受限的设备上实时运行,尤其是在无人机(UAV)跟踪方面。为了在性能和效率之间取得更好的平衡,我们提出了AVTrack,这是一种自适应计算跟踪框架,通过Activation Module (AM)自适应地激活Transformer块,该模块通过选择性启用相关组件来动态优化ViT架构。为了应对极端视角变化,我们提出通过互信息(MI)最大化来学习视角不变表示。此外,我们提出了AVTrack-MD,这是一种结合了基于MI最大化的新型多教师知识蒸馏框架的增强型跟踪器。利用多个现成的AVTrack模型作为教师,我们最大化其聚合软化特征与学生模型相应软化特征之间的MI,从而提升学生的泛化能力和性能,特别是在噪声条件下。大量实验表明,AVTrack-MD实现了与AVTrack相当的性能,同时降低了模型复杂度,并将平均跟踪速度提高了17%以上。代码可在https://github.com/wuyou3474/AVTrack获取。

英文摘要:

Transformer-based models have improved visual tracking, but most still cannot run in real time on resource-limited devices, especially for unmanned aerial vehicle (UAV) tracking. To achieve a better balance between performance and efficiency, we propose AVTrack, an adaptive computation tracking framework that adaptively activates transformer blocks through an Activation Module (AM), which dynamically optimizes the ViT architecture by selectively engaging relevant components. To address extreme viewpoint variations, we propose to learn view-invariant representations via mutual information (MI) maximization. In addition, we propose AVTrack-MD, an enhanced tracker incorporating a novel MI maximization-based multi-teacher knowledge distillation framework. Leveraging multiple off-the-shelf AVTrack models as teachers, we maximize the MI between their aggregated softened features and the corresponding softened feature of the student model, improving the generalization and performance of the student, especially under noisy conditions. Extensive experiments show that AVTrack-MD achieves performance comparable to AVTrack's performance while reducing model complexity and boosting average tracking speed by over 17\%. Codes is available at: https://github.com/wuyou3474/AVTrack.

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

↑