EmbodiTTA: Resource-Efficient Test-Time Adaptation for Embodied Visual Systems
Organizations: Singapore Management University, Singapore · Samsung AI Center–Cambridge, United Kingdom · University of Cambridge, United Kingdom
Abstract
Continual Test-time adaptation (CTTA) continuously adapts the deployed model on every incoming batch of data. While achieving optimal accuracy, existing CTTA approaches present poor real-world applicability on resource-constrained edge devices, due to the substantial memory overhead and energy consumption. In this work, we first introduce a novel paradigm -- on-demand TTA -- which triggers adaptation only when a significant domain shift is detected. Then, we present OD-TTA, an on-demand TTA framework for accurate and efficient adaptation on edge devices. OD-TTA comprises three innovative techniques: 1) a lightweight domain shift detection mechanism to activate TTA only when it is needed, drastically reducing the overall computation overhead, 2) a source domain selection module that chooses an appropriate source model for adaptation, ensuring high and robust accuracy, 3) a decoupled Batch Normalization (BN) update scheme to enable memory-efficient adaptation with small batch sizes. Extensive experiments show that OD-TTA achieves comparable and even better performance while reducing the energy and computation overhead remarkably, making TTA a practical reality.
Figures & tables
| Batch Size = 1 | Batch Size = 16 | Batch Size = 64 | |||||||
| Avg. Acc.(%) | Avg. Acc.(%) | Avg. Acc.(%) | |||||||
| Method | CIFAR10-C | ImageNet-C | Memory (MB) | CIFAR10-C | ImageNet-C | Memory (MB) | CIFAR10-C | ImageNet-C | Memory (MB) |
| Source | 59.5 | 26.9 | 242 | 59.5 | 26.9 | 358 | 59.5 | 26.9 | 809 |
| SAR | 68.3 | 35.1 | 429 | 70.4 | 35.2 | 1723 | 69.2 | 37.3 | 5780 |
| MECTA | 65.0 | 12.0 | 378 | 81.3 | 33.2 | 1231 | 81.3 | 33.7 | 3836 |
| ROID | 10.1 | 0.4 | 698 | 79.5 | 32.5 | 2850 | 82.8 | 37.8 | 9840 |
| Method | Batch size = 1 | Batch size = 16 | Batch size = 64 |
|---|---|---|---|
| Source | 69.9 | 69.9 | 69.9 |
| SAR | 76.8 | 75.9 | 74.6 |
| MECTA | 62.7 | 83.4 | 84.2 |
| ROID | 6.2 | 81.3 | 84.3 |
| L-TTA | 2.2 | 83.0 | 83.3 |
| SURGEON | 2.9 | 79.5 | 80.4 |
| SHIFT types | ||||
|---|---|---|---|---|
| Method | Day Night | Clear Foggy | Clear Rainy | Avg. |
| Source | 27.3 | 17.7 | 11.0 | 18.7 |
| SAR | 23.6 | 7.6 | 4.2 | 11.8 |
| MECTA | 24.2 | 20.6 | 15.4 | 20.1 |
| SURGEON | 30.1 | 23.6 | 18.2 | 24.0 |
| EATA-C | 24.7 | 21.5 | 16.7 | 20.9 |
| Lengths | 1000 | 3000 | 5000 | 7000 | 9000 | 10000 |
|---|---|---|---|---|---|---|
| SAR | 138 (1.2 ) | 414 | 690 | 965 | 1241 | 1379 (1.2 ) |
| MECTA | 160 (1.0 ) | 480 | 800 | 1120 | 1440 | 1600 (1.0 ) |
| ROID | 75 (2.1 ) | 226 | 376 | 526 | 676 | 749 (2.1 ) |
| L-TTA | 50 (3.2 ) | 143 | 240 | 333 | 424 | 477 (3.4 ) |
| SURGEON | 78 (2.1 ) | 227 | 376 | 531 | 678 | 770 (2.1 ) |
| EATA-C | 47 (3.4 ) | 136 | 227 | 317 | 408 | 453 (3.5 ) |
| Sequences | S1 | S2 | S3 | S4 | S5 | S6 | S7 | S8 | S9 | S10 | Mean Std. |
| Triggered | 15 | 13 | 15 | 14 | 13 | 14 | 15 | 13 | 13 | 14 | 13.9 0.8 |
| Missed | 1 | 2 | 0 | 1 | 2 | 1 | 0 | 2 | 2 | 1 | 1.2 0.7 |
| False | 1 | 0 | 0 | 0 | 0 | 0 | 0 | 0 | 0 | 0 | 0.1 0.3 |
| Delay | 301 | 569 | 262 | 837 | 322 | 279 | 477 | 531 | 488 | 523 | 459 167 |
| Avg. acc. | 79.9 | 79.3 | 80.5 | 79.5 | 79.5 | 80.2 | 80.2 | 78.7 | 77.7 | 79.7 | 79.5 0.8 |
| Adapt from. | Batch Size = 1 | Batch Size = 16 | Batch Size = 64 | |||
|---|---|---|---|---|---|---|
| CIFAR10-C | ImageNet-C | CIFAR10-C | ImageNet-C | CIFAR10-C | ImageNet-C | |
| Prev-Domain | 79.5 | 27.8 | 84.4 | 31.4 | 86.0 | 33.4 |
| Initial-Pool | 81.0 | 34.6 | 85.7 | 36.7 | 86.1 | 39.1 |
| Progressive-Pool | 81.7 | 34.5 | 85.1 | 37.5 | 86.0 | 37.7 |
| Method | MobileNetV2 | MobileViT | ||||
|---|---|---|---|---|---|---|
| BS = 1 | BS = 16 | BS = 64 | BS = 1 | BS = 16 | BS = 64 | |
| SAR | 0.1 | 21.4 | 25.7 | 0.1 | 30.6 | 32.9 |
| MECTA | 0.2 | 0.6 | 0.7 | 0.2 | 0.2 | 0.2 |
| ROID | 0.2 | 21.5 | 25.4 | 0.2 | 30.0 | 33.1 |
| SURGEON | 0.2 | 1.38 | 23.5 | 0.2 | 2.51 | 28.4 |
| EATA-C | 0.1 | 21.8 | 26.2 | 0.9 | 27.3 | 32.5 |
Test-Time Adaptation
Test-time adaptation updates or recalibrates pretrained models at inference to handle distribution shifts, often without target labels.
See all 322 papers