Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

TRUST: Test-Time Refinement using Uncertainty-Guided SSM Traverses

About

State Space Models (SSMs) have emerged as efficient alternatives to Vision Transformers (ViTs), with VMamba standing out as a pioneering architecture designed for vision tasks. However, their generalization performance degrades significantly under distribution shifts. To address this limitation, we propose TRUST (Test-Time Refinement using Uncertainty-Guided SSM Traverses), a novel test-time adaptation (TTA) method that leverages diverse traversal permutations to generate multiple causal perspectives of the input image. Model predictions serve as pseudo-labels to guide updates of the Mamba-specific parameters, and the adapted weights are averaged to integrate the learned information across traversal scans. Altogether, TRUST is the first approach that explicitly leverages the unique architectural properties of SSMs for adaptation. Experiments on seven benchmarks show that TRUST consistently improves robustness and outperforms existing TTA methods.

Sahar Dastani, Ali Bahri, Gustavo Adolfo Vargas Hakim, Moslem Yazdanpanah, Mehrdad Noori, David Osowiechi, Samuel Barbeau, Ismail Ben Ayed, Herve Lombaert, Christian Desrosiers• 2025

Related benchmarks

TaskDatasetResultRank
Image ClassificationPACS (test)
Average Accuracy69.9
279
Image ClassificationImageNet V2 (test)
Top-1 Accuracy64
232
Image ClassificationCIFAR-10-C
Accuracy77.5
179
Image ClassificationImageNet-R (test)--
170
Image ClassificationImageNet-C (val)
mCE53.4
105
Image ClassificationImageNet-S (test)
Top-1 Acc41.5
17
ClassificationCIFAR10-C (test)
Gaussian Noise63.1
8
Image ClassificationCIFAR100-C
Top-1 Accuracy54.3
8
Image ClassificationImageNet-C
Top-1 Accuracy0.561
8
ClassificationCIFAR100-C (test)
Gaussian Noise Error32.1
8
Showing 10 of 10 rows

Other info

Follow for update