arrow
Return

Hierarchical Feature Pooling Transformer for Efficient UAV Object Tracking

delete2023-01-01
delete10
PRE
AI
H
Haijun Wang *
W
Wenlai Ma
W
Wei Hao
DOI:10.1109/LGRS.2023.3314435delete
deleteOriginal
deleteOriginal request for help
deleteShare
deleteSave
Abstract

Abstract

En 中文
Recently, owing to the long-range feature dependencies, transformers have achieved considerable progress in the field of visual tracking. One of the most challenging problems in transformer-based tracking is that the large feature length of tokens leads to high computational cost. Meanwhile, only the single feature map from the last layer of convolutional neural networks (CNNs) is employed as the input of transformer, which reduces the tracking accuracy and robustness in complex scenarios. Thus, in this letter, we present an efficient and effective hierarchical feature pooling transformer (HFPT) for UAV object tracking, which is able to inherit the merits from both CNN and transformer architectures. First, we introduce a single pooling operation into multihead self-attention (MHSA) in transformer to build a new backbone network for reducing the concatenated feature length and capturing rich contextual information. Second, hierarchical feature maps generated by multilevel convolutional layers are fed into the pooling transformer to learn interdependencies between high-resolution features and low-resolution features. Third, a feature correction layer is designed to enrich the encoded detailed information for handling the small targets. Finally, we evaluate our proposed method in three well-known UAV benchmarks such as DTB70, UAV20L, and UAV123@10fps. Numerous experimental results demonstrate that our HFPT method is able to achieve better performance than the current top-performing trackers with an average speed of 30.3 frames/s on the edge platform of Nvidia Jetson AGX Orin.
Keywords:
Convolutional neural networks (CNNs)
object tracking
self-attention
transformer
UAV

Journal

IEEE Geoscience and Remote Sensing Magazine cover
IEEE Geoscience and Remote Sensing Magazine
IF:
16.4
Papers:
1.0W
Citations:
5.1K

Organization

S
Shandong University of Aeronautics
Scholars:
1.3K
Papers: 917
Citations: 0