StableTTA: Improving Vision Model Performance by Training-free Test-Time Adaptation Methods
Organizations: Department of Computer Science New York Institute of Technology
Abstract
Ensemble methods improve predictive performance but often incur high memory and computational costs. We identify an aggregation instability induced by nonlinear projection and voting operations. To address both efficiency challenges and this inconsistency, we propose StableTTA, a training-free test-time adaptation method with two variants. StableTTA-I targets coherent-batch inference settings, where temporally or semantically adjacent observations are likely to belong to the same class. Examples include burst photography, video streams, robotics perception, and industrial inspection. Under coherent-batch inference, StableTTA-I substantially improves prediction consistency and accuracy through variance-aware logit aggregation. StableTTA-II establishes feature-level cropping, enabling efficient logit aggregation with a single forward pass on a single model backbone. Experiments on ImageNet-1K across 71 models demonstrate that StableTTA-I consistently improves prediction accuracy under coherent-batch inference, while StableTTA-II provides lightweight and architecture-agnostic accuracy improvements with minimal computational overhead. These results suggest that inference-time semantic coherence and aggregation stability provide useful perspectives for improving practical test-time adaptation systems.
Figures & tables
| Regime | Assumption | Example Applications |
|---|---|---|
| IID inference | Independent samples | Standard image classification |
| Coherent-batch inference | Neighboring samples correlated | Burst imaging, inspection |
| Multi-view inference | Multiple views of same object | 3D recognition |
| Sequential inference | Temporal continuity | Video and robotics |
| Baseline (%) | StableTTA-II (%) | ||||
|---|---|---|---|---|---|
| Weight | Params | Acc@1 | Acc@5 | Acc@1 | Acc@5 |
| ConvNeXt_Tiny [ 18 , 36 ] | 28.6M | 82.520 | 96.146 | 82.570 | 96.218 |
| ConvNeXt_Small [ 18 , 36 ] | 50.2M | 83.616 | 96.65 | 83.692 | 96.778 |
| ConvNeXt_Base [ 18 , 36 ] | 88.6M | 84.062 | 96.87 | 84.208 | 96.952 |
| ConvNeXt_Large [ 18 , 36 ] | 197.8M | 84.414 | 96.976 | 84.524 | 97.128 |
| DenseNet121 [ 10 , 36 ] | 8.0M | 74.434 | 91.972 | 75.980 | 93.062 |
Appendix figures & tables12 assets
Supplementary material from the paper’s appendix.
Appendix
| Weight | Data Augmentation | Baseline | TTA Acc | Source paper |
|---|---|---|---|---|
| VGG(v5) | 10 Crop | 71.93 | [ 7 ] He et al. | |
| PReLU-net | 10 Crop | 75.73 | [ 7 ] He et al. | |
| plain-34 | 10 Crop | 71.46 | [ 7 ] He et al. | |
| ResNet50 | 10 Crop | 77.15 | [ 7 ] He et al. | |
| ResNet101 | 10 Crop | 78.25 | [ 7 ] He et al. | |
| ResNet152 | 10 Crop | 78.57 | [ 7 ] He et al. |
| Random Data | Hard-vot. (%) | Soft-vot. (%) | Logit-avg. (%) | ||||
|---|---|---|---|---|---|---|---|
| Model | Augmentation | Acc@1 | Acc@5 | Acc@1 | Acc@5 | Acc@1 | Acc@5 |
| AlexNet (# Params = 61.1 M) (7.1 GFLOPS) ( ) | baseline | 56.522 | 79.066 | 56.522 | 79.066 | 56.522 | 79.066 |
| flip | 56.630 | 56.760 | 57.306 | 79.674 | 57.322 | 79.714 | |
| crop | 55.016 | 55.158 | 55.460 | 78.240 | 55.002 | 78.082 | |
| affine | 51.176 | 51.342 | 51.966 | 75.262 | 51.116 | 74.564 | |
| erasing | 56.552 | 56.688 | 56.574 | 78.914 | 56.508 | 79.070 | |
| Peak | Baseline (%) | StableTTA-I (%) | ||||
|---|---|---|---|---|---|---|
| Weight | Params | GFLOPS | Acc@1 | Acc@5 | Acc@1 | Acc@5 |
| AlexNet [ 14 , 36 ] | 61.1M | 0.71 | 56.522 | 79.066 | 89.332 | 98.028 |
| ConvNeXt_Tiny [ 18 , 36 ] | 28.6M | 4.46 | 82.52 | 96.146 | 95.958 | 99.884 |
| ConvNeXt_Small [ 18 , 36 ] | 50.2M | 8.68 | 83.616 | 96.65 | 95.932 | 99.914 |
| ConvNeXt_Base [ 18 , 36 ] | 88.6M | 15.36 | 84.062 | 96.87 | 95.964 | 99.906 |
| ConvNeXt_Large [ 18 , 36 ] | 197.8M | 34.36 | 84.414 | 96.976 | 95.998 | 99.932 |
| TTA (%) | StableTTA-I (%) | |||||
|---|---|---|---|---|---|---|
| Model | GFLOPS | Acc@1 | Acc@5 | Acc@1 | Acc@5 | |
| AlexNet (# Params = 61.1M) | 4 | 2.84 | 65.590 | 87.022 | 78.750 | 93.434 |
| 8 | 5.68 | 68.604 | 89.432 | 84.810 | 96.216 | |
| 16 | 11.36 | 70.496 | 90.510 | 87.864 | 97.470 | |
| 32 | 22.72 | 71.258 | 90.938 | 89.532 | 98.030 | |
| ResNet50 † (# Params = 25.6M) | 4 | 16.36 | 88.556 | 98.858 | 91.276 | 99.108 |
| Model | Acc@1 | Acc@5 | Acc@1 | Acc@5 | Acc@1 | Acc@5 | Acc@1 | Acc@5 | |
|---|---|---|---|---|---|---|---|---|---|
| AlexNet (61.1M) (0.71GFLOPS) | 1 | 78.412 | 93.182 | 84.638 | 96.134 | 87.942 | 97.530 | 89.576 | 98.052 |
| 2 | 78.444 | 93.262 | 84.982 | 96.312 | 87.946 | 97.528 | 89.496 | 98.052 | |
| 5 | 78.804 | 93.328 | 84.666 | 96.220 | 87.984 | 97.412 | 89.484 | 98.108 | |
| 10 | 78.750 | 93.434 | 84.810 | 96.216 | 87.864 | 97.470 | 89.532 | 98.030 | |
| 20 | 78.540 | 93.320 | 84.778 | 96.160 | 88.072 | 97.540 | 89.530 | 98.076 | |
| AlexNet (%) | ResNet50 † (%) | ViT_B_16 (%) | ||||
| Acc@1 | Acc@5 | Acc@1 | Acc@5 | Acc@1 | Acc@5 | |
| 89.492 | 98.034 | 95.596 | 99.862 | 95.888 | 99.868 | |
| 89.558 | 98.056 | 95.622 | 99.854 | 95.830 | 99.898 | |
| 89.642 | 98.020 | 95.620 | 99.868 | 95.916 | 99.896 | |
| 89.620 | 98.070 | 95.684 | 99.842 | 95.848 | 99.882 | |
| 89.588 | 98.096 | 95.666 | 99.858 | 95.876 | 99.878 | |
| Model | Head | Crops |
|---|---|---|
| ConvNeXt_Tiny | AvgPool+Norm+Linear(768,1000) | |
| ConvNeXt_Small | AvgPool+Norm+Linear(768,1000) | |
| ConvNeXt_Base | AvgPool+Norm+Linear(1024,1000) | |
| ConvNeXt_Large | AvgPool+Norm+Linear(1536,1000) | |
| DenseNet121 | ReLU+AvgPool+Linear(1024,1000) | |
| DenseNet161 | ReLU+AvgPool+Linear(2208,1000) |
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