PAT-VCM: Plug-and-Play Auxiliary Tokens for Video Coding for Machines
Organizations: Futurewei Technologies Inc. San Jose, CA 95131
Abstract
Existing video coding for machines is often trained for a specific downstream task and model. As a result, the compressed representation becomes tightly coupled to the end task, making it difficult to scale across multiple tasks or adapt to model updates. We propose PAT-VCM, a plug-and-play auxiliary-token framework for video coding for machines. PAT-VCM keeps a shared baseline compressed stream and augments it with lightweight task-aware auxiliary tokens, allowing different downstream tasks to recover the information they need without retraining a separate codec for each task. The framework supports three forms of auxiliary information: visual residual tokens, prompt/control tokens, and semantic tokens. We evaluate PAT-VCM on segmentation, depth estimation, and semantic recognition. A shared detection-oriented auxiliary branch provides a reusable first refinement, task-specific visual branches improve segmentation and depth, prompt tokens provide further segmentation gains at negligible bitrate, and semantic tokens achieve strong recognition performance with extremely low overhead. These results suggest that a shared compressed representation, combined with lightweight task-aware auxiliary tokens, is a practical and scalable alternative to tightly task-coupled VCM design.
Figures & tables
| Dataset | Orig | Cosmos | Det-Aux | + Seg-Aux | + Seg-Aux + 1-pt | + Seg-Aux + FG+BG |
|---|---|---|---|---|---|---|
| DAVIS | 0.774 | 0.677 | 0.686 | 0.709 | 0.751 | 0.764 |
| MOSE | 0.713 | 0.693 | 0.694 | 0.707 | 0.743 | 0.753 |
| VIPSeg | 0.731 | 0.721 | 0.723 | 0.727 | 0.751 | 0.754 |
| Variant | Full AbsRel | ROI AbsRel | ROI |
|---|---|---|---|
| Cosmos | 2.576 | 3.764 | 0.828 |
| + Det-Aux | 2.151 | 3.057 | 0.834 |
| + Det-Aux + Depth-Aux | 1.290 | 1.554 | 0.840 |
| Task | System | bpp | Metric | Value |
| Segmentation | Cosmos DV4 | 0.0833 | IoU | 0.677 |
| PAT-VCM Det-Aux+Seg-Aux | 0.1242 | IoU | 0.687 | |
| PAT-VCM Det-Aux+Seg-Aux+FG+BG | 0.1242 | IoU | 0.764 | |
| Cosmos CV4 (nearest higher-rate) | 0.1671 | IoU | 0.509 | |
| MPA q=1 | 0.0908 | IoU | 0.739 | |
| Detection | Cosmos DV4 | 0.0833 | matched IoU | 0.475 |
Appendix figures & tables21 assets
Supplementary material from the paper’s appendix.
Appendix
| Dataset | No. GT | Cosmos | + Det-Aux | Orig | % gap closed |
|---|---|---|---|---|---|
| DAVIS-2017 val | 59 | 0.390 | 0.475 | 0.610 | 39% |
| MOSE val | 386 | 0.233 | 0.249 | 0.267 | 47% |
| VIPSeg val | 528 | 0.127 | 0.133 | 0.161 | 18% |
| Method | Matched IoU | [email protected] |
|---|---|---|
| Cosmos | 0.475 | 0.516 |
| + Det-Aux | 0.512 | 0.549 |
| Bin | No. instances | Cosmos+SAM | + Text token | |
|---|---|---|---|---|
| Hard ( ) | 26 | 0.098 | 0.246 | +0.149 |
| Medium ( ) | 160 | 0.592 | 0.586 | -0.006 |
| Easy ( ) | 191 | 0.853 | 0.836 | -0.017 |
| Overall | 377 | 0.682 | 0.652 | -0.030 |
| Policy | Avg bits/ROI | Avg bits/clip | bpp | Hard-bin IoU | Overall IoU |
|---|---|---|---|---|---|
| No text | 0 | 0 | 0 | 0.098 | 0.682 |
| Uniform text | 154 | 231 | 0.000088 | 0.246 | 0.652 |
| Adaptive text | 10.4 | 15.6 | 0.000006 | 0.246 | 0.692 |
| Variant | Mean IoU | Gain over Cosmos |
|---|---|---|
| Cosmos | 0.677 | +0.000 |
| + Det-Aux | 0.687 | +0.010 |
| + Det-Aux + 1-pt | 0.740 | +0.063 |
| + Det-Aux + Seg-Aux | 0.710 | +0.033 |
| + Det-Aux + Seg-Aux + 1-pt | 0.751 | +0.074 |
| Orig | 0.774 | +0.097 |
| Difficulty | No. instance | Cosmos | + Det-Aux | + Det-Aux + Seg-Aux | + Det-Aux + Seg-Aux + 1-pt |
|---|---|---|---|---|---|
| easy | 23 | 0.8742 | 0.8780 | 0.8715 | 0.9010 |
| medium | 29 | 0.6281 | 0.6443 | 0.6823 | 0.7236 |
| hard | 6 | 0.1661 | 0.1551 | 0.2208 | 0.3382 |
| Size bin | No. instance | Cosmos | + Det-Aux | + Det-Aux + Seg-Aux | + Det-Aux + Seg-Aux + 1-pt |
|---|---|---|---|---|---|
| Small | 20 | 0.6010 | 0.5953 | 0.6131 | 0.6278 |
| Medium | 32 | 0.7263 | 0.7366 | 0.7611 | 0.8209 |
| Large | 6 | 0.6762 | 0.7222 | 0.7562 | 0.8188 |
| Difficulty | No. instance | Cosmos | + Det-Aux | + Det-Aux + Depth-Aux |
|---|---|---|---|---|
| easy | 4 | 0.0341 | 0.0332 | 0.0394 |
| medium | 31 | 0.1243 | 0.1270 | 0.1320 |
| hard | 23 | 4.2539 | 3.5733 | 2.2300 |
| Dataset | Variant | AbsRel | RMSE | |
|---|---|---|---|---|
| DAVIS full-frame | Cosmos | 2.576 | 0.315 | 0.810 |
| DAVIS full-frame | + Det-Aux | 2.151 | 0.310 | 0.807 |
| DAVIS ROI-only | Cosmos | 3.764 | 0.364 | 0.828 |
| DAVIS ROI-only | + Det-Aux | 3.057 | 0.357 | 0.834 |
| VIPSeg full-frame | Cosmos | 0.484 | 0.187 | 0.893 |
| VIPSeg full-frame | + Det-Aux | 0.476 | 0.187 | 0.895 |
| Pipeline | Latency (ms / clip) | Throughput |
| Cosmos | 101.6 | 88.6 fps |
| Encoder: Cosmos + DETR + Det-Aux | 488.3 | 18.4 fps |
| Decoder: Cosmos + Det-Aux + DETR | 530.5 | 17.0 fps |
| Encoder: Cosmos + DETR + Det-Aux + Seg-Aux | 640.0 | 14.1 fps |
| Decoder: Cosmos + Det-Aux + Seg-Aux + SAM | 692.5 | 13.0 fps |
| Encoder: above + adaptive LLaVA text | 704.0 | 12.8 fps |
| Variant | Extra | Total | Bits / frame | bpp |
|---|---|---|---|---|
| Cosmos DV baseline | 0 | 196,608 | 21,845 | 0.08333 |
| + Det-Aux (measured ROI-masked) | 40,960 | 237,568 | 26,396 | 0.10069 |
| + Det-Aux (full-frame upper bound) | 147,456 | 344,064 | 38,229 | 0.14583 |
| + Det-Aux + Seg-Aux | 96,460 | 293,068 | 32,563 | 0.12422 |
| + Text + box | 228 | 196,836 | 21,871 | 0.08343 |
| + 1-pt | 5 | 196,615 | 21,846 | 0.08334 |
| Task | System | bpp | Metric | Value |
|---|---|---|---|---|
| Segmentation | Cosmos DV4 baseline | 0.0833 | IoU | 0.6761 |
| PAT-VCM (Det-Aux + Seg-Aux + FG+BG) | 0.1242 | IoU | 0.7640 | |
| Cosmos CV4 (nearest higher-rate) | 0.1671 | IoU | 0.5087 | |
| Detection | Cosmos DV4 baseline | 0.0833 | matched IoU | 0.4751 |
| PAT-VCM (Det-Aux) | 0.1007 | matched IoU | 0.5116 | |
| Cosmos CV4 (nearest higher-rate) | 0.1671 | matched IoU | 0.0309 |
| System | bpp | IoU |
|---|---|---|
| Cosmos DV4 | 0.0833 | 0.6761 |
| Cosmos DV4 + Det-Aux + Seg-Aux | 0.1242 | 0.6874 |
| Cosmos CV4 N=1 | 0.0838 | 0.5460 |
| Cosmos CV4 N=2 | 0.1671 | 0.5087 |
| Cosmos CV4 N=3 | 0.2504 | 0.6328 |
| Cosmos CV4 N=10 | 0.8338 | 0.6764 |
| System | bpp | matched IoU |
|---|---|---|
| Cosmos DV4 | 0.0833 | 0.4751 |
| Cosmos DV4 + Det-Aux | 0.1007 | 0.5116 |
| Cosmos CV4 N=1 | 0.0838 | 0.0093 |
| Cosmos CV4 N=2 | 0.1671 | 0.0309 |
| Cosmos CV4 N=3 | 0.2504 | 0.2717 |
| Cosmos CV4 N=10 | 0.8338 | 0.5229 |
| System | bpp | AbsRel |
|---|---|---|
| Cosmos DV4 | 0.0833 | 2.296 |
| Cosmos DV4 + Det-Aux | 0.1007 | 2.412 |
| Cosmos CV4 N=1 | 0.0838 | 23.55 |
| Cosmos CV4 N=2 | 0.1671 | 19.86 |
| Cosmos CV4 N=3 | 0.2504 | 2.361 |
| Cosmos CV4 N=10 | 0.8338 | 1.448 |
| System | bpp | SAM IoU |
|---|---|---|
| MPA q=1 | 0.0908 | 0.7387 |
| MPA q=2 | 0.1320 | 0.7315 |
| MPA q=3 | 0.1914 | 0.7454 |
| MPA q=8 | 1.0189 | 0.7656 |
| Cosmos baseline | 0.0833 | 0.6761 |
| Det-Aux + Seg-Aux | 0.1242 | 0.6874 |
| Method | Class agreement (%) | Bits / ROI | No. instances |
|---|---|---|---|
| Cosmos (no token) | 64.9 | 0 | 57 |
| + Det-Aux (no token) | 63.2 | 0 | 57 |
| + Det-Aux + Class-label token | 100.0 | 7 | 57 |
| Model | ROI MAE ∘ |
|---|---|
| Cosmos | 5.13 |
| + Det-Aux | 5.07 |
| + Normal-Aux | 5.06 |
| + Depth-Aux | 5.17 |
| Method | MKE (px) | Bits / person |
|---|---|---|
| Cosmos (no token) | 33.62 | 0 |
| + Det-Aux (no token) | 31.99 | 0 |
| + Skeleton token | 7.33 | 102 |
| Branch | Training dataset | In-domain held-out |
|---|---|---|
| Det-Aux | YTVIS 2021 splits 0–6 | YTVIS split_7 |
| Seg-Aux feat-distill | YTVIS 2021 splits 0–6 | YTVIS split_7 |
| Seg-Aux SAM-aware | MOSE train | MOSE val |
| Depth-Aux | VIPSeg train | VIPSeg val |
| Normal-Aux | VIPSeg train | – |
| Pose-Aux (primary) | VIPSeg train | – |
Video Compression
Video compression reduces the bits needed to store or transmit moving images while balancing visual quality, computational cost, and downstream usefulness.
See all 73 papers