Single Image Test-Time Adaptation via Multi-View Co-Training
This addresses the need for real-time, per-patient adaptation in clinical settings where large target datasets are unavailable, though it is incremental as it builds on existing test-time adaptation techniques.
The paper tackles the problem of adapting trained models to new domains with only a single test-time image, particularly in medical imaging, by proposing a patch-based multi-view co-training method that achieves performance close to supervised benchmarks and outperforms state-of-the-art methods by an average Dice Similarity Coefficient of 3.75%.
Test-time adaptation enables a trained model to adjust to a new domain during inference, making it particularly valuable in clinical settings where such on-the-fly adaptation is required. However, existing techniques depend on large target domain datasets, which are often impractical and unavailable in medical scenarios that demand per-patient, real-time inference. Moreover, current methods commonly focus on two-dimensional images, failing to leverage the volumetric richness of medical imaging data. Bridging this gap, we propose a Patch-Based Multi-View Co-Training method for Single Image Test-Time adaptation. Our method enforces feature and prediction consistency through uncertainty-guided self-training, enabling effective volumetric segmentation in the target domain with only a single test-time image. Validated on three publicly available breast magnetic resonance imaging datasets for tumor segmentation, our method achieves performance close to the upper bound supervised benchmark while also outperforming all existing state-of-the-art methods, on average by a Dice Similarity Coefficient of 3.75%. We publicly share our accessible codebase, readily integrable with the popular nnUNet framework, at https://github.com/smriti-joshi/muvi.git.