arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2602.16160cs.CV

不确定性引导的推理时间深度适应用于基于变换器的视觉跟踪

Uncertainty-Guided Inference-Time Depth Adaptation for Transformer-Based Visual Tracking

Patrick Poggi, Divake Kumar, Theja Tulabandhula, Amit Ranjan Trivedi

AI总结:

本文提出UncL-STARK,通过动态不确定性引导的深度适应,在不修改网络的情况下,提升基于变换器的视觉跟踪效率,实现计算开销降低和能耗节省。

摘要:

基于变换器的单目标跟踪器在准确率上达到了最先进的水平,但依赖于固定的深度推理,对每个帧执行完整的编码器-解码器堆栈,无论视觉复杂性如何,因此在由时间相干帧主导的长视频序列中会造成不必要的计算开销。我们提出了UncL-STARK,一种在架构上保持不变的方法,能够在基于变换器的跟踪器中实现动态、不确定性感知的深度适应,而无需修改底层网络或添加辅助头。模型通过随机深度训练与知识蒸馏进行微调,以在多个中间深度上保留预测的鲁棒性,从而实现安全的推理时间截断。在运行时,我们直接从模型的角落定位热图中推导出轻量级的不确定性估计,并利用视频中的时间相干性,在预测置信度的基础上,通过反馈驱动的策略选择下一帧的编码器和解码器深度。在GOT-10k和LaSOT上的广泛实验表明,可以实现高达12%的GFLOPs减少、8.9%的延迟减少和10.8%的能耗节省,同时在短期和长期序列中保持跟踪精度在全深度基线的0.2%以内。

英文摘要:

Transformer-based single-object trackers achieve state-of-the-art accuracy but rely on fixed-depth inference, executing the full encoder--decoder stack for every frame regardless of visual complexity, thereby incurring unnecessary computational cost in long video sequences dominated by temporally coherent frames. We propose UncL-STARK, an architecture-preserving approach that enables dynamic, uncertainty-aware depth adaptation in transformer-based trackers without modifying the underlying network or adding auxiliary heads. The model is fine-tuned to retain predictive robustness at multiple intermediate depths using random-depth training with knowledge distillation, thus enabling safe inference-time truncation. At runtime, we derive a lightweight uncertainty estimate directly from the model's corner localization heatmaps and use it in a feedback-driven policy that selects the encoder and decoder depth for the next frame based on the prediction confidence by exploiting temporal coherence in video. Extensive experiments on GOT-10k and LaSOT demonstrate up to 12% GFLOPs reduction, 8.9% latency reduction, and 10.8% energy savings while maintaining tracking accuracy within 0.2% of the full-depth baseline across both short-term and long-term sequences.

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

↑