General Compression Framework for Efficient Transformer Object Tracking

Fuente: arXiv
Gespeichert in:
Bibliographische Detailangaben
Hauptverfasser: Hong, Lingyi, Li, Jinglun, Zhou, Xinyu, Yan, Shilin, Guo, Pinxue, Jiang, Kaixun, Chen, Zhaoyu, Gao, Shuyong, Li, Runze, Sheng, Xingdong, Zhang, Wei, Lu, Hong, Zhang, Wenqiang
Format: Preprint
Veröffentlicht: 2024
Schlagworte:
Online-Zugang:
Tags: Tag hinzufügen
Keine Tags, Fügen Sie den ersten Tag hinzu!
_version_ 1866913915697364992
author Hong, Lingyi
Li, Jinglun
Zhou, Xinyu
Yan, Shilin
Guo, Pinxue
Jiang, Kaixun
Chen, Zhaoyu
Gao, Shuyong
Li, Runze
Sheng, Xingdong
Zhang, Wei
Lu, Hong
Zhang, Wenqiang
author_facet Hong, Lingyi
Li, Jinglun
Zhou, Xinyu
Yan, Shilin
Guo, Pinxue
Jiang, Kaixun
Chen, Zhaoyu
Gao, Shuyong
Li, Runze
Sheng, Xingdong
Zhang, Wei
Lu, Hong
Zhang, Wenqiang
contents Previous works have attempted to improve tracking efficiency through lightweight architecture design or knowledge distillation from teacher models to compact student trackers. However, these solutions often sacrifice accuracy for speed to a great extent, and also have the problems of complex training process and structural limitations. Thus, we propose a general model compression framework for efficient transformer object tracking, named CompressTracker, to reduce model size while preserving tracking accuracy. Our approach features a novel stage division strategy that segments the transformer layers of the teacher model into distinct stages to break the limitation of model structure. Additionally, we also design a unique replacement training technique that randomly substitutes specific stages in the student model with those from the teacher model, as opposed to training the student model in isolation. Replacement training enhances the student model's ability to replicate the teacher model's behavior and simplifies the training process. To further forcing student model to emulate teacher model, we incorporate prediction guidance and stage-wise feature mimicking to provide additional supervision during the teacher model's compression process. CompressTracker is structurally agnostic, making it compatible with any transformer architecture. We conduct a series of experiment to verify the effectiveness and generalizability of our CompressTracker. Our CompressTracker-SUTrack, compressed from SUTrack, retains about 99 performance on LaSOT (72.2 AUC) while achieves 2.42x speed up. Code is available at https://github.com/LingyiHongfd/CompressTracker.
format Preprint
id arxiv_https___arxiv_org_abs_2409_17564
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle General Compression Framework for Efficient Transformer Object Tracking
Hong, Lingyi
Li, Jinglun
Zhou, Xinyu
Yan, Shilin
Guo, Pinxue
Jiang, Kaixun
Chen, Zhaoyu
Gao, Shuyong
Li, Runze
Sheng, Xingdong
Zhang, Wei
Lu, Hong
Zhang, Wenqiang
Computer Vision and Pattern Recognition
Previous works have attempted to improve tracking efficiency through lightweight architecture design or knowledge distillation from teacher models to compact student trackers. However, these solutions often sacrifice accuracy for speed to a great extent, and also have the problems of complex training process and structural limitations. Thus, we propose a general model compression framework for efficient transformer object tracking, named CompressTracker, to reduce model size while preserving tracking accuracy. Our approach features a novel stage division strategy that segments the transformer layers of the teacher model into distinct stages to break the limitation of model structure. Additionally, we also design a unique replacement training technique that randomly substitutes specific stages in the student model with those from the teacher model, as opposed to training the student model in isolation. Replacement training enhances the student model's ability to replicate the teacher model's behavior and simplifies the training process. To further forcing student model to emulate teacher model, we incorporate prediction guidance and stage-wise feature mimicking to provide additional supervision during the teacher model's compression process. CompressTracker is structurally agnostic, making it compatible with any transformer architecture. We conduct a series of experiment to verify the effectiveness and generalizability of our CompressTracker. Our CompressTracker-SUTrack, compressed from SUTrack, retains about 99 performance on LaSOT (72.2 AUC) while achieves 2.42x speed up. Code is available at https://github.com/LingyiHongfd/CompressTracker.
title General Compression Framework for Efficient Transformer Object Tracking
topic Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2409.17564