Fast computation of the TGOSPA metric for multiple target tracking via unbalanced optimal transport
Viktor Nevelius Wernholm, Alfred Wärnsäter, Axel Ringh
TL;DR
The paper tackles the computational bottleneck of evaluating the trajectory generalized optimal sub-pattern assignment (TGOSPA) metric in multi-target tracking by reformulating the LP-relaxed TGOSPA as an entropy-regularized, unbalanced multimarginal OT problem. It derives a Sinkhorn-type algorithm via block coordinate ascent in the Lagrangian dual and shows how the cost structure enables efficient projection computations with $ ext{O}(m^2 n^2)$ complexity per update, yielding linear convergence. Numerical results demonstrate substantial speed-ups over exact LP solvers while maintaining accurate approximations (≈1% relative error on larger problems), with potential for GPU acceleration. The approach provides a scalable, principled metric computation for large MTT scenarios and suggests future work on differentiable implementations for gradient-based data-driven tracking.
Abstract
In multiple target tracking, it is important to be able to evaluate the performance of different tracking algorithms. The trajectory generalized optimal sub-pattern assignment metric (TGOSPA) is a recently proposed metric for such evaluations. The TGOSPA metric is computed as the solution to an optimization problem, but for large tracking scenarios, solving this problem becomes computationally demanding. In this paper, we present an approximation algorithm for evaluating the TGOSPA metric, based on casting the TGOSPA problem as an unbalanced multimarginal optimal transport problem. Following recent advances in computational optimal transport, we introduce an entropy regularization and derive an iterative scheme for solving the Lagrangian dual of the regularized problem. Numerical results suggest that our proposed algorithm is more computationally efficient than the alternative of computing the exact metric using a linear programming solver, while still providing an adequate approximation of the metric.
