Search papers, labs, and topics across Lattice.
This paper introduces UniTAC, a novel image codec that adapts to varying tasks without the need for retraining, utilizing a per-component importance vector derived from gradient attribution. By conditioning both the encoder and decoder on this low-overhead side information, UniTAC achieves a balance between universal and task-specific performance. The results show that UniTAC can maintain competitive accuracy (91.4%) compared to task-based codecs while significantly outperforming universal codecs, demonstrating its efficiency in high-dimensional sensory signal processing under constraints.
A single learned codec can adapt to diverse tasks on-the-fly, achieving near task-specific performance without retraining.
Physical AI systems such as autonomous vehicles and robots rely on timely exchange of high-dimensional sensory signals under tight bandwidth, latency, and energy budgets. Because the task driving downstream decisions evolves over time, a task-specific codec is brittle and retraining one per task is infeasible in the field. We propose UniTAC, a single learned image codec spanning universal (task-agnostic) to task-specialized operation, re-targeted at runtime without retraining. The task is abstracted as a per-component importance vector, derived, e.g., from gradient attribution of any downstream model, and transmitted as low-overhead side information that conditions both encoder and decoder. Trained once over a broad, randomized family of such vectors against weighted-reconstruction distortion, UniTAC keeps a fixed backbone and a single human-viewable reconstruction whose fidelity is steered to the active task by swapping the injected vector. We analyze the underlying weighted rate-distortion problem, characterizing when a diagonal weighted distortion is task-consistent and how weights relate to task sensitivity. Guided by this, we design a Vision Transformer (ViT) codec whose token-level conditioning natively realizes this weight-driven code. On a localized task at 0.034 bpp, a single UniTAC model reaches 91.4% accuracy, only 1.9% below a task-based codec (93.3%) and above universal codecs (76.9%).
