Causal neural set filtering for online multi-target tracking
Organizations: Hangzhou Applied Acoustics Research Institute, Hangzhou 310023, China · Independent Researcher, Hangzhou, China
Abstract
Transformer-based multi-target tracking (MTT) jointly learns data association and state estimation, but MT3/Track-MT3-style trackers repeatedly re-encode measurement windows, incurring redundant computation. We propose Causal Neural Set Filtering (CNSF)\footnote{Code: https://github.com/daihuangyu/CNSF}, a neural set filter that encodes only current measurements while carrying past evidence in a structured recursive track state. CNSF combines exclusive Sinkhorn association, association-conditioned Kalman-shaped updates with moment matching, and recurrent Bernoulli lifecycle modeling with measurement-driven birth. These mechanisms impose soft one-to-one constraints, propagate association-induced state uncertainty, and support existence estimation under missed detections and birth--death transitions. On a held-out three-regime simulated test set, CNSF reduces mean GOSPA and T-GOSPA relative to Track-MT3 by 19.3% and 30.4%, with 55.9% fewer parameters and a speedup in single-thread CPU inference.
Figures & tables
| Method | S1 | S2 | S3 | CPU | Par. | |||
|---|---|---|---|---|---|---|---|---|
| MHT | .576 | 1.601 | 2.579 | 1.585 | 1.585 † | 1.676 | 89.04 | – |
| -GLMB | .250 | .998 | 2.034 | 1.094 | 1.094 † | 1.251 | 1614.57 | – |
| PMBM | .270 | .982 | 1.945 | 1.066 | 1.250 | 1.179 | 81.47 | – |
| Track-MT3 | .609 | .990 | 1.680 | 1.093 | 1.245 | 1.443 | 117.37 | 19.38 |
| Track-MT3-CM | .642 | 1.050 | 1.750 | 1.148 | 1.427 | 1.595 | 39.68 | 8.48 |
| CNSF | .166 | .828 | 1.652 | .882 | .990 | 1.004 | 31.18 | 8.54 |
| Variant | |||||
|---|---|---|---|---|---|
| Full CNSF | .882 | .990 | 1.004 | .759 | .246 |
| w/o excl. Sinkhorn | 1.524 | 1.539 | 1.659 | .724 | 1.325 |
| w/o pairwise mix. | .931 | 1.053 | 1.060 | .847 | .250 |
| w/o rec. lifecycle | 1.112 | 1.194 | 1.248 | 1.195 | .177 |
Multi-Object Tracking
Multi-object tracking (MOT) links detections across video frames to maintain object identities through motion, occlusion, and changing viewpoints.
See all 189 papers