UniMergeTrack: Hierarchical Model Merging for General RGB-X Visual Object Tracking
Organizations: School of Computer Science and Technology, Anhui University, Hefei 230601, China · Peng Cheng Laboratory, Shenzhen, China · School of Computer Science, Peking University, China · School of Electronic and Computer Engineering, Shenzhen Graduate School, Peking University, China
Abstract
RGB-X tracking leverages complementary modalities, such as thermal infrared, event, or depth, to improve robustness under challenging conditions. Existing general-purpose RGB-X trackers typically follow two paradigms: independent task-specific training and joint training across modality combinations. However, the former suffers from knowledge isolation and model redundancy, while the latter introduces optimization interference across heterogeneous tasks. In this work, we present UniMergeTrack, a new model merging paradigm that preserves task-specific optimization while enabling effective cross-task knowledge integration. UniMergeTrack first trains modality-specific trackers independently and then performs hierarchical model merging. Specifically, transferable backbones are merged across RGB-X tasks via high-pass-filtered merging, and complemented by lightweight task-specific experts that preserve modality-specific cues. Tracking heads are merged within each task using optimal transport-guided channel alignment to mitigate hidden channel mismatch. A subsequent distribution alignment fine-tuning stage further improves compatibility between the unified backbone and task-specific heads. Extensive experiments on the LasHeR, RGBT234, VisEvent, VOT-RGBD22, COESOT, and CDTB benchmarks demonstrate that UniMergeTrack achieves state-of-the-art or competitive performance across RGB-Thermal, RGB-Event, and RGB-Depth tracking tasks, while exhibiting strong generalization to unseen datasets. The source code will be released upon acceptance. The source code of this paper will be released on https://github.com/Event-AHU/OpenEvTracking.
Figures & tables
| Method | Source | LasHeR | RGBT234 | VisEvent | VOT-RGBD22 | |||||
|---|---|---|---|---|---|---|---|---|---|---|
| AUC | PR | MSR | MPR | AUC | PR | EAO | Acc. | Rob. | ||
| OSTrack ( Ye et al., 2022 ) | ECCV22 | 41.2 | 51.5 | 54.9 | 72.9 | 53.4 | 69.5 | 67.6 | 80.3 | 83.3 |
| ProTrack ( Yang et al., 2022 ) | ACM MM22 | 42.0 | 53.8 | 59.9 | 79.5 | 47.1 | 63.2 | 65.1 | 80.1 | 80.2 |
| ViPT ( Zhu et al., 2023 ) | CVPR23 | 52.5 | 65.1 | 61.7 | 83.5 | 59.2 | 75.8 | 72.1 | 81.5 | 87.1 |
| SeqTrackv2 ( Chen et al., 2023a ) | ArXiv23 | 55.8 | 70.4 | 64.7 | 88.0 | 61.2 | 78.2 | 74.4 | 81.5 | 91.0 |
| OneTracker ( Hong et al., 2024 ) | CVPR24 | 53.8 | 67.2 | 64.2 | 85.7 | 60.8 | 76.7 | 72.7 | 81.9 | 87.2 |
| Dataset | Metric | SMSTracker | XTrack | SUTrack | SEATrack | UETrack | UTPTrack | UniMergeTrack |
|---|---|---|---|---|---|---|---|---|
| COESOT (RGB-Event) | SR | 63.6 | 65.2 | 64.6 | 64.8 | 61.5 | 64.7 | 66.3 |
| PR | 76.8 | 78.8 | 77.7 | 78.9 | 74.7 | 78.2 | 80.5 | |
| CDTB (RGB-Depth) | F-score | 69.7 | 69.8 | 70.8 | 68.5 | 67.7 | 71.5 | 72.0 |
| Precision | 70.5 | 68.5 | 69.8 | 69.4 | 68.2 | 71.4 | 72.0 |
| No. | Variant | LasHeR | VisEvent | Avg. Change | |||
| AUC | PR | AUC | PR | AUC | PR | ||
| 1 | UniMergeTrack (Ours) | 61.8 | 77.2 | 64.1 | 81.5 | – | – |
| 2 | w/o Model Merging (Baseline) | 58.8 | 73.4 | 62.2 | 79.2 | -2.5 | -3.1 |
| 3 | w/o Distribution Alignment Fine-Tuning | 57.7 | 73.2 | 60.9 | 78.6 | -3.7 | -3.5 |
| 4 | w/o High-pass Filtering | 60.1 | 75.2 | 62.9 | 80.0 | -1.5 | -1.8 |
| 5 | w/o Lightweight Modality Expert | 60.3 | 75.3 | 63.0 | 80.4 | -1.3 | -1.5 |
Appendix figures & tables3 assets
Supplementary material from the paper’s appendix.
Appendix
| Hyperparameter | RGB-Thermal | RGB-Event | RGB-Depth |
|---|---|---|---|
| Expert sparsity ( ) | 0.01 | 0.02 | 0.02 |
| Reference head contribution ( ) | 0.50 | 0.70 | 0.50 |
Visual Object Tracking
Visual object tracking follows objects across video frames, maintaining their locations and identities despite motion, occlusion, and changing appearance.
See all 174 papers