I Can't Believe TTA Is Not Better: When Test-Time Augmentation Hurts Medical Image Classification
Not on the map. Why?
Abstract (arXiv)
Test-time augmentation (TTA)--aggregating predictions over multiple augmented copies of a test input--is widely assumed to improve classification accuracy, particularly in medical imaging where it is routinely deployed in production systems and competition solutions. We present a systematic empirical study challenging this assumption across three MedMNIST v2 benchmarks and four architectures spanning three orders of magnitude in parameter count (21K to 11M). Our principal finding is that TTA with standard augmentation pipelines consistently degrades accuracy relative to single-pass inference, with drops as severe as 31.6 percentage points for ResNet-18 on pathology images. This degradation affects all architectures, including convolutional models, and worsens with more augmented views. The sole exception is ResNet-18 on dermatology images, which gains a modest +1.6%. We identify the distribution shift between augmented and training-time inputs--amplified by batch normalization statistics mismatch--as the primary mechanism. Our ablation studies show that augmentation strategy matters critically: intensity-only augmentations preserve more performance than geometric transforms, and including the original unaugmented image partially mitigates but does not eliminate the accuracy drop. These findings serve as a cautionary note for practitioners: TTA should not be applied as a default post-hoc improvement but must be validated on the specific model-dataset combination.
Key facts. No citing papers in the map yet (data through 7 Oct 2026). Builds on 9 of its 12 references that are in the map. First posted on arXiv 6 Apr 2026.
Builds on
9 of 12 references.
References · 9
- 2014Very Deep Convolutional Networks for Large-Scale Image Recognitioncited by 15,267
- 2020Scaling Laws for Neural Language Modelscited by 6,411
- 2017On Calibration of Modern Neural Networkscited by 4,860
- 2024Scaling LLM Test-Time Compute Optimally can be More Effective than Scaling Model Parameterscited by 1,928
- 2018Aleatoric uncertainty estimation with test-time augmentation for medical image segmentation with convolutional neural networkscited by 278
- 2020Better Aggregation in Test-Time Augmentationcited by 121
- 2020Greedy Policy Search: A Simple Baseline for Learnable Test-Time Augmentationcited by 61
- 2017Image Classification of Melanoma, Nevus and Seborrheic Keratosis by Deep Neural Network Ensemblecited by 23
- 2024Understanding Test-Time Augmentationcited by 23
Built on by
Not cited yet.