Trajectory Prediction for Heterogeneous Agents: A Performance Analysis on Small and Imbalanced DatasetsShow others and affiliations
2024 (English)In: IEEE Robotics and Automation Letters, E-ISSN 2377-3766, Vol. 9, no 7, p. 6576-6583Article in journal (Refereed) Published
Abstract [en]
Robots and other intelligent systems navigating in complex dynamic environments should predict future actions and intentions of surrounding agents to reach their goals efficiently and avoid collisions. The dynamics of those agents strongly depends on their tasks, roles, or observable labels. Class-conditioned motion prediction is thus an appealing way to reduce forecast uncertainty and get more accurate predictions for heterogeneous agents. However, this is hardly explored in the prior art, especially for mobile robots and in limited data applications. In this paper, we analyse different class-conditioned trajectory prediction methods on two datasets. We propose a set of conditional pattern-based and efficient deep learning-based baselines, and evaluate their performance on robotics and outdoors datasets (TH & Ouml;R-MAGNI and Stanford Drone Dataset). Our experiments show that all methods improve accuracy in most of the settings when considering class labels. More importantly, we observe that there are significant differences when learning from imbalanced datasets, or in new environments where sufficient data is not available. In particular, we find that deep learning methods perform better on balanced datasets, but in applications with limited data, e.g., cold start of a robot in a new environment, or imbalanced classes, pattern-based methods may be preferable.
Place, publisher, year, edition, pages
IEEE, 2024. Vol. 9, no 7, p. 6576-6583
Keywords [en]
Datasets for human motion, deep learning methods, human and humanoid motion analysis and synthesis, human detection and tracking
National Category
Computer graphics and computer vision
Identifiers
URN: urn:nbn:se:oru:diva-115005DOI: 10.1109/LRA.2024.3408510ISI: 001246186700011Scopus ID: 2-s2.0-85195392481OAI: oai:DiVA.org:oru-115005DiVA, id: diva2:1885610
Funder
Wallenberg AI, Autonomous Systems and Software Program (WASP)2024-07-242024-07-242025-02-07Bibliographically approved