Spectral-Aware Proposal Enhancement for Diffusion-Based Multispectral UAV Multi-Object Tracking
Organizations: Beijing Institute of Technology, Beijing 100081, China. · Beijing Institute of Technology Chongqing Innovation Center, Chongqing 401135, China.
Abstract
Multispectral UAV multi-object tracking (MOT) with oriented bounding boxes remains challenging due to small targets, cluttered backgrounds, and unstable inter-frame associations. Although diffusion-based MOT provides an effective iterative proposal refinement paradigm, existing diffusion-based trackers lack dedicated designs for exploiting multispectral cues in proposal initialization and confidence calibration, limiting tracking robustness in complex UAV scenarios. This letter proposes DSATrack, a diffusion-guided spectral-aware tracking framework for multispectral UAV MOT. Specifically, a lightweight spectral encoder injects multispectral information into the RPN branch through feature fusion, enabling the RPN head to generate high-quality object-aware proposals. These proposals are then injected into the initial proposal set to provide reliable spatial priors for diffusion refinement. Furthermore, a spectral-saliency confidence modulation strategy recalibrates refined tracking confidence according to local multispectral saliency contrast before score thresholding, enhancing spectrally reliable candidates while suppressing unreliable ones. Without modifying the original diffusion refinement head, DSATrack improves both proposal quality and post-refinement candidate selection. Experiments on the MMOT benchmark demonstrate that DSATrack outperforms the adapted DiffusionTrack baseline by 4.6 HOTA, 9.0 MOTA, and 7.7 IDF1 points, showing its effectiveness for multispectral UAV MOT.
Figures & tables
| Type | Method | Class-Averaged | Detection-Averaged | ||||||||
| HOTA | MOTA | IDF1 | DetA | AssA | HOTA | MOTA | IDF1 | DetA | AssA | ||
| Tracking by Detection | SORT [ 2 ] | 27.2 | 24.3 | 29.1 | 25.7 | 30.0 | 35.0 | 25.7 | 33.7 | 27.6 | 44.8 |
| ByteTrack [ 3 ] | 40.5 | 34.2 | 44.1 | 37.0 | 46.2 | 46.0 | 37.8 | 46.7 | 41.9 | 51.5 | |
| OC-SORT [ 4 ] | 29.5 | 25.1 | 31.9 | 27.3 | 32.8 | 37.5 | 27.5 | 37.0 | 29.5 | 48.0 | |
| BoT-SORT [ 5 ] | 53.6 | 46.2 | 61.0 | 45.7 | 64.6 | 60.7 | 59.4 | 69.4 | 55.0 | 68.7 | |
| Tracking by Query | MOTR [ 6 ] | 39.0 | 26.5 | 44.6 | 27.1 | 60.1 | 48.4 | 32.2 | 54.7 | 35.4 | 68.4 |
| A | B | C | HOTA | MOTA | IDF1 | DetA | AssA |
| 47.9 | 31.4 | 51.3 | 41.4 | 58.1 | |||
| 49.6 | 35.1 | 54.6 | 41.7 | 61.4 | |||
| 49.8 | 35.3 | 54.0 | 42.8 | 60.4 | |||
| 51.1 | 37.8 | 56.5 | 43.5 | 62.2 | |||
| 52.5 | 40.4 | 59.0 | 43.8 | 65.5 |
| #RPN | HOTA | MOTA | IDF1 | DetA | AssA |
| 32 | 52.29 | 40.42 | 58.58 | 43.78 | 64.85 |
| 64 | 52.53 | 40.44 | 58.98 | 43.76 | 65.53 |
| 128 | 52.50 | 40.36 | 58.91 | 43.76 | 65.43 |
| 256 | 52.12 | 40.36 | 58.10 | 43.87 | 64.40 |
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