Search papers, labs, and topics across Lattice.
This paper introduces MetaDAT, a meta-learning framework for trajectory prediction that optimizes for fast and accurate online adaptation to distribution shifts. MetaDAT uses bi-level optimization during pre-training to simulate test-time adaptation tasks, and employs a data-adaptive model updating mechanism at test time that dynamically adjusts learning rates and focuses on hard samples. Experiments on nuScenes, Lyft, and Waymo datasets demonstrate that MetaDAT achieves superior adaptation accuracy compared to state-of-the-art test-time training methods, especially under suboptimal learning rates and high FPS demands.
Trajectory prediction models can now adapt to new environments far more effectively thanks to a meta-learning approach that dynamically adjusts learning rates based on online data characteristics.
Existing trajectory prediction methods exhibit significant performance degradation under distribution shifts during test time. Although test-time training techniques have been explored to enable adaptation, current approaches rely on an offline pre-trained predictor that lacks online learning flexibility. Moreover, they depend on fixed online model updating rules that do not accommodate the specific characteristics of test data. To address these limitations, we first propose a meta-learning framework to directly optimize the predictor for fast and accurate online adaptation, which performs bi-level optimization on the performance of simulated test-time adaptation tasks during pre-training. Furthermore, at test time, we introduce a data-adaptive model updating mechanism that dynamically adjusts the predefined learning rates and updating frequencies based on online partial derivatives and hard sample selection. This mechanism enables the online learning rate to suit the test data, and focuses on informative hard samples to enhance efficiency. Experiments are conducted on various challenging cross-dataset distribution shift scenarios, including nuScenes, Lyft, and Waymo. Results demonstrate that our method achieves superior adaptation accuracy, surpassing state-of-the-art test-time training methods for trajectory prediction. Additionally, our method excels under suboptimal learning rates and high FPS demands, showcasing its robustness and practicality.
