S.T.A.R.-Track: Latent Motion Models for End-to-End 3D Object Tracking with Adaptive Spatio-Temporal Appearance Representations

Simon Doll,Niklas Hanselmann,Lukas Schneider,Richard Schulz,Markus Enzweiler,Hendrik P.A. Lensch
DOI: https://doi.org/10.1109/LRA.2023.3342552
2024-10-13
Abstract:Following the tracking-by-attention paradigm, this paper introduces an object-centric, transformer-based framework for tracking in 3D. Traditional model-based tracking approaches incorporate the geometric effect of object- and ego motion between frames with a geometric motion model. Inspired by this, we propose S.T.A.R.-Track, which uses a novel latent motion model (LMM) to additionally adjust object queries to account for changes in viewing direction and lighting conditions directly in the latent space, while still modeling the geometric motion explicitly. Combined with a novel learnable track embedding that aids in modeling the existence probability of tracks, this results in a generic tracking framework that can be integrated with any query-based detector. Extensive experiments on the nuScenes benchmark demonstrate the benefits of our approach, showing state-of-the-art performance for DETR3D-based trackers while drastically reducing the number of identity switches of tracks at the same time.
Computer Vision and Pattern Recognition,Artificial Intelligence,Robotics
What problem does this paper attempt to address?
The paper attempts to address the problem of achieving robust 3D object tracking in autonomous driving scenarios. Specifically, the paper proposes a new attention-based multi-object tracking framework called S.T.A.R.-Track, aiming to solve the following key issues: 1. **Geometric Transformation and Appearance Variation**: - Traditional object tracking methods usually only consider the impact of geometric transformations (such as object and ego-vehicle motion) on the target state, while ignoring appearance changes caused by variations in viewpoint and lighting conditions. This can lead to a decline in tracking performance in complex environments. - The paper proposes a new Latent Motion Model (LMM) that can simultaneously adjust geometric and appearance information in the latent space, thereby improving tracking consistency and accuracy. 2. **Target Existence Probability Modeling**: - In multi-object tracking, correctly estimating the existence probability of each target is crucial for reducing tracking fragments and identity switches. - The paper introduces learnable track embeddings that can implicitly model the lifecycle of targets, thus better handling short-term occlusions and target disappearances. 3. **Joint Detection and Tracking**: - Traditional methods that separate detection and tracking have two main drawbacks: (1) detection models optimize for detection metrics rather than tracking performance, leading to error accumulation; (2) it is difficult to effectively integrate appearance information, affecting consistent association. - S.T.A.R.-Track unifies the detection and tracking tasks within a single module, using attention mechanisms to jointly perform object detection and re-identification, thereby improving overall performance. 4. **Reducing Identity Switches and Fragmentation**: - In crowded scenes, multiple partially occluded targets can easily lead to identity confusion and frequent identity switches. - Through the aforementioned methods, S.T.A.R.-Track significantly reduces identity switches and trajectory fragmentation, demonstrating state-of-the-art performance, especially in the nuScenes benchmark. In summary, the main contribution of this paper is the proposal of a new latent motion model and track embedding method, which enables more robust and accurate 3D object tracking in complex autonomous driving environments.